알고리즘/알고리즘 문제풀이

[백준] 15654: N과 M(5)

mint* 2024. 1. 2. 21:02
728x90

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

 

15654번: N과 M (5)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열

www.acmicpc.net

 

오랜만에 쉬운 DFS 유형 문제를 풀었다.

dfs가 오랜만이다보니 헤맸는데, 역시 계속해서 다양한 문제 유형을 접하는 것이 중요한 것 같다.

 

코드

#include <bits/stdc++.h>
using namespace std;
int n, m, tmp;
int visited[10], num[10];
vector<int> arr;
void dfs(int here, int cnt){
    if (cnt==m){
        for(int i=0;i<m;i++) cout << arr[num[i]] << " ";
        cout << "\n";
        return;
    }
    for (int i=0;i<n;i++){
        if (!visited[i]){
            visited[i] = 1; 
            num[cnt] = i;
            dfs(i, cnt+1);
            visited[i]=0;
        }
    }
    
}


int main(){
    //1. 입력받기
    cin >> n >> m;
    for (int i=0;i<n;i++){
        cin >> tmp;
        arr.push_back(tmp);
    } 
    sort(arr.begin(), arr.end());
    //2. dfs 수행
    dfs(0, 0);
}

 

728x90