codefoces 632E

题目大意

给你n 种物品,每种物品有一个价值ai ,每一种物品可以取任意次,问恰好取k 次物品能取到的所有可能价值,从小到大输出价值。
n,ai,k1000

解题思路

可以容易写出一个dp,fi,j,k 表示已经做到第i 种物品,取了j 次,是否存在和为k 的方案,这个方法的时间复杂度是O(n4) .

我们可以消掉一维,首先,我让每个数都减去最小的数,这样做有什么好处呢,如果我选了x 个数和为sum ,那么我真正选的数就是sum+ka1 ,其中由于我选了x 个数,这些数原来都减了a1 ,我就加回去xa1 ,而没选的kx 个数,就相当于选第一个数。这样我就可以设fi,j 表示做到第i 个数,和为j 的最小取数次,这样时间是O(n3) .加几个剪枝可以过。

参考代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define maxn 1005
#define maxm (maxn*maxn)
using namespace std;

int f[maxm],a[maxn];

int n,fir,mx,k;

int main(){
    scanf("%d%d",&n,&k);
    fo(i,1,n) scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    n=unique(a+1,a+n+1)-(a+1);
    fir=a[1];
    fo(i,1,n) a[i]-=fir;
    mx=a[n]*k;
    memset(f,63,sizeof(f));
    f[0]=0;
    fo(j,2,n)
        fo(i,a[j],k*a[j])
            f[i]=min(f[i],f[i-a[j]]+1);
    fo(i,0,mx)
        if (f[i]<=k) printf("%d ",i+fir*k);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章