再学一道算法题: 食物链(带权并查集)

简介: 再学一道算法题: 食物链(带权并查集)

动物王国中有三类动物 A,B,C,这三类动物的食物链构成了有趣的环形。

A 吃 B,B 吃 C,C 吃 A。

现有 N 个动物,以 1∼N 编号。

每个动物都是 A,B,C 中的一种,但是我们并不知道它到底是哪一种。

有人用两种说法对这 N 个动物所构成的食物链关系进行描述:

第一种说法是 1 X Y,表示 X 和 Y 是同类。

第二种说法是 2 X Y,表示 X 吃 Y。

此人对 N 个动物,用上述两种说法,一句接一句地说出 K 句话,这 K 句话有的是真的,有的是假的。

当一句话满足下列三条之一时,这句话就是假话,否则就是真话。

当前的话与前面的某些真的话冲突,就是假话;
当前的话中 X 或 Y 比 N 大,就是假话;
当前的话表示 X 吃 X,就是假话。
你的任务是根据给定的 N 和 K 句话,输出假话的总数。

输入格式
第一行是两个整数 N 和 K,以一个空格分隔。

以下 K 行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中 D 表示说法的种类。

若 D=1,则表示 X 和 Y 是同类。

若 D=2,则表示 X 吃 Y。

输出格式
只有一个整数,表示假话的数目。

数据范围
1≤N≤50000,
0≤K≤100000
输入样例:
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例:
3
说明:
image.png

using namespace std;
const int maxn = 5e4 + 233;
int fa[maxn], d[maxn];
int ff(int x)
{
    if (fa[x] == x) return x;
    int r = ff(fa[x]);
    d[x] += d[fa[x]];
    return fa[x] = r;
}
int main()
{
    int n, k;
    cin >> n >> k;
    for (int i = 0; i <= n; i++) fa[i] = i;
    int ans = 0;
    for (int i = 1; i <= k; i++)
    {
        int t, a, b;
        scanf("%d%d%d", &t, &a, &b);
        if (a > n || b > n)
        {
            ans ++;
            continue;
        }
        else if (t == 2 && a == b)
        {
            ans++;
            continue;
        }
        else
        {
            int rel;
            if (t == 2) rel = 1;
            else rel = 0;
            int x = ff(a), y = ff(b);
            if (x == y)
            {
                if ((((d[a] - d[b]) % 3) + 3) % 3 != rel)
                    ans++;
            }
            else
            {
                fa[x] = y;
                d[x] = d[b] - (d[a] - rel);
            }
        }
    }
    cout << ans;
}
相关文章
|
1月前
|
机器学习/深度学习 存储 算法
【算法沉淀】刷题笔记:并查集 带权并查集+实战讲解
【算法沉淀】刷题笔记:并查集 带权并查集+实战讲解
|
4月前
|
算法
并查集的实现【学习算法】
并查集的实现【学习算法】
23 0
|
4月前
|
算法
算法专栏 ---- trie树,并查集
算法专栏 ---- trie树,并查集
|
9月前
|
算法
算法 - 蓝桥杯并查集题型
算法 - 蓝桥杯并查集题型
|
10月前
|
存储 算法
从小白开始刷算法 并查集篇 leetcode.547
从小白开始刷算法 并查集篇 leetcode.547
|
10月前
|
存储 算法 索引
从小白开始刷算法 并查集篇 leetcode.200
从小白开始刷算法 并查集篇 leetcode.200
|
10月前
|
存储 算法 API
LeetCode算法小抄 -- 经典图论算法 之 并查集算法
LeetCode算法小抄 -- 经典图论算法 之 并查集算法
|
存储 算法
给我三分钟,带你领略热血江湖中的并查集算法
给我三分钟,带你领略热血江湖中的并查集算法
给我三分钟,带你领略热血江湖中的并查集算法
|
存储 算法
【数据结构与算法】并查集
【数据结构与算法】并查集
【数据结构与算法】并查集
|
算法
最小生成树:Kruskal算法(邻接表+最小堆+并查集)
最小生成树:Kruskal算法(邻接表+最小堆+并查集)
214 0
最小生成树:Kruskal算法(邻接表+最小堆+并查集)