백준 17610번 양팔저울 C++

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

17610번: 양팔저울

 

저울에 추를 올리는 경우 :

1. 추를 올리지 않는 경우

2. 왼쪽에 올리는 경우

3. 오른쪽으로 올리는 경우

 

추가 최대 13개이기때문에 dfs로 가능한 문제이다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int k;
int maxSum = 0;
vector<int> v;

set<int> s;

void Dfs(int sum, int idx) {
    if (idx == k) {
        if (abs(sum) > maxSum) return;
        if (sum == 0) return;
        s.insert(abs(sum));
        return;
    }
    Dfs(sum, idx + 1);
    Dfs(sum + v[idx], idx + 1);
    Dfs(sum - v[idx], idx + 1);
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> k;
    v = vector<int>(k);
    for (int i = 0; i < k; i++) {
        cin >> v[i];
        maxSum += v[i];
    }
    Dfs(0, 0);
    cout << maxSum - s.size();
}