코딩/공부

[Python] 백준 5217번 - 쌍의 합

취미니스트 2024. 1. 31. 14:57
728x90
반응형

문제

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

 

입력받은 자연수만큼 반복.

- 정수 하나 입력.

- 1부터 입력받은 자연수까지 두 자연수 쌍을 만들어 리스트에 저장.

- 쌍의 합이 입력한 자연수와 같은 쌍만을 filter를 통해 얻음.

- generate 문법과 join을 통해 쌍들을 하나의 문자열로 이어붙임

for _ in range(int(input())):
    n = int(input())
    l = []
    for i in range(1, n+1):
        for j in range(i+1, n+1):
            l.append((i, j))
    l = list(filter(lambda x: sum(x) == n, l))
    print(f'Pairs for {n}: {", ".join([f"{t[0]} {t[1]}" for t in l])}')
728x90
반응형