ACM zb的生日

OpenSoucre發表於2014-05-01

zb的生日

時間限制:3000 ms  |  記憶體限制:65535 KB
難度:2
 
描述
今天是陰曆七月初五,acm隊員zb的生日。zb正在和C小加、never在武漢集訓。他想給這兩位兄弟買點什麼慶祝生日,經過調查,zb發現C小加和never都很喜歡吃西瓜,而且一吃就是一堆的那種,zb立刻下定決心買了一堆西瓜。當他準備把西瓜送給C小加和never的時候,遇到了一個難題,never和C小加不在一塊住,只能把西瓜分成兩堆給他們,為了對每個人都公平,他想讓兩堆的重量之差最小。每個西瓜的重量已知,你能幫幫他麼?
 
輸入
多組測試資料(<=1500)。資料以EOF結尾
第一行輸入西瓜數量N (1 ≤ N ≤ 20)
第二行有N個數,W1, …, Wn (1 ≤ Wi ≤ 10000)分別代表每個西瓜的重量
輸出
輸出分成兩堆後的質量差
樣例輸入
5
5 8 13 27 14
樣例輸出
3
本題的用暴力搜尋,想好解答樹,一邊搜尋一邊剪枝
#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <numeric>
using namespace std;

int ans = 0;
int sumWeight = 0;

void dfs(int index,int v, vector<int>& weight){
    if (index == weight.size()) return ;
    ans = min(ans,abs(sumWeight-2*v));
    dfs(index+1,v,weight);
    dfs(index+1,v+weight[index],weight);
}

int main(){
    int n;
    while(cin >> n){
        vector<int> weight(n);
        for(int i = 0; i < n ; ++ i) cin >> weight[i];
        sumWeight = accumulate(weight.begin(),weight.end(),0);
        ans = sumWeight;
        dfs(0,0,weight);
        cout<<ans<<endl;
    }
}