[BOJ/백준] 5356 - Triangles [python]

문제 링크

5356번: Triangles

문제 풀이

이 문제도 단순한 출력 문제이다. 미리 A부터 Z까지를 순서대로 문자열에 추가한 뒤에 사용해준다. 만약 출력하다가 Z이상으로 인덱스가 넘어간다면, 다시 인덱스를 0으로 만들어서 처음부터 돌아갈 수 있도록 해준다.

정답 코드

alpha = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
for _ in range(int(input())) :
    n, l = input().split()
    idx = alpha.index(l)
    for i in range(1, int(n) + 1) :
        print(alpha[idx] * i)
        idx += 1
        if idx == 26 : idx = 0
    print()