1349: [蓝桥杯2017决赛]对局匹配 （dp）

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+5;
int q[maxn];;///q[i]=k 代表积分为i的用户有k个人
int n,k;
ll dp[maxn];
ll max(ll a,ll b)
{
    if(a>b) return a;
    else return b;
}
ll f(int beg,int ed)
{
    memset(dp,0,sizeof(dp));
    int i;
    dp[beg]=q[beg];;///初始化 前 2个 连续
    dp[beg+k]=q[beg+k];
 ///i从相互冲突的第3个人开始
    for(int i=beg+2*k; i<=ed; i=i+k)
    {
        dp[i]=dp[i-2*k]+q[i];
        dp[i-k]=max(dp[i-k],dp[i-2*k]);
    }
    return max(dp[ed],dp[ed-k]);///比较一下dp数组最后两个值的大小，返回较大的那个
}
int main()
{
    ios::sync_with_stdio(false);
    memset(q,0,sizeof(q));
    scanf("%d%d",&n,&k);
    ll sum=0;
    int maxx=0;
    for(int i=1; i<=n; i++)
    {
        int x;
        scanf("%d",&x);
        q[x]++;
        if(maxx<x)
        {
            maxx=x;
        }
    }
    if(k==0)///为0的时候特殊处理 每种积分 只允许1个人
    {
        for(int i=0; i<=maxx; i++)
        {
            if(q[i]!=0)
            {
                sum++;
            }
        }
    }
    else
    {
        for(int i=0; i<=maxx; i++)
        {
            int t=1;
            while(q[t*k+i]!=0)///找出 有多少分数差k的用户（就是同时在线会冲突）
            {
                t++;
            }
            if(t!=1)
            {
                sum+=f(i,(t-1)*k+i);
      ///有冲突的人 的起始用户积分与结束用户积分
            }
            else sum+=q[i];///该积分就一个人就不存在冲突加上就好

            t=1;
            while(q[t*k+i]!=0)///找过了就清零
            {
                q[t*k+i]=0;
                t++;
            }
        }
    }
    cout<<sum<<endl;
    return 0;
}