題干：

鏈接：https://ac.nowcoder.com/acm/contest/696/D
來源：牛客網(wǎng)
?

小K有n個(gè)雕塑，每個(gè)雕塑上有一個(gè)整數(shù)
若集合T中的每一個(gè)元素在n個(gè)雕塑上都能找得到，則稱這個(gè)集合為一個(gè)優(yōu)秀的集合

小K想知道所有大小<=k優(yōu)秀的集合的價(jià)值和是多少

一個(gè)優(yōu)秀的集合的價(jià)值可以用來表示

輸入描述:

兩個(gè)整數(shù)n,k，分別表示雕塑的數(shù)量與集合大小的上限? 接下來n個(gè)整數(shù)表示第i個(gè)雕塑上的數(shù)字aiai是多少

輸出描述:

一行表示所有大小<=k優(yōu)秀集合的權(quán)值和

示例1

輸入

復(fù)制

3 0 1 2 3

輸出

復(fù)制

1

示例2

輸入

復(fù)制

3 1 1 2 3

輸出

復(fù)制

7

示例3

輸入

復(fù)制

3 2 1 2 3

輸出

復(fù)制

18

示例4

輸入

復(fù)制

3 3 1 2 3

輸出

復(fù)制

24

示例5

輸入

復(fù)制

8 1 1 6 35 45 65 3 56 8

輸出

復(fù)制

220

備注:

有10%10%的數(shù)據(jù)滿足k=1

有30%30%的數(shù)據(jù)滿足k=n

以上兩個(gè)部分包含在100%100%的數(shù)據(jù)中，但不包含在前30%30%的數(shù)據(jù)中

對(duì)于前30%30%的數(shù)據(jù)，滿足k≤n≤5k≤n≤5

對(duì)于100%100%的數(shù)據(jù)，滿足n≤106,1≤ai≤5×103n≤106,1≤ai≤5×103

由于答案可能過大請(qǐng)對(duì)109+7取模后輸出

解題報(bào)告：

? 題意好難懂啊、、、

?注意到集合的性質(zhì)：互異性。

根據(jù)鴿巢原理，很容易知道最多有m=5e3個(gè)數(shù)（去重之后）

然后設(shè)F[n][k]表示前n個(gè)數(shù)，取k個(gè)數(shù)的價(jià)值。（且已知這k個(gè)數(shù)一定互不相同。因?yàn)榧系幕ギ愋?#xff09;

????F[n][k]=F[n-1][k]+F[n-1][k-1]*a[i]

效率O(m^2)。?

當(dāng)然這題也可以用01背包的方法將空間復(fù)雜度優(yōu)化到一維。

AC代碼：

#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<set> #include<string> #include<cmath> #include<cstring> #define F first #define S second #define ll long long #define pb push_back #define pm make_pair using namespace std; typedef pair<int,int> PII; const int MAX = 1e6 + 5; const ll mod = 1e9+7; ll dp[5005][5005]; int a[MAX],b[MAX]; int n,k; int main() {cin>>n>>k;for(int i = 1; i<=n; i++) cin>>a[i],b[i]=a[i];sort(b+1,b+n+1);int N = unique(b+1,b+n+1) - b - 1;dp[0][0]=1;for(int i = 1; i<=N; i++) {dp[i][0] = 1;for(int j = 1; j<=k; j++) {dp[i][j] = (dp[i-1][j] + dp[i-1][j-1] * b[i])%mod;}}ll ans = 0;for(int i = 0; i<=N && i<=k; i++) ans = (ans + dp[N][i])%mod;cout << ans << endl;return 0 ; }

?

總結(jié)

以上是生活随笔為你收集整理的【牛客 - 696D】小K的雕塑（dp，鸽巢原理，01背包类问题）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。