题目大意: 从 n n n 种物品中拿 k k k 个,每种物品有一个价值且数量无限,求出总价值的所有情况。
题解
把所有物品放在一起造一个多项式 f f f, f i = 1 f_i=1 fi=1 表示存在物品的价值为 i i i。
然后 f k f^k fk 就是答案,看一下哪一位不为 0 0 0 就输出。
而且似乎需要双模数 N T T NTT NTT,即用两个不同的模数来做 N T T NTT NTT,然后只要某一位在两个答案多项式中的任意一个里不为 0 0 0 就输出。
由于这题答案多项式的最高次只有 1 0 6 10^6 106,所以不需要多项式快速幂,直接搞即可,具体看代码吧:
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
#define maxn 3000010
#define bin(x) (1<<(x))
int n,k,mod1=998244353,mod2=1004535809;
int f[maxn],g[maxn];
int ksm(int x,int y,int mod){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
int limit,r[maxn];void ntt(int *f,int lg,int mod,int type=0)
{
limit=bin(lg);if(type)reverse(f+1,f+limit);
for(int i=1;i<limit;i++)if(i<r[i])swap(f[r[i]],f[i]);
for(int mid=1;mid<limit;mid<<=1)
{
int wn=ksm(3,(mod-1)/mid/2,mod);
for(int j=0;j<limit;j+=(mid<<1))
for(int i=0,w=1;i<mid;i++,w=1ll*w*wn%mod)
{
int x=f[j+i],y=1ll*f[j+i+mid]*w%mod;
f[j+i]=(x+y)%mod;f[j+i+mid]=(x-y+mod)%mod;
}
}
if(type)for(int i=0,in=ksm(limit,mod-2,mod);i<limit;i++)f[i]=1ll*f[i]*in%mod;
}
int main()
{
scanf("%d %d",&n,&k);int N=0,lg;
for(int i=1,x;i<=n;i++)scanf("%d",&x),f[x]=g[x]=1,N=max(N,x);
N*=k,lg=ceil(log2(N));for(int i=0;i<bin(lg);i++)r[i]=(r[i>>1]>>1)|((i&1)<<(lg-1));
ntt(f,lg,mod1);for(int i=0;i<bin(lg);i++)f[i]=ksm(f[i],k,mod1);ntt(f,lg,mod1,1);
ntt(g,lg,mod2);for(int i=0;i<bin(lg);i++)g[i]=ksm(g[i],k,mod2);ntt(g,lg,mod2,1);
for(int i=0;i<=N;i++)if(f[i]||g[i])printf("%d ",i);
}
这题还有另外一个做法,是 n 3 n^3 n3 的dp。(但是因为常数奇小所以跑的甚至比上面快很多)
先让所有物品减去价值最小的物品的价值 p p p,这样就弄出了一个价值为 0 0 0 的物品。
然后设 f [ i ] f[i] f[i] 表示组成价值 i i i 至少需要用几个物品,跑一发 n 2 n^2 n2 的dp,最后看一下,假如 f [ i ] ≤ k f[i]\leq k f[i]≤k,那么就输出 k p + i kp+i kp+i。
因为只要满足
f
[
i
]
≤
k
f[i]\leq k
f[i]≤k,那么剩下的
k
−
f
[
i
]
k-f[i]
k−f[i] 个物品全部放价值
0
0
0 的物品,就能做到用k个物品凑成价值kp+i
。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 1010
int n,k,a[maxn],f[maxn*maxn];
int main()
{
scanf("%d %d",&n,&k);int mi=99999999,ma=0;
for(int i=1;i<=n;i++)scanf("%d",&a[i]),mi=min(mi,a[i]),ma=max(ma,a[i]);
memset(f,63,sizeof(f));f[0]=0;
for(int i=1;i<=n;i++)
{
a[i]-=mi;
for(int j=0;j+a[i]<=ma*k;j++)
f[j+a[i]]=min(f[j]+1,f[j+a[i]]);
}
for(int i=0;i<=ma*k;i++)if(f[i]<=k)printf("%d ",i+k*mi);
}