본문 바로가기
Algorithm

[Brute Force] 백준 15655 N과 M (6) - 파이썬(Python)

by jangThang 2022. 4. 7.
반응형

백준 온라인 저지

 

[ Contents ]

     

     

    1. 문제 (링크 참조)

     

    15655번: N과 M (6)

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

    www.acmicpc.net

     

     

     

    2. 문제 풀이

     주어진 수열에서 M개를 뽑는 조합을 구하는 문제입니다.

     

    itertools.combinations(lst, n): lst에서 n개를 뽑는 조합

     itertools 라이브러리의 combinations() 함수를 사용하면 쉽게 구할 수 있습니다.

     

     

     

    3. 코드

    from itertools import combinations
    
    N, M = map(int, input().split())
    numlist = list(map(int, input().split()))
    case = combinations(sorted(numlist), M)
    
    for i in case:
        for j in i:
            print(j, end=" ")
        print()

     

     

    star가 되고나서 Tistory

    반응형

    댓글