题目链接:点击打开链接
思路:
用d[i][j][k]表示, 前i个数, 选取的数之和为j, 再次选取的数之和为k的种类数。 那么当这种情况存在时, 我们把此时的k保存即可。
细节参见代码:
#include <cstdio> #include <cstring> #include <algorithm> #include <iostream> #include <string> #include <vector> #include <stack> #include <bitset> #include <cstdlib> #include <cmath> #include <set> #include <list> #include <deque> #include <map> #include <queue> #define Max(a,b) ((a)>(b)?(a):(b)) #define Min(a,b) ((a)<(b)?(a):(b)) using namespace std; typedef long long ll; typedef long double ld; const ld eps = 1e-9, PI = 3.1415926535897932384626433832795; const int mod = 1000000000 + 7; const int INF = 0x3f3f3f3f; // & 0x7FFFFFFF const int seed = 131; const ll INF64 = ll(1e18); const int maxn = 500 + 10; int T,n,K,cnt[maxn*maxn],d[2][maxn][maxn],kase = 0, a[maxn]; vector<int> res; int main() { scanf("%d%d", &n, &K); int sum = 0; for(int i = 1; i <= n; i++) { scanf("%d", &a[i]); sum += a[i]; } int u = 1; d[u][0][0] = 1; if(sum > K) sum = K; for(int i = 1; i <= n; i++) { for(int j = 0; j <= sum; j++) { for(int k = 0; k <= sum; k++) { d[u^1][j][k] += d[u][j][k]; if(j >= a[i] && k >= a[i]) d[u^1][j][k] += d[u][j-a[i]][k-a[i]]; if(j >= a[i]) d[u^1][j][k] += d[u][j-a[i]][k]; if(i == n && d[u^1][j][k]) { if(j == K) { if(!cnt[k]) { res.push_back(k); cnt[k] = 1; } } } } } u ^= 1; } sort(res.begin(), res.end()); int len = res.size(); printf("%d\n", len); for(int i = 0; i < len; i++) { printf("%d%c", res[i], i == len-1 ? '\n' : ' '); } return 0; }