[SWEA] D3 - 13547. 팔씨름 in 파이썬

문제 출처

SW Expert Academy

 

문제

소정이와 세정이는 점심 값을 누가 낼 지 정하기 위해 팔씨름을 하기로 했다. 공정하고 재밌는 경기를 위해 둘은 15번 팔씨름을 하여 8번 이상 이기는 사람이 점심 값을 면제받기로 하였다.

둘은 지금까지 k번의 팔씨름을 진행했다. 이 결과는 길이가 k인 ‘o’ 또는 ‘x’로만 구성된 문자열 S[1..k]로 나타낼 수 있다. S[i]가 ‘o’면 소정이가 i번째 경기에서 승리했다는 것이고, ‘x’면 패배했다는 것이다.

소정이는 앞으로 팔씨름을 15번째 경기까지 진행했을 때 자신이 점심값을 면제받을 가능성이 있는지 알고자 한다. 이를 대신해 주는 프로그램을 작성하라.

[입력]
첫 번째 줄에 테스트 케이스의 수 T가 주어진다.
각 테스트 케이스는 하나의 줄로 이루어진다. 각 줄에는 ‘o’ 또는 ‘x’로만 구성된 길이가 1 이상 15 이하인 문자열 S가 주어진다.

[출력]
각 테스트 케이스마다, 소정이가 점심값을 면제받을 가능성이 있다면 ‘YES’, 없다면 ‘NO’를 출력한다.

 

 

문제 풀이

 15번 중 8번 이상 이겨야 점심값을 면제받을 수 있다는 것은 다른 말로 하면 15번 중 15-8=7번 이하로 졌을 때까지는 면제 가능성이 있다는 것과 같다.

 그러므로 문자열을 입력받은 뒤, x의 개수를 세고 만약 x의 개수가 15-8=7 번 이하일때는 면제가능성을 뜻하는 'YES'를 출력한다.

 

작성 코드

import sys
sys.stdin = open("input.txt", "r")

T = int(input())
for test_case in range(1, T+1):
    S = input()
    cnt = 0
    for _ in S:
        if _ == 'x':    cnt += 1

    if cnt <= 15 - 8: print(f'#{test_case} YES')
    else: print(f'#{test_case} NO')