[백준] 13413번 오셀로 재배치 [Python]

 

 

오셀로 말$($2개$)$의 위치를 바꾸거나, 말 1개를 뒤집기

위 두 가지 작업만으로 초기 상태 -> 목표 상태 까지의 최소 작업 횟수 구하는 문제

 

각 상태는 W와 B가 이어진 문자열 형태로 주어짐

 

오셀로 게임

 

문제

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

초기 상태	목표 상태
○●●○○	○●○●○

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

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

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

입력

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

출력

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

13413번: 오셀로 재배치 (acmicpc.net)

 

W와 B를 1과 0으로 치환한 후 풀이

import sys
T = int(sys.stdin.readline())
for _ in range(T):
    sys.stdin.readline()
    # 입력 받은 문자열을 W는 1로 B는 0으로 바꿈
    org = [1 if s=="W" else 0 for s in list(sys.stdin.readline().rstrip())]
    goal = [1 if s=="W" else 0 for s in list(sys.stdin.readline().rstrip())]
    
    # 두 리스트를 순서대로 비교하며 다른 원소의 개수를 셈
    diff = sum([1 if o!=g else 0 for o,g in zip(org, goal)])
    
    # 1의 개수 차이
    temp = abs(sum(org) - sum(goal))

    # temp + (diff-temp)//2
    # 1과 0의 개수를 맞추고(2번 작업) -> temp 수만큼
    # 다른 부분을 서로 바꿔치기(1번 활동) -> (diff-temp)//2
    count = (diff+temp)//2 
    print(count)

 

문자열로 풀이

import sys
T = int(sys.stdin.readline())
for _ in range(T):
    sys.stdin.readline()
    org = sys.stdin.readline().rstrip()
    goal = sys.stdin.readline().rstrip()
    diff = sum([1 if o!=g else 0 for o,g in zip(org, goal)])
    temp = abs(org.count("W") - goal.count("W"))
    count = (diff+temp)//2 
    print(count)

 

예제 입력 1 

3
5
WBBWW
WBWBW
7
BBBBBBB
BWBWBWB
4
WWBB
BBWB

예제 출력 1 

1
3
2