codeforces 632E Thief in a Shop 题解

题目传送门

题目大意： 从 $n$ 种物品中拿 $k$ 个，每种物品有一个价值且数量无限，求出总价值的所有情况。

题解

把所有物品放在一起造一个多项式 $f$，$f_i=1$ 表示存在物品的价值为 $i$。

然后 $f^k$ 就是答案，看一下哪一位不为 $0$ 就输出。

而且似乎需要双模数 $NTT$，即用两个不同的模数来做 $NTT$，然后只要某一位在两个答案多项式中的任意一个里不为 $0$ 就输出。

由于这题答案多项式的最高次只有 $10^6$，所以不需要多项式快速幂，直接搞即可，具体看代码吧：

#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$ 的dp。（但是因为常数奇小所以跑的甚至比上面快很多）

先让所有物品减去价值最小的物品的价值 $p$，这样就弄出了一个价值为 $0$ 的物品。

然后设 $f[i]$ 表示组成价值 $i$ 至少需要用几个物品，跑一发 $n^2$ 的dp，最后看一下，假如 $f[i]\le k$，那么就输出 $kp+i$。

因为只要满足 $f[i]\le k$，那么剩下的 $k-f[i]$ 个物品全部放价值 $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);
}