nyoj ACM:zb的生日(DFS 递归 回溯)

来源:互联网 发布:java getclasspath 编辑:程序博客网 时间:2024/06/16 04:34

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

思路:
和前几次一样,算法用回溯算法,方式是DFS的递归,这里为了剪枝,提前将瓜的重量W[]进行了排序,当当前重量大于总重量的一半,那么久不用再递归了。

//你永远不知道你当前的差值是不是最小,所以你要遍历所有的情况 #include<iostream>#include<cstdio>#include<cstring>using namespace std;int N;int W[30];bool visited[30];int ans;//ans用于记录当前的最小值 int sum=0;int SUM=0;void kuaipai(int l,int r){    int i=l,j=l;    int x=W[r];    int y=0;    if(l>=r)        return;    for(int i=l;i<=N;i++)    {        if(W[i]<x)        {            y=W[i];            W[i]=W[j];            W[j]=y;            j++;        }    }    y=W[r];    W[r]=W[j];    W[j]=y;    kuaipai(l,j-1);    kuaipai(j+1,r);}void DFS(int i){    if(i>N)        return;        for(int j=i;j<=N;j++)    {        if(sum+W[j]>SUM/2) //剪枝            return;        if(visited[j]==false)        {            sum+=W[j];            if(ans>(SUM-sum)-sum)                ans=(SUM-sum)-sum;            visited[j]=true;            DFS(j+1);            sum-=W[j];            visited[j]=false;        }    }}int main(){    while(scanf("%d",&N)!=EOF)    {        memset(W,0,sizeof(W));        memset(visited,false,sizeof(visited));        ans=0x3f3f3f3f;        sum=0;        SUM=0;        for(int i=1;i<=N;i++)        {            scanf("%d",W+i);            SUM+=W[i];        }        kuaipai(1,N);        if(N==1)            ans=W[1];        else            DFS(1);         printf("%d\n",ans);    }    return 0;}
0 0