【Leetcode】2316. Count Unreachable Pairs of Nodes in an Undirected Graph

记录算法题解

于 2023-11-02 02:45:37 发布

阅读量74

点赞数

分类专栏： LC 栈、队列、串及其他数据结构文章标签： leetcode 算法 c++ 数据结构

本文链接：https://blog.csdn.net/qq_46105170/article/details/134173588

版权

LC 栈、队列、串及其他数据结构专栏收录该内容

678 篇文章

订阅专栏

题目地址：

https://leetcode.com/problems/count-unreachable-pairs-of-nodes-in-an-undirected-graph/description/

有一个 $n$ 个顶点的无向图，给出图中的 $e$ 条边，问有多少个点对是不连通的。 $(a, b)$ 和 $(b, a)$ 视为同一个点对。

先用并查集统计一下每个连通块的点数，设点数为 $v_i$ ，则答案为 $\frac{1}{2}\sum_i v_i(n-v_i)$ 代码如下：

class Solution {
 public:
  using ll = long long;
  ll countPairs(int n, vector<vector<int>>& es) {
    vector<int> p(n), sz(n, 1);
    for (int i = 0; i < n; i++) p[i] = i;
    function<int(int)> find = [&](int x) {
      if (x != p[x]) p[x] = find(p[x]);
      return p[x];
    };

    for (auto& e : es) {
      int px = find(e[0]), py = find(e[1]);
      if (px != py) {
        p[px] = py;
        sz[py] += sz[px];
      }
    }

    ll res = 0;
    for (int i = 0; i < n; i++)
      if (i == p[i]) res += (ll)sz[i] * (n - sz[i]);

    return res / 2;
  }
};

时间复杂度