并查集有关
食物链
分析:
存在A吃B,B吃C,C吃A这三种关系,即存在三种生物,三个集合,而这三种生物又是等价的,对于每一个生物x,他都可以是A,也可以是B,也可以是C,假定fa[x],fa[n + x], fa[2 * n + x]分别是生物x是A,B,C的情况下的父节点。对于每次的两个生物都要把他们当作A,B,C分别看一遍。
eg.
2 2 3
1 1 3
第一句话表示2吃3,即2是A的时候,3不能是A,和C,2是B的时候,3不能是B和A,2是C的时候,3不能是C和B,判断后合并
第二句话同样判断
AC代码:
#include <cstdio> using namespace std; int n, k; int fa[150050]; int ans; int d, x, y; int findfa(int x) { return x == fa[x] ? x : fa[x] = findfa(fa[x]); } void merge_(int x,int y) { fa[findfa(y)] = findfa(x); } int main() { scanf("%d%d", &n, &k); for(int i = 1; i <= 3 * n; ++i) fa[i] = i; for(int i = 1; i <= k; ++i) { scanf("%d%d%d", &d, &x, &y); if(x > n || y > n) ans++; else if(d == 2 && x == y) ans++; else if(d == 1) { if(findfa(x) == findfa(y + n) || findfa(x) == findfa(2 * n + y)) ans++; else { merge_(x, y); merge_(x + n, y + n); merge_(x + 2 * n, y + 2 * n); } } else if(d == 2) { if(findfa(x) == findfa(y) || findfa(x) == findfa(2 * n + y)) ans++; else { merge_(x, y + n); merge_(x + n, y + 2 * n); merge_(x + 2 * n, y); } } } printf("%d\n", ans); return 0; }
关押罪犯
分析:
方法1:同食物链
AC代码:
#include <bits/stdc++.h> using namespace std; int n, m; struct node { int a, b, c; }ar[100050]; int fa[40050]; bool flag; bool cmp(struct node A, struct node B) { return A.c > B.c; } int findfa(int x) { return fa[x] == x ? x : fa[x] = findfa(fa[x]); } void merge_(int x, int y) { fa[findfa(y)] = findfa(x); } int main() { scanf("%d%d", &n, &m); for(int i = 1; i <= m; ++i) scanf("%d%d%d", &ar[i].a, &ar[i].b, &ar[i].c); for(int i = 1; i <= 2 * n; ++i) fa[i] = i; sort(ar + 1, ar + m + 1, cmp); for(int i = 1; i <= m; ++i) { if(findfa(ar[i].a) == findfa(ar[i].b) || findfa(ar[i].a + n) == findfa(ar[i].b + n)) { printf("%d\n", ar[i].c); flag = 1; break; } else { merge_(ar[i].a, ar[i].b + n); merge_(ar[i].a + n, ar[i].b); } } if(!flag) printf("0\n"); return 0; }
二分图
关押罪犯
方法二:二分图
如果a,b打架会造成影响值c,那么就在a,b之间连一条长为c的边。
然后利用二分的思想,那剩下的就是如何判断。
我们利用一个数组color[i]来记录点i的颜色(1或2),如果两个点之间有边的话,一个点的颜色是1,另一个点就必须是2,不然就表明这两个点在一块了,即两个人在一个监狱。
AC代码:
#include <bits/stdc++.h> using namespace std; int n, m; int a, b, c; int tot; int head[20050]; struct node { int to, l, next; }edge[200050]; int color[20050]; int l, mid, r; void addedge(int u, int v, int w) { edge[++tot].to = v; edge[tot].l = w; edge[tot].next = head[u]; head[u] = tot; } //进入dfs,首先把点u染成颜色c,然后遍历与点u相连的点 //如果某个相连的点的边权小于了k跳过就好,因为要让最大的最小,这个一定不是最大的 //在判断该点是否染色,染了的话看是不是一样,一样直接return false //如果没染色,就染色,做同样的操作(递归) //另外,对于染色了但是不冲突的,不用管就好,仔细想想确实不用管哈 bool dfs(int u, int c, int k) { color[u] = c; for(int i = head[u]; i != -1; i = edge[i].next) { if(edge[i].l <= k) continue; if(color[edge[i].to]) { if(color[edge[i].to] == c) return false; } else if(!dfs(edge[i].to, 3 - c, k)) return false; } return true; } bool check(int k) { memset(color, 0, sizeof(color)); for(int i = 1; i <= n; ++i) { if(!color[i]) { if(!dfs(i, 1, k)) return false; } } return true; } int main() { memset(edge, -1, sizeof(edge)); memset(head, -1, sizeof(head)); scanf("%d%d", &n, &m); for(int i = 1; i <= m; ++i) { scanf("%d%d%d", &a, &b, &c); addedge(a, b, c); addedge(b, a, c); } l = 0, r = 1e9; //这里二分条件写l<=r的话,带等号,下面l一定要等于mid + 1,r一定要等于mid - 1; while(l <= r) { mid = (l + r) >> 1; //cout << mid << endl; if(check(mid)) r = mid - 1; else l = mid + 1; } printf("%d\n", r + 1); return 0; }