[Baekjoon] 2490 Python

예제 입력1

0 1 0 1
1 1 1 0
0 0 1 1

예제 출력 1

B
A
B

 

풀이과정

윷의 앞뒷면의 갯수를 세어주면 되는 단순한 문제.
처음엔 for문 안에 if문으로 분기해서 카운트 해줬다. 두번째 코드는 팀원이 푼 방법인데 코드가 깔끔하고 좋은 것 같아서 참고.

import sys; input = sys.stdin.readline

for _ in range(3):
    game = list(map(int, input().rstrip().split()))
    if game.count(0) == 1:
        print('A')
    elif game.count(0) == 2:
        print('B')
    elif game.count(0) == 3:
        print('C')
    elif game.count(0) == 4:
        print('D')
    else:
        print('E')

import sys; input = sys.stdin.readline

for _ in range(3):
    yut = list(map(int, input().split()))
    print({0: 'E', 1: 'A', 2: 'B', 3: 'C', 4: 'D'}.get(yut.count(0)))