[boj][c++] N과 M (9) (15663)

문제
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
- N개의 자연수 중에서 M개를 고른 수열
입력
첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.
출력
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다.
중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

풀이 방법
수열을 모두 구하는 것이므로 -> 백트래킹
중복 선택이 아니므로 -> visited 배열을 이용해 확인
중복되는 수열은 한 번만 -> M과 N이 최대 8이므로 수열의 최대 갯수가 많지 않음 -> 전부 저장 후 중복 제거

 

#include <bits/stdc++.h>
using namespace std;

int N, M;
vector<int> v;
bool visited[9]={0};
int arr[9] = {0};
vector<vector<int>> route;

void pick(){
    if(v.size() == M){ // M만큼 방문했을 경우 루트에 추가
        route.push_back(v);
        return;
    }
    for(int i=0; i<N; i++){
        if(!visited[i]){ // 중복 방문 제거
            visited[i] = 1;
            v.push_back(arr[i]);
            pick();
            v.pop_back();
            visited[i] = 0;
        }
    }
}

int main(){  
    cin >> N >> M;
    for(int i=0; i<N; i++) cin >> arr[i];
    
    // 사전 증가 순이므로 정렬
    sort(arr, arr + N);
    pick();
    
    // 중복 경로 제거
    sort(route.begin(), route.end());
    route.erase(unique(route.begin(), route.end()), route.end());
    
    // 경로 출력
    for(auto tmp : route){
        for(int i=0; i<tmp.size(); i++){
            cout << tmp[i] << ' ';
        }
        cout << '\n';
    }
}

기억하고 넘어갈 점
1. vector에서 중복 삭제하기
- unique(v.begin(), v.end()) 는 중복을 완전히 제거하고 반환하는 것이 아니라 1,2,3,4,2,2,2 와 같이 중복된 값들을 뒤로 보내고, 중복된 값이 시작하는 지점을 포인터로 반환한다.
- 따라서 v.erase(unique(v.begin(), v.end()), v.end()) 로 중복된 값이 시작하는 지점부터 맨 끝까지 삭제해준다.