codeforces 1133E 线性DP

codeforces 1133E

---

题意：

$给定n位学生的能力值，要求从中选出若干学生组成不超过k支队伍，且队伍内学生的能力差值不能超过5。$
$问组成的所有队伍中的最大人数。$

---

题解：

$dp[i][j]表示从前i位学生中挑选组成的j支队伍中的最大人数。$

• $dp[i][t]=max(dp[i-1][t],dp[j-1][t-1]+i-j+1)$

---

#include <bits\stdc++.h>
using namespace std;
const int N = 5001;
int a[N];
int dp[N][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 j = i;
        while(j > 1 && a[i]-a[j-1] <= 5)  j--;
        for(int t = 1 ; t <= k ; t++)
        dp[i][t] = max(dp[i-1][t], dp[j-1][t-1]+i-j+1);
    }
    cout << dp[n][k] << endl;
    return 0;
}