[백준 13413 파이썬] 그리디 알고리즘 오셀로 재배치

문제

로봇을 좋아하는 세희는 로봇동아리에서 카메라와 센서, 라즈베리 파이, 집게발을 이용해 로봇을 완성하였다. 이 로봇을 통해서 오셀로 재배치라는 작업을 하려고 한다. 오셀로 말은 앞면이 검정, 뒷면이 흰색으로 된 말이다. 세희의 목표는 로봇을 이용하여 처음 배치된 오셀로 말을 주어진 형태로 바꾸는 일을 하는 것이다. 아래의 예시를 참고하자.

초기 상태목표 상태

○●●○○

○●○●○

세희는 로봇을 이용해 2가지 작업 중 하나를 골라 진행할 수 있다.

1. 배치된 말 중 임의의 2개의 말을 골라 서로의 위치를 바꾼다.
2. 말 1개를 들어 뒤집어 놓아 색상을 변경한다.

위의 예시에서, 3번째와 4번째 말을 2번 작업을 통해 각각 뒤집으면 2번의 작업으로 목표 상태를 만들 수 있다. 하지만 1번 작업을 통해 3번째와 4번째 말을 골라 서로의 위치를 바꾸어주면 1번 만에 목표 상태에 도달할 수 있다. 초기 상태의 말과 목표 상태의 말이 주어질 때, 목표 상태에 도달할 수 있는 최소 횟수를 구하는 프로그램을 작성하시오.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 각 입력의 첫 번째 줄에는 오셀로 말의 개수 N(1≤N≤100,000)이 주어진다. 각 입력의 두 번째 줄과 세 번째 줄에는 각각 오셀로 말의 초기 상태와 목표 상태가 주어진다. 초기 상태와 목표 상태의 말의 개수는 항상 N과 일치한다. 흰색 면이 보이는 경우에는 W, 검은색 면이 보이는 경우에는 B로 주어진다.

출력

출력은 표준 출력을 사용한다. 입력받은 데이터에 대해, 한 줄에 1개씩 초기 상태에서 목표 상태를 만들기 위한 작업의 최소 횟수를 구한다.

풀이

최소한의 움직임으로 오셀로의 초기상태를 목표상태로 만들어야 한다.

움직임으로는 두 개의 자리를 바꾸거나 하나를 뒤집는 방법이 있다.

움직임을 최소로 만들기 위해서는 두 개의 자리를 바꿔서 목표상태로 만드는 것이 좋다.

이 문제를 해결하기 위해 입력값들에 대해 반복문을 돌려서 다른 돌이 있으면 각각의 리스트에 추가하고 정렬한다.

정렬한 두 리스트를 비교하여 순회 할 때 같은 돌이라면 반대 돌의 같은것이 분명히 존재한다.

따라서 두 개의 자리를 바꾸는 것으로 목표 상태를 만들 수 있다. 따라서 count에 0.5씩 더해준다.

순회 할 때 다른 돌이라면 반드시 돌을 뒤집어서 목표상태를 만들 수 있다. 따라서 count에 1씩 더해준다.

예를 들어 살펴보면,

WBBWW, WBWBW 가 입력값으로 주어지면

각 정렬한 리스트는 BW, BW 가 된다. 두개의 인덱스는 모두 같은 값이므로 0.5가 두 번 더해져 결과값이 1이된다.

BBBBBBB, BWBWBWB 가 입력값으로 주어지면

각 정렬한 리스트는 BBB, WWW 가 되고 3개의 인덱스가 모두 다른값이므로 결과값은 3이 된다.

 

import sys
testCases = int(sys.stdin.readline().rstrip("\n"))
for _ in range(testCases):
    n = int(sys.stdin.readline().rstrip("\n"))
    mapSt1 = sys.stdin.readline().rstrip("\n")
    mapSt2 = sys.stdin.readline().rstrip("\n")
    count=0
    Bcount=0
    Wcount=0
    ans=0
    mapList1=[]
    mapList2=[]
    for i in range(n):
        if(mapSt1[i]!=mapSt2[i]):
            mapList1.append(mapSt1[i])
            mapList2.append(mapSt2[i])


    mapList1 = sorted(mapList1)
    mapList2 = sorted(mapList2)
    for i in range(len(mapList1)):
        if (mapList1[i] == mapList2[i]):
            ans += 0.5
        else:
            ans += 1
    print(int(ans))