알고리즘
백준 1182 부분수열의 합
중국고대사
2019. 4. 3. 18:38
sum은 인자로 전달해준다.
중간에 원소 몇 개로 합을 완성할 수도 있지만, 아래 코드에서 sum과 s가 같더라도 이후의 원소가 들어오는 것에 따라서 다시 sum과 s가 같아질 수 있다.
따라서 경우의 수를 (s==sum && i == n)과 (s != sum && i == n) 두 가지로 나누어서 풀어야 한다.
또한 합이 0이 입력되면, 아무것도 선택하지 않아도 0이 되므로, 공집합은 허용되지 않는다. 따라서 답을 출력할 때 1을 빼준다.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | #include <iostream> #include <algorithm> #include <vector> using namespace std; int n, s; vector<int> v(20); int ans; void go(int i, int sum) { if (s == sum && i == n) { ans++; return; } if (s != sum && i == n) return ; go(i + 1, sum); go(i + 1, sum + v[i]); } int main(int argc, char* argv[]) { cin >> n >> s; for (int i = 0; i < n; i++) { cin >> v[i]; } go(0, 0); if (s == 0)ans--; cout << ans << '\n'; return 0; } | cs |