[Python] 백준 15652, 15654번 - N과 M (4), (5)

N과 M (4)

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

15652번: N과 M (4)

 

n과 m (3)의 방식에서 고른 수열이 비내림차순이어야 한다는 조건만 고려해주면 된다.

 

s 변수를 사용하여 시작하는 지점부터 for 문을 돌려주었다.

 

import sys
input = sys.stdin.readline

def dfs(s): #s: start
    if len(ans) == m:
        print(*ans)
        return
    
    for i in range(s, n+1):
        ans.append(i)
        dfs(i)
        ans.pop()

n, m = map(int, input().split())
ans = []
dfs(1)

 

 

 

 

 

 

---

 

 

 

N과 M (5)

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

15654번: N과 M (5)

 

n과 m (1)과 같은 유형이다.

 

(1)~(4)는 1~n 까지의 자연수의 수열을 구하는 문제였고,

(5)는 n개의 자연수가 주어지면 직접 입력받아서 오름차순으로 정렬한 후 문제를 푸는 방식이다.

 

(아마 뒷 문제들은 다 이런 방식으로 한 번 더 같은 유형을 연습하는 게 아닐까 싶다).

 

import sys
input = sys.stdin.readline

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

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

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