문제
상근이는 가위 바위 보를 할 수 있는 로봇을 가지고 있다. 게임은 총 k번의 연속적인 라운드로 이루어져 있다.
각 로봇은 길이가 k인 문자열을 가지고 있고, 이 문자열에 의해서 게임을 하게 된다. 각 라운드에서 패배한 로봇은 더 이상 게임을 참여할 수 없고, 남은 로봇끼리 게임을 하게 된다. 로봇이 하나만 남은 경우에는 그 로봇이 게임을 승리하게 되고, 게임이 끝난다. k 라운드가 지난 후에 남아있는 로봇이 하나보다 많다면, 게임은 무승부로 끝난다.
가위는 S, 바위는 R, 보는 P로 나타낼 수 있다. 만약, 로봇이 가지고 있는 문자열이 RSPSRSSP 라면, 첫 라운드에서 로봇은 주먹을 내고, 둘째 라운드에서는 가위를 낸다. 여덟번째 라운드까지 그 로봇이 살아있다면 보를 낸다.
가위 바위 보 게임에 참여하는 로봇의 수와 각 로봇의 문자열이 주어졌을 때, 게임의 승자를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 참여하는 로봇의 수 N이 주어진다. 다음 N개 줄에는 각 로봇에 저장되어 있는 문자열이 주어진다. 모든 로봇의 문자열의 길이는 k이다. (2 ≤ N ≤ 10, 3 ≤ k ≤ 30) 로봇은 주어지는 순서대로 1번부터 번호를 매긴다.
출력
각 테스트 케이스마다, 게임을 승리한 로봇의 번호를 출력한다. k 라운드가 지난 후에도 승자가 없는 경우 (무승부) 에는 0을 출력한다.
내 풀이
import sys
input = sys.stdin.readline
t = int(input())
for _ in range(t):
n = int(input())
robots = [(y+1, input().strip()) for y in range(n)]
for i in range(len(robots[0][1])):
round = [x[1][i] for x in robots]
if len(round) <= 1:
break
# tie
if set(round) == {"R", "S", "P"} or set(round) == {"R"} or set(round) == {"S"} or set(round) == {"P"}:
pass
# "R" wins
elif "R" in round and "S" in round and "P" not in round:
remove = [x for x in robots if x[1][i]=="S"]
for r in remove:
robots.remove(r)
# "S" wins
elif "R" not in round and "S" in round and "P" in round:
remove = [x for x in robots if x[1][i] == "P"]
for r in remove:
robots.remove(r)
# "P" wins
elif "R" in round and "S" not in round and "P" in round:
remove = [x for x in robots if x[1][i] == "R"]
for r in remove:
robots.remove(r)
if len(robots) >= 2:
print(0) # tie
elif len(robots) == 1:
print(robots[0][0])코멘트
먼저 경우의 수를 잘 정리해보자.
비기는 경우
가위, 바위, 보가 모두 포함된 경우
모두가 (가위/바위/보)를 낸 경우
바위가 이기는 경우 : 바위와 가위만 포함되어 있고, 보는 없음
가위가 이기는 경우 : 가위와 보만 포함되어 있고, 바위는 없음
보가 이기는 경우 : 바위와 보만 포함되어 있고, 가위는 없음
이 경우의 수를 잘 구현하면 된다.
이 때 탈락한 로봇을 없앨 때 pop()을 사용했는데 그러면 먼저 삭제된 요소 때문에 다음에 요소를 삭제할 때 인덱스가 바뀌면서 엉뚱한 요소가 삭제되어서 오답을 받았다. 엉뚱한 로봇을 탈락시키지 않게 조심하자.