백준 N과 M (8) [c++]

15657 N과 M (8) 실버 3

문제 링크

문제

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

N개의 자연수 중에서 M개를 고른 수열 같은 수를 여러 번 골라도 된다. 고른 수열은 비내림차순이어야 한다. 길이가 K인 수열 A가 A1 ≤ A2 ≤ … ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

입력

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

둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

출력

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

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

풀이

다른 N과 M 문제들과 마찬가지로 백트래킹을 사용하며, queue와 vector를 이용한다. ( N과 M (1) )

차이점은

1. 중복이 가능하고
2. 비내림차순이어야 한다.

중복이 가능하게 하기 위해 중복을 피하려고 사용한 promising함수의 verified를 제거해준다.

비내림차순은 수열이 증가하거나 같아야 하므로 새로운 숫자를 vector에 추가할 때 새로 들어오는 값이 현재 백터의 마지막 값보다 크거나 같은지 확인하면 된다.

코드

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

queue <vector<int> > q;
vector <vector<int> > answer;
vector <int> ins;
int n, m;

void promising(vector<int> v)
{
    int size = v.size();

    if(size == m){
        answer.push_back(v);
        return;
    }

    for(int i = 0; i < n; i++)
    {
        vector<int> add(v);
        if(add.back() <= ins[i]) {
            add.push_back(ins[i]);
            q.push(add);
        }
    }
}

int main(){
    cin >> n >> m;

    for(int i = 1; i <= n; i++)
    {
        vector<int> v;
        int in;
        cin >> in;
        v.push_back(in);
        ins.push_back(in);
        q.push(v);
    }

    while(!q.empty()){
        int qsize = q.size();
        //cout << "q_size : " << qsize << endl;
        for(int i = 0; i < qsize; i++)
        {
            vector<int> t = q.front();
            q.pop();
            promising(t);

        }
    }

    sort(answer.begin(), answer.end());
    for(int i = 0; i < answer.size(); i++)
    {
        vector<int> out = answer[i];
        for(int j = 0; j < out.size(); j++)
        {
            printf("%d ", out[j]);
        }
        printf("\n");
    }
    
    return 0;
}

• ps
• back-tracking