ZZULI-2019年3月份月赛（个人赛） 问题 A: 小P参加相亲大会 思维

题目描述

小P最近人生得意，去参加了一次相亲大会，相亲大会上每个人有一个密码牌（密码牌上的密码是一个正整数m，m<231 ），相互之间在交流之前先交换密码牌，密码牌上的密码可能相同，也可能不同，如果相同，两人牵手离开，如果不保同，各自再寻找下一位，保证最后只有1个人或2个人留下来。 
 

输入

第一行两个数 n,k (n≤3000000,1≤k≤2)，n表示参加相亲大会的人数，接下来 n行每行一个正整数表示相亲大会上每一个人的密码，k表示最后留在相亲大会的人数。

输出

从小到大输出一行 k个数，表示相亲不成功留在相亲大会人的密码，中间用空格分隔。

样例输入 Copy

3 1
2
2
2

样例输出 Copy

2

这个题注意卡内存，16MB，3个1e6的数组好像都卡不过。

我们想，数据保证有答案，那我们对数组排序，记录每个数的个数即可。如果k为1，则一定只有一个数的个数为奇数。

因为k最大为2，所以我们完全可以用两个变量来记录。

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e6+10;
int b[maxn];
int main()
{
    //freopen("C://input.txt","r",stdin);
    int n,k;
    cin >> n >> k;
    int pos=0;
    int k1,k2;
    k1=k2=-1;
    memset(b,0,sizeof(b));
    for(int i=0;i<n;i++)
    {
        cin >> b[i];
    }
    sort(b,b+n);
    int sum=0;
    for(int i=0;i<n;i++)
    {
        int ans=b[i];
        sum++;
        if(b[i]!=b[i+1]&&sum%2==1)
        {
            if(k1==-1)
            {
                k1=ans;
                sum=0;
            }
            else if(k2==-1)
            {
                k2=ans;
                sum=0;
            }
        }
        else if(b[i]!=b[i+1]&&sum%2==0)
        {
            sum=0;
        }
    }
    if(k==1)
    {
        printf("%d\n",k1);
    }
    if(k==2)
    {
        printf("%d %d\n",k1,k2);
    }
    return 0;
}
/**************************************************************
    Problem: 1519
    User: 1610101013
    Language: C++
    Result: 正确
    Time:1703 ms
    Memory:13372 kb
****************************************************************/