B题:Two Sets CodeForces - 468B(并查集)
**题意:**输入n,a,b。再输入n个数p1,p2,…pn。有两个集合A和B,这n个数中如果x属于集合A,a-x也必须属于集合A;如果x属于集合B,b-x也必须属于集合B。如果能分为两个集合的话,输出“YES”并输出每个元素属于集合A还是B,0表示属于集合A,1表示属于集合B。
**思路:**先假设存在一种合理的方案,即任意一个x,一定属于集合A和B中的一个。如果最后结果合理集合A、B应该没有交集。设n+1和n+2分别为集合A和B的根节点。用一个map存储元素对应的下标。
- 对于任意一个x,如果a-x不存在,则x一定属于集合B,把x下标和集合B(即n+2)并起来;如果b-x也不存在,则x又一定属于集合A,把x下标和集合A(即n+1)并起来,此时集合A,B并了起来,不合理。
- 对于任意一个x,如果a-x存在,则把x和a-x下标并起来,此时x不一定属于集合A,因为可能存在n=2,a=8,b=20 4,16的数据。只有当b-x不存在时,才能把x的下标和A并起来。
- 如果b-x也存在,就把x和b-x的下标也并起来,此时如果x!=a-x!=b-x时,说明集合A和B公用x,显然不合理。此时如果x==a-x!=b-x,显然应属于集合B。特例就是a和b相等,此时任何分组都可以。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<map>
using namespace std;
int fa[100010],p[100010];
int n, a, b, x, fx, fy;
map<int, int>mp;
int find(int x) {
return x == fa[x] ? x : fa[x] = find(fa[x]);
}
void baba(int x, int y) {
int fx = find(x), fy = find(y);
fa[fx] = fy;
}
int main() {
scanf("%d %d %d", &n, &a, &b);
for (int i = 1; i <= n + 2; i++) fa[i] = i;
for (int i = 1; i <= n; i++) {
scanf("%d", &p[i]);
mp[p[i]] = i;
}
for (int i = 1; i <= n; i++) {
x = p[i];
if (mp[a-x] != 0) baba(i, mp[a-x]);
else baba(i, n + 2);
if (mp[b-x] != 0) baba(i, mp[b-x]);
else baba(i, n + 1);
}
int fx = find(n + 1);
int fy = find(n + 2);
if (fx == fy) {
printf("NO\n");
return 0;
}
printf("YES\n");
for (int i = 1; i <= n; i++) {
if (find(i) == fx) printf("0 ");
else printf("1 ");
}
}
F题:The LCMs Must be Large CodeForces - 1166E(思维)
**题意:**第一行输入m,n。接下来m行每行的第一位输入si,后跟si个数为下标值。这si个数字的LCM(输入的si个数字) > LCM(ai序列中剩余的n-si个数字)。判断是否存在这个ai序列。
**思路:**假设A是B 的子集,那么一定存在 LCM(B)≥LCM(A)LCM(B) 。反证法,某行输入集合Di,补集为Si,再取另一行集合为Dj,补集为Sj。假设Di和Dj的交集为空,那么Dj一定为Si的真子集,同理,Di也是Sj的真子集,那么:LCM(Di)>LCM(Si)>=LCM(Dj)>LCM(Sj)>=LCM(Di)。显然产生矛盾。结论:任意两行的集合D都必须有交集。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
using namespace std;
int m, n, s, x, flag;
bool vis[10010][10010];
int main() {
scanf("%d %d", &m, &n);
for (int i = 1; i <= m; i++) {
scanf("%d", &s);
for (int j = 1; j <= s; j++) {
scanf("%d", &x);
vis[i][x] = 1;
}
}
for (int i = 1; i <= m; i++) {
for (int j = 1; j <= m; j++) {
flag = 0;
for (int k = 1; k <= n && flag == 0; k++) {
if (vis[i][k] && vis[j][k]) flag = 1;
}
if (flag == 0) {
printf("impossible");
return 0;
}
}
}
printf("possible");
}