[APIO2018] Duathlon 铁人两项（圆方树+树形DP）

[APIO2018] Duathlon 铁人两项

洛谷题目传送门
如果没学过圆方树的可以进入我的博客 圆方树学习记录及例题

解题思路

首先这是一道无向图计数问题，因此我们考虑在圆方树上计算
建出圆方树后，考虑如果一条路径的起点和终点已经确定，那么中点有几种选择呢？
首先，这条路径上的圆点肯定可以作为中点，其次路径上的方点连的圆点也可以是中点（因为方点代表一个点双连通分量，它可以联通）
因此对于一个路径，我们给每个点付一个权值，如果是圆点权值是1，就代表这一个点，方点的权值就是它对应的点双大小减二，这是因为它的点双里有两个圆点本身就在路径上，那么这条路径的中点个数就是起点与重点之间的点的权值之和。
但是如果枚举起点和终点的话，复杂度太大，因此我们把问题转换成，对于一个点，计算有多少条路径经过它 ，然后把条数乘上这个点的权值。
这就是经典的树上技术问题了，对于一个节点，路径条数分两种，在它子树中和跨过他的
第一种的个数就是任意两个不同子树中圆点个数的乘积之和可以在算子树大小时顺便算出来

	ans+=((siz[x]-is[x])*1ll*siz[y]*1ll*v[x]);//siz是子树中的圆点个数，is表示当前这个点是圆点还是方点
	siz[x]+=siz[y];//这两个的顺序一定不能弄反

第二种就是子树中的圆点个数乘上不在子树中的圆点个数
注意有可能图不连通因此可能有若干棵树，每一棵树的圆点总数都需要记录

	ans+=(siz[x]-is[x])*1ll*(C[root]-siz[x])*1ll*v[x];//C代表每一棵树的圆点个数

因为起点和终点可以交换，因此最终答案需要乘2

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+7;
const int M = 2e5+7;
typedef long long LL;
struct node
{
	LL y,next;
}e[2*M];
LL link[M],t=0,n,m;
void add(LL x,LL y)
{
	e[++t].y=y;
	e[t].next=link[x];
	link[x]=t;
}
LL dfn[N],low[N],cnt=0,num=0;
vector<LL> dcc[N];
stack<LL> st;
LL root;
LL C[N];
void tarjan(LL x)
{
	dfn[x]=low[x]=++num;
	C[root]++;
	st.push(x);
	if(x==root&&link[x]==0)
	{
		dcc[++cnt].push_back(x);
		return;
	}
	for(LL i=link[x];i;i=e[i].next)
	{
		LL y=e[i].y;
		if(!dfn[y])
		{
			tarjan(y);
			low[x]=min(low[x],low[y]);
			if(dfn[x]<=low[y])
			{
				cnt++;
				LL z;
				do
				{
					z=st.top();
					st.pop();
					dcc[cnt].push_back(z);
				}while(y!=z);
				dcc[cnt].push_back(x);
			}
		}
		else low[x]=min(low[x],dfn[y]);
	}
}
LL s=0;
LL v[M];
void construct()
{
	t=0;
	s=n;
	memset(link,0,sizeof(link));
	for(LL i=1;i<=n;i++)
	v[i]=1;
	for(LL i=1;i<=cnt;i++)
	{
		LL x=++s;
		for(LL j=0;j<dcc[i].size();j++)
		{
			LL y=dcc[i][j];
			add(x,y);
			add(y,x);
			v[x]++;
		}
		v[x]-=2;
	}
}
LL ans=0;
LL siz[M],is[M];
LL vis[M];
void dfs(LL x,LL pre)
{
	vis[x]=1;
	siz[x]=(x<=n);
	is[x]=(x<=n);
	for(LL i=link[x];i;i=e[i].next)
	{
		LL y=e[i].y;
		if(y==pre) continue;
		dfs(y,x);
		ans+=((siz[x]-is[x])*1ll*siz[y]*1ll*v[x]);
		siz[x]+=siz[y];
	}
	ans+=(siz[x]-is[x])*1ll*(C[root]-siz[x])*1ll*v[x];
}
inline LL read()
{
	LL X=0; bool flag=1; char ch=getchar();
	while(ch<'0'||ch>'9') {if(ch=='-') flag=0; ch=getchar();}
	while(ch>='0'&&ch<='9') {X=(X<<1)+(X<<3)+ch-'0'; ch=getchar();}
	if(flag) return X;
	return ~(X-1);
}
int main()
{
	n=read();
	m=read();
	for(LL i=1;i<=m;i++)
	{
		LL x,y;
		x=read();
		y=read();
		add(x,y);
		add(y,x);
	}
	for(LL i=1;i<=n;i++)
	{
		root=i;
		if(!dfn[i]) tarjan(i);
	}

	construct();
	for(LL i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			root=i;
			dfs(i,i);
		}
	}
	cout<<2*1ll*ans;
	return 0;
}