본문 바로가기

코딩테스트/백준

[Python] 백준 15650번. N과 M(2)

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

 

15650번: N과 M (2)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

문제 설명

예제2 출력 결과를 보면 조합이라는 것을 바로 알 수 있다.

1 2
1 3
1 4
2 3
2 4
3 4

 

풀이설계

파이썬 itertools모듈의 combinations 를 사용하면 금방 풀 수 있다.

 


import sys
from itertools import combinations

input = sys.stdin.readline
n, m = map(int, input().split())
lst = [i for i in range(1,n+1)]
for ele in combinations(lst, m):
    #print(str(list(ele)).replace("[", "").replace("]", "").replace(",", ""))
    print(*ele)