URAL 1244. Gentlemen

滚动数组+动态规划

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;
}