loj2587「APIO2018」铁人两项（圆方树+树形dp）

我们考虑暴力枚举起点和终点a,c，那么合法的b的就是a到c的所有简单路径的并。
我们搞出圆方树，圆点点权为-1，方点点权为点双大小，那么要求的就是树上圆点a到c的路径和。
所以答案就是圆方树上所有圆点两两之间的路径和。
考虑树形dp解决这个问题。
计算每一个点在多少条路径上即可。
复杂度$O(n+m)$

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,h[N],num=0,dfn[N],low[N],dfnum=0,sz[N<<1],w[N<<1],sumsz;
ll ans=0;
struct edge{
    int to,next;
}data[N<<2];
vector<int>Son[N<<1];
stack<int>qq;
inline void tarjan(int x,int Fa){
    dfn[x]=low[x]=++dfnum;qq.push(x);sz[x]=1;w[x]=-1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;
        if(!dfn[y]){
            tarjan(y,x);low[x]=min(low[x],low[y]);
            if(low[y]<dfn[x]) continue;++m;w[m]=1;
            while(1){
                int z=qq.top();qq.pop();Son[m].push_back(z);sz[m]+=sz[z];w[m]++;
                if(z==y) break;
            }Son[x].push_back(m);sz[x]+=sz[m];
        }else low[x]=min(low[x],dfn[y]);
    }
}
void dfs(int x){
    int tmp=x<=n;
    for(int i=0;i<Son[x].size();++i){
        int y=Son[x][i];dfs(y);ans+=(ll)tmp*sz[y]*w[x];tmp+=sz[y];
    }ans+=(ll)sz[x]*(sumsz-sz[x])*w[x];
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    while(m--){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }m=n;
    for(int i=1;i<=n;++i){
        if(dfn[i]) continue;
        tarjan(i,0);sumsz=sz[i];dfs(i);
    }printf("%lld\n",ans<<1);
    return 0;
}