【ACWing】363. B城

题目地址：

https://www.acwing.com/problem/content/description/365/

$B$城有$n$个城镇，$m$条双向道路。每条道路连结两个不同的城镇，没有重复的道路，所有城镇连通。把城镇看作节点，把道路看作边，容易发现，整个城市构成了一个无向图。

输入格式：
第一行包含两个整数$n$和$m$。接下来$m$行，每行包含两个整数$a$和$b$，表示城镇$a$和$b$之间存在一条道路。

输出格式：
输出共$n$行，每行输出一个整数。第$i$行输出的整数表示把与节点$i$关联的所有边去掉以后（不去掉节点$i$本身），无向图有多少个有序点$(x,y)$，满足$x$和$y$不连通。

数据范围：
$n\le 100000,m\le 500000$

对于点$u$，如果其不是割点，则去掉其所有边后，其余点都连通，那么点对就是$(u,v),(v,u),v\ne u$，个数为$2(n-1)$；如果$u$是割点，则用Tarjan算法，然后看$u$能割出多少个子树，设能割出$v_1,...,v_k$这些子树（不包括$u$的父亲可能割出的子树），设每个子树的点数为$s[v_i]$，则以$v_i$为第一维的点对数量是$s[v_i](n-s[v_i])$，总数为${\sum }_{i=1}^{k}s[v_i](n-s[v_i])$，第一维为$u$的点对数为$n-1$，还需要累加一下$u$的父亲的子树为第一维的点对，为$(n-1-{\sum }_{i=1}^{k}s[v_i])(1+{\sum }_{i=1}^{k}s[v_i])$，综上有$$\sum _{i=1}^{k}s[v_i](n-s[v_i])+(n-1)+(n-1-\sum _{i=1}^{k}s[v_i])(1+\sum _{i=1}^{k}s[v_i])$$个点对。其实如果$u$不是割点，这个公式也是对的，即此时$\forall i,s[v_i]=0$， 上面的公式算出来恰好也是$2(n-1)$。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 1e5 + 10, M = 1e6 + 10;
int n, m;
int h[N], e[M], ne[M], idx;
int dfn[N], low[N], timestamp;
int cnt[N];
long res[N];

void add(int a, int b) {
  e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void tarjan(int u, int from) {
  dfn[u] = low[u] = ++timestamp;
  // 记录一下u子树的点数
  cnt[u] = 1;
  
  // 存一下u能割出的子树总点数
  int sum = 0;
  for (int i = h[u]; ~i; i = ne[i]) {
    int v = e[i];
    if (!dfn[v]) {
      tarjan(v, u);
      cnt[u] += cnt[v];
      low[u] = min(low[u], low[v]);
      // 如果u能割出v子树，则累加v子树为第一维的点对数
      if (low[v] >= dfn[u]) {
        res[u] += (long) cnt[v] * (n - cnt[v]);
        // 累加割出的子树的总点数
        sum += cnt[v];
      }
    } else if (v != from) low[u] = min(low[u], dfn[v]);
  }
  
  // 累加一下以u自己为第一维的点对数和以u上面的点为第一维的点对数
  res[u] += (long) (n - 1 - sum) * (1 + sum) + (n - 1);
}

int main() {
  scanf("%d%d", &n, &m);
  memset(h, -1, sizeof h);
  while (m--) {
    int a, b;
    scanf("%d%d", &a, &b);
    add(a, b), add(b, a);
  }

  tarjan(1, 1);
  for (int i = 1; i <= n; i++)
    printf("%ld\n", res[i]);
}

时间复杂度$O(n+m)$，空间$O(n)$。