zoukankan      html  css  js  c++  java
  • Codeforces 544E K Balanced Teams (DP)

    题目:

    You are a coach at your local university. There are nn students under your supervision, the programming skill of the ii-th student is aiai.

    You have to form kk 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 kk (and at least one) non-empty teams so that the totalnumber 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 55. 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 kk (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 nn and kk (1kn50001≤k≤n≤5000) — the number of students and the maximum number of teams, correspondingly.

    The second line of the input contains nn integers a1,a2,,ana1,a2,…,an (1ai1091≤ai≤109), where aiai is a programming skill of the ii-th student.

    Output

    Print one integer — the maximum possible total number of students in no more than kk (and at least one) non-empty balanced teams.

    Examples
    input
    5 2
    1 2 15 15 15
    
    output
    5
    
    input
    6 1
    36 4 1 25 9 16
    
    output
    2
    
    input
    4 4
    1 10 100 1000
    
    output
    4

    题意:
    有n个学生 要求组成k个小组 每个小组中两两差值不得超过5 可以有学生不被编入组中 求最多可以有多少个学生被编入组中

    思路:
    采取线性dp的思路
    将每个学生的值a[i]从小到大排序
    排序后 每个小组必定可以视为一段连续区域 则将pos设为一段连续区间的左端点
    dp[i][j]表示到第i个学生为止 前面分为j组的最多学生数
    状态转移方程为
    dp[i][j]=max(dp[i-1][j],dp[pos-1][j-1]+i-pos+1);

    代码:
    #include <iostream>
    #include <cstdio>
    #include <cstdlib>
    #include <cmath>
    #include <string>
    #include <cstring>
    #include <algorithm>
    
    using namespace std;
    typedef long long ll;
    typedef unsigned long long ull;
    const int inf=0x3f3f3f3f;
    const int maxn=5010;
    int n,k;
    int a[maxn],dp[maxn][maxn];
    
    int main(){
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        sort(a+1,a+1+n);
        int pos=1;
        dp[0][0]=0;
        for(int i=1;i<=n;i++){
            while(a[i]-a[pos]>5 && pos<=n) pos++;
            for(int j=1;j<=min(i,k);j++){
                dp[i][j]=dp[i-1][j];
                dp[i][j]=max(dp[i][j],dp[pos-1][j-1]+i-pos+1);
            }
        }
        int ans=0;
        for(int i=1;i<=k;i++){
            ans=max(ans,dp[n][i]);
        }
        printf("%d
    ",ans);
        return 0;
    }
  • 相关阅读:
    在线音视频(MP3/MP4)测试文件
    jQuery 找包含内容的节点,但不搜索子节点
    bash 脚本小练习:查看 git 提交对应的提交数
    桌面应用开发的日子(二):实现文件资源管理列表树加载
    桌面应用开发的日子(一):初识WPF
    Winform同一数据源多个控件保持同步
    Windows Form中DataGridView的基本玩法
    jsp第七周作业
    jsp第五周作业
    jsp第六周作业
  • 原文地址:https://www.cnblogs.com/whdsunny/p/10525719.html
Copyright © 2011-2022 走看看