51Nod1557 两个集合(set容器)

来源:互联网 发布:linux db2数据库名 编辑:程序博客网 时间:2024/06/06 20:13

小X有n个互不相同的整数: p1,p2,…,pn 。他想把这些整数分到两个集合A和B里边。但是要符合下面两个条件。
· 如果x属于A,那么a-x也肯定属于A。
· 如果x属于B,那么b-x也肯定属于B。
判断一下是否存在一种方案来分配这些数字到集合A,B中。
注意:如果一个集合为空也是可以的。



Input
单组测试数据。
第一行有三个整数n,a,b (1≤n≤10^5; 1≤a,b≤10^9)。
第二行有n个不一样的整数 p1,p2,…,pn (1≤pi≤10^9).
Output
如果可行,那么输出YES,否则输出NO。
Input示例

样例输入1
4 5 9
2 3 4 5

Output示例

样例输出1
YES

wa了14次。。。
各种细节 各种不读题。。
n个互不相同的数
每次对当前数进行判断,如果属于集合A或者集合B,通过,若同时属于两个集合。继续判断(a-x)和(b-x)属于哪个集合。如果都不属于 那么直接退出遍历

#include <stdio.h>#include <set>using namespace std;#define N 100000+10typedef long long LL;set<LL>s;int main(){    LL n,a,b;    s.clear();    scanf("%lld %lld %lld",&n,&a,&b);    LL x;    for(int i=0;i<n;i++)    {        scanf("%lld",&x);        s.insert(x);    }    bool find;    set<LL>::iterator pr=s.begin();    while(pr!=s.end())    {        x=*pr;        find=false;        if(s.count(a-x))        {            find=true;        }        if(a!=b&&s.count(b-x))        {            if(find)            {                if(!s.count(a-b+x)&&!s.count(b-a+x))                find=false;            }            else            {                find=true;            }        }        if(!find)break;        pr++;    }       if(find)    puts("YES");    else    puts("NO");}
1 0
原创粉丝点击