본문 바로가기
백준/백트래킹

[백준🥈3] #15649 N과 M(1) (Python)

by 똥먹는낙타 2023. 2. 9.
728x90
반응형

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

 

15649번: N과 M (1)

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

www.acmicpc.net

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

  • 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

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

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

예제 입력 1

3 1

예제 출력 1

1
2
3

예제 입력 2

4 2

예제 출력 2

1 2
1 3
1 4
2 1
2 3
2 4
3 1
3 2
3 4
4 1
4 2
4 3

예제 입력 3

4 4

예제 출력 3

1 2 3 4
1 2 4 3
1 3 2 4
1 3 4 2
1 4 2 3
1 4 3 2
2 1 3 4
2 1 4 3
2 3 1 4
2 3 4 1
2 4 1 3
2 4 3 1
3 1 2 4
3 1 4 2
3 2 1 4
3 2 4 1
3 4 1 2
3 4 2 1
4 1 2 3
4 1 3 2
4 2 1 3
4 2 3 1
4 3 1 2
4 3 2 1

✔️ Code 1 (완전탐색 - permutations)

import sys
from itertools import permutations
input = sys.stdin.readline

n, m = map(int, input().split())
nums = [i for i in range(1, n+1)]

per = list(permutations(nums, m))
for p in per:
    print(*p)

✔️ Code 2 (백트래킹)

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
visited = [False] * n
result = []

def backtracking(k, n, m):
    if k == m:
        print(*result)
        return
    
    for i in range(len(visited)):
        if not visited[i]:
            visited[i] = True
            result.append(i+1)
            backtracking(k+1, n, m)
            visited[i] = False
            result.pop()

backtracking(0, n, m)

✏️ Comment

완전탐색으로 먼저 풀었는데, 백트래킹 대표 문제이기에 ... 백트래킹 개념 좀 찾아보고 풀이 보면서 로직을 이해했다. N과 M 시리즈 다 풀어야지 ~~ ! N값이 작아서 그런가 이 문제는 완탐이 시간이 더 적게 걸렸당 

728x90
반응형

댓글