BZOJ 1123 tarjan求割点+组合数学

https://www.lydsy.com/JudgeOnline/problem.php?id=1123
Byteotia城市有n个 towns m条双向roads. 每条 road 连接 两个不同的 towns ,没有重复的road. 所有towns连通。
Input

输入n<=100000 m<=500000及m条边

Output

输出n个数，代表如果把第i个点去掉，将有多少对点不能互通。

Sample Input
5 5
1 2
2 3
1 3
3 4
4 5
Sample Output
8
8
16
14
8
Hint

思路：根据割点的定义，若节点$i$不是割点，那么把节点$i$关联的边都去掉之后，只有$i$与其他$n-1$个点不连通，因此答案为$2\ast (n-1)$；若节点$i$为割点，那么去掉与节点$i$关联的边后，会把图分成几个连通块，我们应该分别求得这些连通块的大小，再计算答案。不妨设节点$i$的子节点集合中，有$t$个点$s_1,s_2\dots \dots s_t$满足割点判定法则$dfn[i]<=low[s_k]$，那么删除与节点$i$关联的边后，无向图至多分成$t+2$个连通块：(1)节点$i$自身单独成一个连通块。(2)有$t$个连通块，分别由搜索树上以$s_k$为根的子树中的节点构成。(3)还可能有一个连通块由剩下的点构成。如图：

设$siz[x]$表示以$x$为根的子树的大小，那么删掉一个割点$i$的贡献为：$$\sum _{k=1}^{t}siz[s_k]\ast (n-siz[s_k])+1\ast (n-1)+(n-1-sum)\ast (1+sum)$$$$其中sum=\sum _{k=1}^{t}siz[s_k]$$
这些都是可以在$tarjan$的$dfs$过程中维护的。

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;

const int maxn=1e5+5;
const int maxm=5e5+5;

struct edge
{
    int to,nxt;
}Edge[maxm<<1];

bool cut[maxn];
int head[maxn],dfn[maxn],low[maxn],siz[maxn];
ll ans[maxn];
int n,m,tot,num,root;

inline void addedge(int x,int y)
{
    Edge[++tot].to=y,Edge[tot].nxt=head[x],head[x]=tot;
}

void tarjan(int x)
{
    int y,flag=0,sum=0;
    dfn[x]=low[x]=++num;
    siz[x]=1;
    for(int i=head[x];i;i=Edge[i].nxt)
    {
        y=Edge[i].to;
        if(!dfn[y])//未访问过的节点
        {
            tarjan(y);
            siz[x]+=siz[y];
            low[x]=min(low[x],low[y]);
            if(low[y]>=dfn[x]) //桥
            {
                ++flag;
                ans[x]+=ll(siz[y])*(n-siz[y]);
                sum+=siz[y];
                if(x!=root||flag>=2)
                    cut[x]=1;
            }
        }
        else
            low[x]=min(low[x],dfn[y]);
    }
    if(cut[x])
        ans[x]+=n-1+ll(n-1-sum)*(1+sum);
    else
        ans[x]=2*(n-1);
}

int main()
{
    scanf("%d%d",&n,&m);
    int x,y;
    tot=1;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&x,&y);
        addedge(x,y),addedge(y,x);
    }
    root=1;
    tarjan(1);
    for(int i=1;i<=n;i++)
        printf("%lld\n",ans[i]);
    return 0;
}