[蓝桥杯][2017年第八届真题]对局匹配（思维）

题目描述

小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分，代表他的围棋水平。
小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是K的两名用户匹配在一起。如果两人分差小于或大于K，系统都不会将他们匹配。
现在小明知道这个网站总共有N名用户，以及他们的积分分别是A1, A2, ... AN。
小明想了解最多可能有多少名用户同时在线寻找对手，但是系统却一场对局都匹配不起来(任意两名用户积分差不等于K)？  

输入

第一行包含两个个整数N和K。
第二行包含N个整数A1, A2, ... AN。  
对于30%的数据，1 <= N <= 10
对于100%的数据，1 <= N <= 100000, 0 <= Ai <= 100000, 0 <= K <= 100000

输出

一个整数，代表答案。

样例输入

10 0
1 4 2 8 5 7 1 4 2 8

样例输出

6

提示

无

来源

无

思路：去抉择a[t]和a[t]+K数量最多的

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<vector>
#include<cmath>

const int maxn=1e5+5;
typedef long long ll;
using namespace std;
int a[maxn];
int vis[2*maxn];
int sum[maxn];
set<int>s;
int main()
{
    int N,K;
    cin>>N>>K;
    if(K==0)
    {
    for(int t=0;t<N;t++)
    {
        scanf("%d",&a[t]);
        s.insert(a[t]);
        sum[a[t]]++;
        
    } 
    printf("%d\n",s.size());
    }
    else
    {
    for(int t=0;t<N;t++)
    {
        scanf("%d",&a[t]);
        sum[a[t]]++;
    } 
    int s=0;
    for(int t=0;t<N;t++)
    {
        //cout<<sum[a[t]]<<endl;
        if(vis[a[t]]==0)
        {
        //    cout<<t<<" "<<sum[a[t]]<<endl;
            if(sum[a[t]]>=sum[a[t]+K])
            {
                vis[a[t]+K]=1;
                vis[a[t]]=1; 
                s+=sum[a[t]];
            }
            else
            {
                vis[a[t]]=1; 
                vis[a[t]+K]=1;
                s+=sum[a[t]+K];
            }
        }
    }
    cout<<s<<endl;
    }
    return 0;
}