[acmicpc] 17615. 볼 모으기(python)

import sys
input=sys.stdin.readline

N=int(input())
balls=input().strip()
def find_index(color): # 맨 우측과 좌측 중, 볼이 많이 모여있는 곳 찾기
    left, right = 0, 0
    for i in range(N):
        if balls[i] != color: break
        left += 1
    for i in range(N - 1, -1, -1):
        if balls[i] != color: break
        right += 1
    if left<right:
        return find_move(False, right, color) #맨 우측
    else:
        return find_move(True, left, color) #맨 좌측
        
def find_move(loc, cnt, color): # 옮길 볼의 개수 구하기
    if loc:
        return balls[cnt:].count(color)
    else:
        return balls[:N - cnt].count(color)

red=find_index('R')
blue=find_index('B')
print(red if red<blue else blue)

 

---

 

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

17615번: 볼 모으기