poj1014——Dividing

多重背包问题。                 

转化为01背包问题。不过 需优化，否则会TLE。

优化部分程序中标出。

#include #include #include using namespace std; int f[60005]; int a[7]; int cnt; int b[60005],c[60005]; int main() { int i,sum,ca=0; int v; while(1) { sum=0; for(i=1;i<=6;i++) { cin>>a[i]; sum+=a[i]; } if(sum==0) break; printf("Collection #%d:/n",++ca); sum=0; for(i=1;i<=6;i++) sum+=i*a[i]; if(sum%2!=0) { printf("Can't be divided./n"); } else { cnt=0; for(i=1;i<=6;i++)//优化部分 { int k=1; while(a[i]-k+1>0) { b[++cnt]=i*k; c[cnt]=i*k; a[i]-=k; k=2*k; } if(a[i]>0) { b[++cnt]=i*a[i]; c[cnt]=i*a[i]; } } v=sum/2; memset(f,-1,sizeof(f)); f[0]=0; int j; for(i=1;i<=cnt;i++) for(j=v;j>=0;j--) { if(j-b[i]>=0) { f[j]=f[j-b[i]]+c[i]>f[j]?f[j-b[i]]+c[i]:f[j]; } else f[j]=f[j]; } if(f[v]==v) printf("Can be divided./n"); else printf("Can't be divided./n"); } printf("/n"); } return 0; }