Algorithm
백준 14225번 부분수열의 합
이무쿤
2020. 9. 5. 10:51
반응형
문제
수열 S가 주어졌을 때, 수열 S의 부분 수열의 합으로 나올 수 없는 가장 작은 자연수를 구하는 프로그램을 작성하시오.
예를 들어, S = [5, 1, 2]인 경우에 1, 2, 3(=1+2), 5, 6(=1+5), 7(=2+5), 8(=1+2+5)을 만들 수 있다. 하지만, 4는 만들 수 없기 때문에 정답은 4이다.
입력
첫째 줄에 수열 S의 크기 N이 주어진다. (1 ≤ N ≤ 20)
둘째 줄에는 수열 S가 주어진다. S를 이루고있는 수는 100,000보다 작거나 같은 자연수이다.
출력
첫째 줄에 수열 S의 부분 수열의 합으로 나올 수 없는 가장 작은 자연수를 출력한다.
#include<iostream>
#include<algorithm>
#include<string.h>
#include<queue>
#include<stack>
#include<vector>
#include<map>
using namespace std;
int N;
vector<int> vec;
vector<int> res;
map<int, bool> m;
void DFS(int cur)
{
int sum = 0;
for (int i = 0; i < res.size(); i++)
{
sum += res[i];
}
m[sum] = true;
for (int i = cur + 1; i < N; i++)
{
res.push_back(vec[i]);
DFS(i);
res.pop_back();
}
}
int main()
{
cin >> N;
for (int i = 0; i < N; i++)
{
int a;
cin >> a;
vec.push_back(a);
}
for (int i = 0; i < N; i++)
{
res.push_back(vec[i]);
DFS(i);
res.pop_back();
}
int idx = 1;
while (1)
{
if (!m[idx])
{
cout << idx << endl;
return 0;
}
idx++;
}
}반응형