본문 바로가기
알고리즘/BOJ

[BOJ / 백준] 15654번 N과 M (5) python / 파이썬 조합

by seohmoon 2022. 7. 28.

출처 : https://www.acmicpc.net/problem/15654

 

15654번: N과 M (5)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열

www.acmicpc.net


한 개의 리스트 안에서 나올 수 있는 모든 조합 구하기

우선 itertools에서 permutations 가져오기!!

 이걸 이용해서 모든 조합 리스트를 구해준다.

 

그리고 for문을 돌면서 그 안의 값만 출력해준다.

# BOJ15654 N과 M (5)
from itertools import permutations

N, M = map(int, input().split())
lst = list(map(int, input().split()))

ans = list(permutations(lst, M))
ans.sort()

for i in ans:
    print(*i, sep=" ")

댓글