[BOJ][C++] 백준 28702번: FizzBuzz (Bronze I)

https://www.acmicpc.net/problem/28702

문제

FizzBuzz 문제는 $i = 1, 2, \cdots$ 에 대해 다음 규칙에 따라 문자열을 한 줄에 하나씩 출력하는 문제입니다.
FizzBuzz 문제에서 연속으로 출력된 세 개의 문자열이 주어집니다. 이때, 이 세 문자열 다음에 올 문자열은 무엇일까요?

입력

FizzBuzz 문제에서 연속으로 출력된 세 개의 문자열이 한 줄에 하나씩 주어집니다. 각 문자열의 길이는 $8$ 이하입니다. 입력이 항상 FizzBuzz 문제에서 연속으로 출력된 세 개의 문자열에 대응됨이 보장됩니다.

출력

연속으로 출력된 세 개의 문자열 다음에 올 문자열을 출력하세요. 여러 문자열이 올 수 있는 경우, 아무거나 하나 출력하세요.

 

풀이

입력받은 문자열들 중 숫자를 찾으면

그 숫자를 기준으로 다음(또는 다다음 또는 다다다음) 인덱스를 구해서

해당 인덱스에 맞는 문자열을 출력하면 끝

def get_fb_with_i(i):
    if i%3 == 0:
        if i%5 == 0:
            return "FizzBuzz"
        else:
            return "Fizz"
    else:
        if i%5 == 0:
            return "Buzz"
        else:
            return i

for i in range(3):
    string = input()
    
    if string.isdigit():
        idx = int(string) + 3 - i
        print(get_fb_with_i(idx))
        break