题目大意:
给定(n)个数(a[1]sim a[n]),让你把它分为若干个集合,使每个集合内最大值与最小值的差的总和不超过(K)。问总方案数。
解题思路:
一道很神的dp题。
首先将数进行排序,然后将这些数扔数轴上,则集合价值相当于在数轴上覆盖这些点所用的最短线段的长度(当然长度可以为0)。
考虑dp,设(f[i][j][k])表示考虑了前(i)个点,目前还未确定右端点的集合还有(j)条,目前的总价值为(k),则
不论如何,新加进(a[i+1]-a[i])这条线段,对(j)个集合均有影响,因此会增加价值为((a[i+1]-a[i]) imes j),设其为(t)。
转移分四种情况:
1. 该点单独成一个集合,则(j)不变,仅有1种情况,(f[i+1][j][k+t]+=f[i][j][k]);
2. 该点作为某个集合的中间元素,则(j)也不变,对于(j)个集合都有可能,因此(f[i+1][j][k+t]+=f[i][j][k] imes j);
3. 该点作为一个集合的起点,则集合数多了1,仅有1种情况,(f[i+1][j+1][k+t]+=f[i][j][k]);
4. 该点作为一个集合的终点,则集合数少了1,对于(j)个集合都有可能,因此(f[i+1][j−1][k+t]+=f[i][j][k] imes j)。
答案为(sumlimits_{i=0}^K f[n][0][i])。
时空复杂度均为(O(n^2K)),空间复杂度可以用滚动数组优化到(O(nK))。
C++ Code:
#include<bits/stdc++.h>
using namespace std;
const int md=1e9+7;
int n,K,cur,a[201],dp[2][201][1001];
int main(){
ios::sync_with_stdio(0),cin.tie(0);
cin>>n>>K;
for(int i=1;i<=n;++i)cin>>a[i];
sort(a+1,a+n+1);
dp[1][1][0]=dp[1][0][0]=cur=1;
for(int i=1;i<n;++i){
const int otr=cur;
memset(dp[cur^=1],0,sizeof dp[0]);
for(int j=0;j<=i;++j){
const int t=j*(a[i+1]-a[i]);
for(int k=0;k+t<=K;++k){
dp[cur][j][k+t]=(dp[cur][j][k+t]+dp[otr][j][k]*(j+1ll)%md)%md;
if(j<n)dp[cur][j+1][k+t]=(dp[cur][j+1][k+t]+dp[otr][j][k])%md;
if(j)dp[cur][j-1][k+t]=(dp[cur][j-1][k+t]+1ll*dp[otr][j][k]*j%md)%md;
}
}
}
int ans=0;
for(int i=0;i<=K;++i)
ans=(1ll*dp[cur][0][i]+ans)%md;
cout<<ans<<endl;
return 0;
}