【Codeforces Round #544 (Div. 3) E. K Balanced Teams】DP

E. K Balanced Teams

题意

给你n个数，分成k组，要求每组内最大值与最小值的差值不超过5。求k组最多可以放多少个数。

$1\le n,k\le 5000$
做法

首先对数组排序，我们可以预处理每个数最多可以向左扩展的长度。
之后我们用$dp[i][j]$表示前i个数分为j组最多可以放多少个数。
对于每个i，一定是从之前预处理的位置转移过来，因为一段应该放尽量多的值。
我们设位置$i$能够扩展的最左位置为$l[i]$。那么转移方程为:
$dp[i][j]=max(dp[i?1][j],dp[i][j?1],dp[l[i]?1][j?1]+i?l[i]+1)$
最后$dp[n][k]$便是答案。
代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 5005;
int a[maxn],l[maxn];
int dp[maxn][maxn];
int main()
{
    int n,k;scanf("%d%d",&n,&k);for(int i=1;i<=n;i++) scanf("%d",&a[i]);sort(a+1,a+1+n);dp[0][0]=0;for(int i=1;i<=n;i++) l[i]=lower_bound(a+1,a+1+n,a[i]-5)-a;for(int i=1;i<=n;i++){
    for(int j=1;j<=i;j++){
    dp[i][j]=max(dp[i-1][j],dp[i][j-1]);dp[i][j]=max(dp[i][j],dp[l[i]-1][j-1]+i-l[i]+1);}}printf("%d\n",dp[n][k]);return 0;
}