描述

三種動物 A吃B, B吃C, C吃A.

1, a, b 表示a,b同類

2, a, b 表示a吃b

判斷假話個數

分析

• NOIp前看過, 當時不會, 現在理解起來也有點抽象.

• 按權值合并的并查集

• 能確定關系的 x, y 的find到的祖先相同.

• pa[x] 是 x 的祖先, r[x] 是 x 到祖先的距離, 有三種, 0, 1, 2. ?0 表示和祖先同類, 1表示可以吃祖先, 2表示被吃. ?r[find(x)] 一定為 0 我用 assert 驗證了一下.

• 查找時先記錄下 x 當前的祖先 px, 可知當 x 和 px 最初確定關系的時候 r[px] 一定為0. 而遞歸 find(px) 之后 r[px] 的數值就是 px 到現在的祖先的距離, x 和 px 現在的祖先是相同的, 那么 x 到現在祖先的距離就是 x 到原來 px 的距離加上 px 到現在祖先的距離.

• 合并 x, y 時, 先px = find(x), py = find(y). 比如要把 pa[px] 設成 py, 那 py 是不會變了, 考慮 px 的秩怎么變. 如果合并后要求 x, y 同類, 那么 x 到 py 的距離應當等于此時的 r[y], 那么 r[x] + r[px] == r[y], 所以 r[px] = r[y] - r[x]. 這樣在路徑壓縮后新的 r[x] 就等于 r[y] 了.

#include #include using namespace std;const int maxn = 50000 + 10; int pa[maxn], r[maxn];int find(int x) {if(pa[x] == x) return x;int px = pa[x];pa[x] = find(px);r[x] = (r[px] + r[x]) % 3;return pa[x]; }int main() {int n, m, ans = 0;scanf("%d %d", &n, &m);for(int x = 1; x <= n; x++) pa[x] = x;for(int i = 1; i <= m; i++) {int k, x, y, px, py;scanf("%d %d %d", &k, &x, &y);if(x > n || y > n) { ans++; continue; }px = find(x); py = find(y);assert(r[px] == 0);assert(r[py] == 0);if(k == 1) {if(px == py) { if(r[x] != r[y]) ans++; }else pa[px] = py, r[px] = (r[y]-r[x]+3) % 3;} else {if(px == py) { if(r[x] != (r[y]+1) % 3) ans++; }else pa[px] = py, r[px] = (r[y]-r[x]+4) % 3;}}printf("%d\n", ans);return 0; }

總結

以上是生活随笔為你收集整理的CODEVS-1074-食物链-并查集的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。