알고리즘/백준

[백준/c++] 14225번: 부분수열의 합

녕이 2022. 7. 29. 20:56
728x90

 

https://www.acmicpc.net/problem/14225

 

14225번: 부분수열의 합

수열 S가 주어졌을 때, 수열 S의 부분 수열의 합으로 나올 수 없는 가장 작은 자연수를 구하는 프로그램을 작성하시오. 예를 들어, S = [5, 1, 2]인 경우에 1, 2, 3(=1+2), 5, 6(=1+5), 7(=2+5), 8(=1+2+5)을 만들

www.acmicpc.net

 

 

우선 수열을 정렬하고, 1개부터 수열의 개수만큼 백트래킹으로 부분 수열을 선택한다.

각 부분수열들의 합을 set에 넣어주고 자연수 1부터 set에 있는지 없는지 체크하고 없다면 바로 출력하고 끝낸다.

set은 자동 정렬, 자동 중복 제거를 하기 때문에 사용하기 좋다.

 

 

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <set>
using namespace std;
int n, S[21], sum = 0;
vector<int> v;
set<int> num;
bool check[21];
int m = 1, ans = 0;

void BT(int cnt, int s, int k){
    if(cnt == k){
        sum = 0;
        for(auto i : v) sum += i;
        num.insert(sum);
        return;
    }
    
    for(int i=s; i<n; i++){
        if(!check[i]){
            check[i] = true;
            v.push_back(S[i]);
            BT(cnt+1,i,k);
            v.pop_back();
            check[i] = false;
        }
    }
}

int main(){
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> n;
    for(int i=0; i<n; i++) cin >> S[i];
    sort(S, S+n);
    
    for(int i=1; i<=n; i++){ //부분수열 N=3 -> 1, 2, 3
        BT(0,0,i);
    }
    
    int ans = 1;
    while(1){
        if(num.find(ans) == num.end()){
            cout << ans << '\n';
            return 0;
        }else{
            ans++;
        }
    }
    return 0;
}

 

 

 

 

728x90