BOJ 15651 - N과 M (3)

문제

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

 

 

풀이

다음 순서로 문제를 해결한다 

1. 만들 수 있는 모든 수열을 구해야 하므로, 완전 탐색을 이용한다

2. 수열의 길이는 m이고 중복되지 않게 오름차순으로 출력한다

 

재귀함수를 이용해 백트래킹 알고리즘으로 문제를 해결한다

 

 

코드

python

def backtracking(depth):
    if depth == m:
        print(*answer)
        return
    for i in range(1, n + 1):
        answer.append(i)
        backtracking(depth + 1)
        answer.pop()
 
 
answer = []
 
n, m = map(int, input().split())
 
backtracking(0)
 

 

java

import java.util.Scanner;
 
public class Main {
    
    private static int n, m;
    private static int[] a;
    private static StringBuilder sb = new StringBuilder();
 
    public static void main(String[] args) {
        input();
        solve();
    }
 
    private static void solve() {
        rec(1);
        System.out.println(sb.toString());
    }
 
    private static void rec(int count) {
        if (count == m + 1) {
            for (int i = 1; i<=m; i++) {
                sb.append(a[i]).append(' ');
            }
            sb.append('\n');
        } else {
            for (int i = 1; i<=n; i++) {
                a[count] = i;
                rec(count + 1);
                a[count] = 0;
            }
        }
        
    }
 
    private static void input() {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        a = new int[m + 1];
        sc.close();
    }
 
}
 

 

C++

#include <bits/stdc++.h> 
using namespace std; 
 
int arr[9];
bool visit[9];
int N, M;
 
void solve(int cnt)
{
    if (cnt == M)
    {
        for (int i=0; i<M; i++)
            cout << arr[i] << ' ';
        cout << '\n';
        return;
    }
 
    for (int i=1; i<=N; i++)
    {
        arr[cnt] = i;
        solve(cnt + 1);
    }
}
 
int main() 
{ 
    ios_base::sync_with_stdio(false); 
    cin.tie(NULL); 
 
    cin >> N >> M;
 
    solve(0);
    
    return 0; 
}