백준 코딩

백준 1182번: 부분수열의 합

까르르꿍꿍 2022. 2. 6. 17:04

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

 

1182번: 부분수열의 합

첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.

www.acmicpc.net

 

이 문제는 처음에 순열을 이용해서 일일이 위치값 변경하면서 더했는데 이럴경우 시간복잡도가 (n*n!)가 나와 시간초과가 나온다

그 다음으로 재귀함수를 구현해서 모든 부분집합을 구햇다. 시간복잡도는 (2^n-1)이 나오고 시간초과는 나오지 않았다.

함수에는 부분집합의 합과 배열 몇 번째인지 를 인자로 받고  

몇번쨰 배열 인자를 더하는 경우와 안더하는 경우를 나누어서 재귀를 했다.

탈출조건은 배열인지가 끝에 도달하면 탈출하게 했다.

코드를 보자

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int n,s;
int ans;
void check(vector<int>& v,int sum,int i){
    if(i==n){                            //배열의 끝을 도달했을 떄
        if(sum==s){                      //s와 합이 같으면 1증가
            ans++;
        }
        return;
    }
    check(v,sum+v[i],i+1);                //i번쨰 값을 더할때
    check(v,sum,i+1);                      //i번쨰 값을 안 더할때
}
int main(){
    cin>>n>>s;
    vector<int> v(n);
    for(int i=0;i<n;i++){
        cin>>v[i];
    }
    check(v,0,0);
    if(s==0){                        //s가 0이면 공집합은 빼야하므로 -1해준다
        cout<<ans-1;
    }
    else{
        cout<<ans;
    }
}