小K的雕塑 01背包变形

链接：https://ac.nowcoder.com/acm/contest/696/D
来源：牛客网
 

题目描述

小K有n个雕塑，每个雕塑上有一个整数
若集合T中的每一个元素在n个雕塑上都能找得到，则称这个集合为一个优秀的集合

小K想知道所有大小<=k优秀的集合的价值和是多少

一个优秀的集合的价值可以用F(T)=∏x  , x∈T来表示

特别的F(∅)=1, |∅|=0

即求  

输入描述:

两个整数n,k，分别表示雕塑的数量与集合大小的上限 
接下来n个整数表示第i个雕塑上的数字ai是多少

输出描述:

一行表示所有大小<=k优秀集合的权值和

示例1

输入

3 0
1 2 3

输出

1

示例2

输入

3 1
1 2 3

输出

7

示例3

输入

3 2
1 2 3

输出

18

题解

首先，一个集合里的元素肯定没有重复值。那么先将数组去重，设去重后的数组长度为n。

那么k=min（n，k）;

然后每个数组逐个考虑 ：

用dp[i][j]表示前i个数取j个数的f(T)值。

每个数要么取，要么不取，思路和01背包的思路一样。

那么状态转移方程为：dp[i][j]=dp[i-1][j]+dp[i-1][j-1]*a[i]

dp[i-1][j] :表示不要第i个数, dp[i-1][j-1]*a[i]表示要第i个数.

可以按照01背包的思路，用滚动数组将二维数组降成一维数组。

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
typedef long long ll;
const ll mod=1e9+7;
ll a[maxn];
ll dp[maxn];
int main() {
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        scanf("%lld",a+i);
    }
    sort(a+1,a+1+n);
    n=unique(a+1,a+1+n)-a-1;
    k=min(n,k);
    dp[0]=1;
    for(int i=1;i<=n;i++){
        for(int j=k;j>=1;j--){
            dp[j]=(dp[j]+dp[j-1]*a[i])%mod;
        }
    }
    ll ans=0;
    for(int i=0;i<=k;i++){
        ans=(ans+dp[i])%mod;
    }
    cout<<ans<<endl;
    return 0;
}