E. K Balanced Teams
这个题刚一看,感觉像是一个贪心,又像是DP,我就以DP问题来想这个题了。
挖掘题中的性质,我刚开始想找每个点最多能到右取几个点作为一组,这个很容易想到,可以用排序+二分来处理,可以把每个点都处理出来,但是在想DP的时候却发现这样并不好DP,很难找到转移方程式。
最后看了题解,发现别人是以最大值为终点,向前找,这样就可以很好的找到转移方程式。
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=5005;
int a[N],dp[N][N],num[N];
int main()
{
int n,k;
cin>>n>>k;
for(int i=1 ; i<=n ; i++) cin>>a[i];
sort(a+1,a+1+n);
for(int i=1 ; i<=n ; i++)
{
int low=a[i]-5;
num[i]=i+1-(lower_bound(a+1,a+1+n,low)-a);
}
for(int i=1 ; i<=n ; i++)
{
for(int j=1 ; j<=k ; j++)
{
dp[i][j]=max(dp[i][j],dp[i-1][j]);
dp[i][j]=max(dp[i][j],dp[i-num[i]][j-1]+num[i]);
}
}
cout<<dp[n][k]<<endl;
return 0;
}