题意
给出 N N N个物品,现在有 M M M元,求买这 N N N个物品的方案总数使得每个方案都不能再买一个东西,答案模1e7+7。
思路
把
N
N
N个物品排序先,用
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示
i
+
1
∼
N
i+1\sim N
i+1∼N这些物品花费了
j
j
j元的方案数,可以转移过来:
f
[
i
]
[
j
]
+
=
f
[
i
+
1
]
[
j
]
+
f
[
i
+
1
]
[
j
−
x
[
i
]
]
f[i][j]+=f[i+1][j]+f[i+1][j-x[i]]
f[i][j]+=f[i+1][j]+f[i+1][j−x[i]]
然后我们枚举一个最小价值的物品,使得剩下的钱不能买它,那么就有
a
n
s
+
=
f
[
i
+
1
]
[
M
−
s
[
i
−
1
]
−
j
]
ans+=f[i+1][M-s[i-1]-j]
ans+=f[i+1][M−s[i−1]−j],其中
s
[
i
−
1
]
s[i-1]
s[i−1]代表前
i
−
1
i-1
i−1个物品的总价值,因为如果当前这个物品最小,那么前
i
−
1
i-1
i−1个都买了,
j
j
j代表剩下的不能买这个物品的钱。
代码
#include<cstdio>
#include<algorithm>
#define mod 10000007
using namespace std;
int N, M, ans;
int x[1101], s[1101], f[1101][1101];
int main()
{
scanf("%d %d", &N, &M);
for (int i = 1; i <= N; i++) scanf("%d", &x[i]);
sort(x + 1, x + 1 + N);
for (int i = 1; i <= N; i++) s[i] = s[i - 1] + x[i];
if (s[N] <= M) {
printf("1");
return 0;
}
f[N + 1][0] = 1;//初始化
for (int i = N; i >= 1; i--) {
for (int j = 0; j <= M; j++) {
if (j < x[i]) f[i][j] = f[i + 1][j];
else f[i][j] = (f[i + 1][j] + f[i + 1][j - x[i]]) % mod;
}
}
for (int i = 1; i <= N; i++)
for (int j = 0; j < x[i]; j++) {//不能买x[i]
if (M - j - s[i - 1] < 0) break;
ans = (ans + f[i + 1][M - j - s[i - 1]]) % mod;
}
printf("%d", ans);
}