니노니나니

[백준/18698번] The Walking Adam - B4/Python 본문

알고리즘/백준

[백준/18698번] The Walking Adam - B4/Python

SangJunni 2024. 6. 4. 23:38

https://www.acmicpc.net/problem/18698

문제

이제 막 걸음마를 떼기 시작한 아담은, 자주 넘어지고 있습니다. 아직 아담은 손을 번쩍 들어 올린 상태여야만 걸을 수 있고, 손을 내리게 되면 균형을 잃고 넘어지고 맙니다.

아담의 행동을 나타내는 'U' 와 'D'로만 구성된 문자열이 주어집니다. 각각 'U'는 아담이 손을 들어 올린 상태임을, 'D'는 아담이 손을 내린 상태임을 의미합니다.

아담의 걸음마 연습을 도와주기 위해, 아담이 처음으로 넘어지기 전까지 몇 걸음을 걸었는지 계산해주는 프로그램을 작성해봅시다.

입력

입력의 첫 줄에는 테스트 케이스의 개수를 뜻하는 정수 T (1 ≤ T ≤ 100)가 주어집니다.

각 테스트 케이스에는 아담의 행동을 시간순으로 나타낸 최대 100자 이내의 문자열 한 줄이 입력됩니다. 해당 문자열은 공백이 존재하지 않고 'U'와 'D'로만 구성되어 있습니다.

풀이

t = int(input())
for _ in range(t):
    line = input()
    if 'D' in line:
        print(line.find('D'))
    else:
        print(len(line))

해결방법

D가 입력에 존재하는지 확인해서 D가 존재하는 경우 D의 인덱스를 출력하고 아닌 경우 입력된 문자열의 전체 길이를 반환하면 되는 문제.