본문 바로가기
알고리즘

[BAEKJOON] 17615 볼 모으기

by mAlfred 2023. 4. 30.
반응형

문제

빨간색 볼과 파란색 볼이 <그림 1>에서 보인 것처럼 일직선상에 섞여 놓여 있을 때, 볼을 옮겨서 같은 색 볼끼리 인접하게 놓이도록 하려고 한다. 볼을 옮기는 규칙은 다음과 같다.

  1. 바로 옆에 다른 색깔의 볼이 있으면 그 볼을 모두 뛰어 넘어 옮길 수 있다. 즉, 빨간색 볼은 옆에 있는 파란색 볼 무더기를 한 번에 뛰어 넘어 옮길 수 있다. 유사하게, 파란색 볼은 옆에 있는 빨간색 볼 무더기를 한 번에 뛰어 넘어 옮길 수 있다.
  2. 옮길 수 있는 볼의 색깔은 한 가지이다. 즉, 빨간색 볼을 처음에 옮겼으면 다음에도 빨간색 볼만 옮길 수 있다. 유사하게, 파란색 볼을 처음에 옮겼으면 다음에도 파란색 볼만 옮길 수 있다.

예를 들어, 처음에 볼이 <그림 1>에서 보인 것처럼 있을 때, 빨간 볼을 <그림 2>에서 보인 것처럼 옮긴 후, <그림 3>에서 보인 것처럼 옮긴다면 두 번 만에 같은 색끼리 모을 수 있다.

<그림 1>
<그림 2>
<그림 3>

반면에 파란색 볼을 선택하여 에서 보인 것처럼 옮기면(화살표에 있는 수는 옮기는 순서를 나타낸다) 네 번을 옮겨야 같은 색의 볼끼리 모을 수 있다.

<그림 4>

일직선상에 놓여 있는 볼에 관한 정보가 주어질 때, 규칙에 따라 볼을 이동하여 같은 색끼리 모으되 최소 이동횟수를 찾는 프로그램을 작성하시오.

입력

첫 번째 줄에는 볼의 총 개수 N이 주어진다. (1 ≤ N ≤ 500,000) 다음 줄에는 볼의 색깔을 나타내는 문자 R(빨간색 볼) 또는 B(파란색 볼)가 공백 없이 주어진다. 문자열에는 R 또는 B 중 한 종류만 주어질 수도 있으며, 이 경우 답은 0이 된다.

출력

최소 이동횟수를 출력한다.

서브태스크

번호배점제한
1 15 N ≤ 10
2 22 N ≤ 1,000
3 14 N ≤ 500,000, 처음 상태에서 모든 파란 공은 연속해서 존재한다.
4 49 원래의 제약조건 이외에 아무 제약조건이 없다.

이 문제 같은 경우 쉽게 접근을 했다.

걍 모든 경우의 수를 다 뽑아낸 뒤 작은 이동 값을 추출하는 방법.

 

오른쪽으로 빨간 공 모으고 파란 공 모으고 횟수 

왼쪽으로 빨간 공 모으로 파란 공 모으고 횟수

 

모두 조합해서 제일 작은 횟수 출력

 

n = int(input())
b = list(map(str, input()))

bb = 0
rb = 0
cnt = 0
cnts = []

for i in range(n):
    if(b[i] == 'R'):
        rb += 1
   
    if(b[i] == 'B' and rb):
        cnt += rb
        rb = 0
   
cnts.append(cnt)
cnt = 0
bb = 0
rb = 0

for i in range(n):
    if(b[i] == 'B'):
        bb += 1
   
    if(b[i] == 'R' and bb):
        cnt += bb
        bb = 0
   
cnts.append(cnt)
cnt = 0
bb = 0
rb = 0

b.reverse()

for i in range(n):
    if(b[i] == 'R'):
        rb += 1
   
    if(b[i] == 'B' and rb):
        cnt += rb
        rb = 0

cnts.append(cnt)
cnt = 0
bb = 0
rb = 0

for i in range(n):
    if(b[i] == 'B'):
        bb += 1
   
    if(b[i] == 'R' and bb):
        cnt += bb
        bb = 0
   
cnts.append(cnt)
cnt = 0
bb = 0
rb = 0

print(min(cnts))

 

반응형

'알고리즘' 카테고리의 다른 글

[BAEKJOON] 2606 바이러스  (0) 2023.05.19
[BAEKJOON] 1260 DFS와 BFS  (0) 2023.05.17
[BAEKJOON] 2178 미로 탐색  (0) 2023.05.14
[BAEKJOON] 2667 단지번호붙이기  (0) 2023.05.14
[BAEKJOON] 17614 369  (0) 2023.04.29