time limit per test3 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
You are a coach at your local university. There are n students under your supervision, the programming skill of the i-th student is ai.
You have to form k teams for yet another new programming competition. As you know, the more students are involved in competition the more probable the victory of your university is! So you have to form no more than k (and at least one) non-empty teams so that the total number of students in them is maximized. But you also know that each team should be balanced. It means that the programming skill of each pair of students in each team should differ by no more than 5. Teams are independent from one another (it means that the difference between programming skills of two students from two different teams does not matter).
It is possible that some students not be included in any team at all.
Your task is to report the maximum possible total number of students in no more than k (and at least one) non-empty balanced teams.
If you are Python programmer, consider using PyPy instead of Python when you submit your code.
Input
The first line of the input contains two integers n and k (1≤k≤n≤5000) — the number of students and the maximum number of teams, correspondingly.
The second line of the input contains n integers a1,a2,…,an (1≤ai≤109), where ai is a programming skill of the i-th student.
Output
Print one integer — the maximum possible total number of students in no more than k (and at least one) non-empty balanced teams.
Examples
inputCopy
5 2
1 2 15 15 15
outputCopy
5
inputCopy
6 1
36 4 1 25 9 16
outputCopy
2
inputCopy
4 4
1 10 100 1000
outputCopy
4
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long long LL;
const int maxnnn=0x3f3f3f3f;
struct node {
LL x,y;
} ppppos[100050];
bool cmp(ll a,ll b) {
return a>b;
}
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;
}