Codeforces Round #544 (Div. 3):E. K Balanced Teams

题目链接：http://codeforces.com/contest/1133/problem/E

题意：现在有n个人，你需要组k个队伍出来，每个队伍里面两两之间人能力值的差值不能大于5，问你最多能够选多少个人出来组队。

解题心得：

• 很明显是一个dp题，先对所有人按照能力排序，便于在找能力值的时候二分查找。
• dp[i][j]表示在前j个人，能够组i个队伍可以选择的最大的人数，转移方程也很简单
  dp[i][j] = max(dp[i][j-1], dp[i-1][比第j个人能力值小5的人的位置]+(j-比第j个人能力值小5的人的位置)

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5010;

int num[maxn], n, k, dp[maxn][maxn];

void init() {
    scanf("%d%d", &n , &k);
    for(int i=1;i<=n;i++) {
        scanf("%d",&num[i]);
    }
    sort(num+1, num+n+1);
}

void DP() {
    for(int i=1;i<=k;i++) {
        for(int j=1;j<=n;j++) {
            int va = num[j] - 6;
            int pos = int(upper_bound(num+1, num+n+1, va) - num - 1);
            dp[i][j] = max(dp[i][j-1], dp[i-1][pos] + (j - pos));
        }
    }

    printf("%d\n", dp[k][n]);
}

int main() {
    //freopen("1.in", "r", stdin);
    init();
    DP();
    return 0;
}