URAL 1244. Gentlemen

来源：互联网发布：ps4最终幻想10淘宝编辑：程序博客网时间：2024/05/22 14:20

滚动数组+动态规划

dp[ i ] [ j ] 的值，表示只用前 i 张卡，和为 j 有多少种。

另外，写了个类来保存DP的路径。

#define FOR(i,n) for(long long (i)=1;(i)<=(n);(i)++)#define For(i,n) for(long long (i)=0;(i)<(n);(i)++)using namespace std;struct Mark{//记录路径 unsigned long long M[2];Mark(){M[0]=M[1]=0;}void init(){M[0]=M[1]=0;}void set(int k){M[k/64]|=1ll<<(k%64);}bool is(int k){return ((M[k/64]>>(k%64))&1ll);}};int dp[2][100001]; Mark path[2][100001];int a[100];int Part,Sum,N;int main(void){while(cin>>Part>>N){Sum=0;For(i,N){scanf("%d",&a[i]);Sum+=a[i];}memset(dp,0,sizeof(dp));For(i,2)For(j,100001)path[i][j].init();dp[0][a[0]]=1;dp[0][0]=1;path[0][a[0]].set(0);for(int i=1;i<N;i++){for(int j=0;j<=100000;j++){dp[i&1][j]=0;if(dp[(i+1)&1][j]){dp[i&1][j]+=dp[(i+1)&1][j];path[i&1][j]=path[(i+1)&1][j];}if(j-a[i]>=0&&dp[(i+1)&1][j-a[i]]){dp[i&1][j]+=dp[(i+1)&1][j-a[i]];path[i&1][j]=path[(i+1)&1][j-a[i]];path[i&1][j].set(i);}}}int ANS=dp[(N+1)&1][Sum-Part];int first=0;switch(ANS){case 0:printf("0\n");break;case 1:for(int i=0;i<N;i++){if(path[(N+1)&1][Sum-Part].is(i)) {if(first) printf(" ");printf("%d",i+1);first++;}}cout<<endl;break;default:printf("-1\n");break;}}return 0;}

0 0