[파이썬 실습] 심화 문제_당근 탐지기

당근 탐지기

땅 속에 숨겨져 있는 당근을 찾기 좋아하는 엘리스 토끼는 당근을 탐지할 수 있는 당근 탐지기를 가지고 왼쪽 혹은 오른쪽으로만 갈 수 있는 길이가 5인 길 어디인가에 떨어졌습니다.

예를 들어 아래와 같이 5칸으로 구성된 길이 있고 O은 당근이 있는 곳, X은 당근이 없는 곳이며 엘리스 토끼가 왼쪽에서 세번째 칸에 떨어졌다고 가정합니다. 이때 왼쪽 끝으로 이동하게 되면 총 1개의 당근을 획득할 수 있으며 오른쪽으로 이동한 경우 2개의 당근을 획득할 수 있습니다.

• O와 X는 알파벳 대문자 O, X를 의미합니다.

지시사항을 참고하여 코드를 작성하세요.

 

지시사항

1. 첫번째 줄에 사용자로부터 O, X 그리고 공백으로 구성된 문자열을 입력 받으세요.
2. 두번째 줄에 엘리스 토끼가 떨어진 위치(왼쪽으로부터 떨어진 위치)를 입력받으세요.
   • 엘리스 토끼는 당근이 없는 곳에서만 떨어집니다.
3. 떨어진 위치를 기준으로 당근을 최대한 많이 획득할 수 있는 방향(왼쪽, 오른쪽)을 출력하세요.
   • 왼쪽과 오른쪽의 당근의 수가 동일한 경우 동일을 출력하세요.

입력 예시

O X X O O
3
Copy

출력 예시

오른쪽
Copy

Tips

사용자로부터 입력을 받을 때 입력을 요구하는 안내 내용이 있거나 출력 예시 이외의 다른 내용이 포함되어 출력되는 경우 오답 처리될 수 있습니다.

---

정답

# 지시사항을 참고하여 코드를 작성하세요.
arr = input().split()
location = int(input())

# O X X O O
# 3

#왼쪽
def left(a):
    count = 0
    i = a-2
    while i >= 0:
        if arr[i] == 'O': count += 1
        i -= 1
    
    return count

# 오른쯕
def right(a):
    count = 0
    i = a
    while i < len(arr):
        if arr[i] == 'O': count += 1
        i += 1
    
    return count

if left(location) > right(location) : print('왼쪽')
elif left(location) < right(location) : print('오른쪽')
else : print('동일')

이문제를 풀때 보기 편하게 짜고 싶어서 함수를 이용해서 풀었습니다.