[백준] 15663번 N과 M c++ 풀이

---

알고리즘 분류

• 백트래킹

---

문제 설명

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

• N개의 자연수 중에서 M개를 고른 수열

---

입력 설명

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

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

---

출력 설명

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

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

---

예제 입력

3 1
4 4 2

4 2
9 7 9 1

4 4
1 1 1 1

---

예제 출력

2
4

1 7
1 9
7 1
7 9
9 1
9 7
9 9

1 1 1 1

---

아이디어

백트래킹을 이용해서 푸는데 중복되는 수열이 나오면 안 되기 때문에 map에 지금까지 출력했던 모든 수열을 전부 기록해서 map에 기록되지 않은 수열만 출력한다. 또한 string의 한 글자씩 저장 시 12 34 이런 숫자가 1 2 3 4 이렇게 출력되므로 숫자+" "를 함께 저장해서 여러 자리의 숫자라도 문제없이 저장하도록 구현한다.

---

알고리즘

vec에 숫자를 입력받는다.

vec을 오름차순으로 정렬한다.

vec에 들어있는 숫자를 하나씩  ans에 push_back()하고 재귀함수로 들어간다.

이후 함수를 빠져 나오면 ans에서 pop_back()해준다.

만약 ans의 길이가 m과 같아지면 map에 현재 ans와 같은 수열을 출력한 기록이 있는지 찾는다.

없다면 map에 ans를 기록하고 ans를 출력한다.

---

코드

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;
int n,m;
vector<pair<int,bool>> vec;
vector<int> ans;
map<string, bool> mp;
void prt()
{
	if(ans.size()==m)
	{
	    string s="";
		for(auto &b : ans)
		    s+=to_string(b)+" ";
		if(mp.find(s) ==mp.end())
		{
		    mp.insert({s,true});
		    for(int i=0;i<s.size();i++)
		        cout<<s[i];
		    cout<<"\n";
		}
		return;
	}
	for(auto &a : vec)
	{   
	    if(a.second)
	    {
	        ans.push_back(a.first);
	        a.second =false;
		    prt();
		    a.second =true;
		    ans.pop_back();
	    }
	}
}
int main()
{
	cin>>n>>m;
	for(int i=0; i<n; i++)
	{
	    int temp;
	    cin>>temp;
	    vec.push_back({temp,true});
	}
	sort(vec.begin(), vec.end());
	prt();
	return 0;
}

---

후기

문제 링크 -> https://www.acmicpc.net/problem/15663

 

N과 M문제를 많이 풀다 보니까 꽤나 익숙해졌는데 이번 문제에서는 vector, map을 모두 사용했다.

map을 오랜만에 사용해서 사용법이 조금 헷갈렸지만 검색해서 알아내서 풀었다.

---

본 블로그의 모든 글은 개인적인 학습 내용이므로 다양한 오류가 있을 수 있습니다.

오류를 발견하신다면 해당 내용 댓글로 알려주시면 감사하겠습니다!