본문 바로가기
Algorithm

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

by jangThang 2022. 4. 6.
반응형

백준 온라인 저지

 

[ Contents ]

     

     

    1. 문제 (링크 참조)

     

    15651번: N과 M (3)

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

    www.acmicpc.net

     

     

     

    2. 문제 풀이

     수열에서 M개를 뽑는 중복 순열을 구하는 문제입니다. 중복 순열은 itertools의 product() 함수를 이용해서 구할 수 있습니다.

     

     

     

    3. 코드

    from itertools import product
    
    N, M = map(int, input().split())
    case = product(range(1, N+1), repeat=M)
    
    for i in case:
        for j in i:
            print(j, end=" ")
        print()

     product() 함수를 이용해서 수열에서 M개를 뽑은 중복 순열을 구합니다.

     

     

    itertools.product(lst, repeat=n): lst에서 n개를 중복해서 뽑은 순열

     

    star가 되고나서 Tistory

    반응형

    댓글