UVA 1099 Sharing Chocolate(记忆化搜索)

来源：互联网发布：美工做好图怎么上传编辑：程序博客网时间：2024/06/07 12:49

如果x*y!=sum[ALL]就直接剪枝。

否则就进行dp，状态定义是dp(S,r),r=min(x,y),S为题目中给出的巧克力块数的集合，然后就一直记忆化搜索下去，如果搜索成功则返回1，否则返回0，首先枚举子集，然后判断该子集能不能继续搜索下去，判断条件是切一刀之后有一条边是不会变的，如果判断有一条边，没有变，那么就说明切了一刀可以把当前巧克力分成两块了，那么就可以继续搜索了，当然切了一刀后的两块巧克力都要搜索。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int maxn=16;int A[maxn],sum[1<<maxn],f[1<<maxn][150],vis[1<<maxn][150];int n;int bitcount(int x) {return x==0?0:bitcount(x/2)+(x&1);}int dp(int S,int x){    if(vis[S][x]) return f[S][x];    vis[S][x]=1;    int &ans=f[S][x];    if(bitcount(S)==1) return ans=1;    int y=sum[S]/x;    for(int S0=(S-1)&S;S0;S0=(S0-1)&S)    {        int S1=S-S0;        if(sum[S0]%x==0&&dp(S0,min(x,sum[S0]/x))&&dp(S1,min(x,sum[S1]/x))) return ans=1;        if(sum[S0]%y==0&&dp(S0,min(y,sum[S0]/y))&&dp(S1,min(y,sum[S1]/y))) return ans=1;    }    return ans=0;}int main(){    int kase=0;    while(scanf("%d",&n)&&n)    {        int x,y;        scanf("%d%d",&x,&y);        for(int i=0;i<n;i++) scanf("%d",&A[i]);        memset(sum,0,sizeof(sum));        for(int s=0;s<(1<<n);s++){            for(int j=0;j<n;j++)                if(s&(1<<j)) sum[s]+=A[j];        }        int ALL=(1<<n)-1;        int ans;        memset(vis,0,sizeof(vis));        if(sum[ALL]!=x*y) ans=0;        else ans=dp(ALL,min(x,y));        printf("Case %d: %s\n",++kase,(ans==1)?"Yes":"No");    }    return 0;}

0 0