两个集合 51Nod - 1557

https://www.51nod.com/Challenge/Problem.html#!#problemId=1557

不考虑特殊情况 将x-a与 x与x-b建边的话 得到很多条链 判一下长度奇偶就行

但是会出现a-x=x即a=2*x的情况 比如n=2 a=2 b=3 x[1]=2,x[2]=1 这时就要分个先后了 可以贪心的从链的端点开始 如果碰到a=b=2*x这种情况就不用管x了

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;

map <int,int> mp;
queue <int> que;
int val[maxn];
int n,a,b;

int main()
{
    int i,cur,tmp,flag,gou;
    scanf("%d%d%d",&n,&a,&b);
    for(i=1;i<=n;i++){
        scanf("%d",&val[i]);
        mp[val[i]]=i;
    }
    for(i=1;i<=n;i++){
        if(!mp[a-val[i]]||!mp[b-val[i]]){
            que.push(i);
        }
    }
    flag=1;
    while(!que.empty()){
        cur=que.front();
        que.pop();
        if(mp[val[cur]]){
            tmp=mp[a-val[cur]],gou=0;
            if(tmp){
                mp[val[cur]]=0,mp[a-val[cur]]=0;
                que.push(tmp);
                gou=1;
            }
            tmp=mp[b-val[cur]];
            if(tmp){
                mp[val[cur]]=0,mp[b-val[cur]]=0;
                que.push(tmp);
                gou=1;
            }
            if(!gou){
                flag=0;
                break;
            }
        }
    }
    if(flag) printf("YES\n");
    else printf("NO\n");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sunyutian1998/article/details/84280054
今日推荐