51Nod1557 两个集合（set容器）

小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+10 typedef 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"); }