Codeforces 1133E

给n个数，选k段，每一段中最大的数和最小数之差不能大于5，问最多能选多少个数。
先排序。
dp[i][j]表示前i个数选j段能选的最大的个数，用upperbound()函数可以求出小于等于a[i] + 5的个数，
考虑到下标为i时，有d[i+1+cnt][j] = max(dp[i+1+cnt][j],dp[i][]-1] + fr + 1);
答案就是dp[n][k]

#include<bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i = 0;i<int(n);i++)
typedef long long LL;
int n,k,a[5050],res,dp[5050][5050];

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>n>>k;
	for(int i =0;i<n;i++) cin>>a[i];
	sort(a,a+n); 
	for(int i = 0;i<n;i++){ 	
		int fr = upper_bound(a,a+n,a[i]+5) - a - i - 1; //fr 小于等于a[i]+5 的数的个数 
		for(int j = 0;j<k;j++){
			dp[i+1][j] = max(dp[i+1][j],dp[i][j]);
			if(j+1<=k) dp[i+1+fr][j+1] = max(dp[i+1+fr][j+1],dp[i][j] + fr+1);
		} 
	}
	cout<<dp[n][k];
	return 0; 
}