[Python] 백준 15656, 15657번 - N과 M (7), (8)

N과 M (7)

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

15656번: N과 M (7)

 

앞서 풀었던 n과 m 문제들과 유사하나, 같은 수를 여러 번 골라도 되기 때문에 dfs 함수의 for문에서 if i not in ans: 조건문을 삭제해주면 된다.

 

import sys
input = sys.stdin.readline

def dfs():
    if len(ans) == m:
        print(*ans)
        return

    for i in num:
        ans.append(i)
        dfs()
        ans.pop()

n, m = map(int, input().split())
num = list(map(int, input().split()))
num.sort()
ans = []
dfs()

 

 

 

 

 

 

---

 

 

N과 M (8)

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

15657번: N과 M (8)

 

고른 수열이 비내림차순이어야 하므로 시작하는 숫자를 정해서 그것보다 작은 숫자는 재귀 루프에 들어가지 않도록 한다.

 

import sys
input = sys.stdin.readline

def dfs(s): #s: start
    if len(ans) == m:
        print(*ans)
        return

    for i in num:
        if i >= s:
            ans.append(i)
            dfs(i)
            ans.pop()

n, m = map(int, input().split())
num = list(map(int, input().split()))
num.sort()
ans = []
dfs(0)