LOJ.2587.[APIO2018]铁人两项Duathlon(圆方树)

题目链接 LOJ
洛谷P4630

先对这张图建圆方树。
对于S->T这条(些)路径,其对答案的贡献为可能经过的所有点数,那么我们把方点权值设为联通分量的大小,求树上路径权值和就行了。
因为两方点之间的圆点会计算两次,所以圆点权值设为-1就好了。
那么现在有 \(n^2\) 个点对,求每个点对之间的路径上点的权值和。
那对每个点计算一下被计算次数就可以了。这个路径次数计算注意考虑全。。

另外点对是圆点的,所以方点初始sz[]为0,圆点才是1。
方点其实建一条边就可以了。

LOJ为什么找不到代码框的位置了。。交了两次文件CE了两次,是代码不能加注释??→_→现在还没过
zz地调了半上午,做做月考卷冷静一下。先用1h做完英语。

#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
const int N=100005<<1,M=4e5+7;//2N!→_→

int n,m,tot,sk[N],top,dfn[N],low[N],Index,fa[N],sz[N],val[N];
long long Ans;
struct Graph
{
    int H[N],Enum,to[M],nxt[M];
    inline void Add_E(int u,int v){
        to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
    }
    inline void AddEdge(int u,int v){
        Add_E(u,v), Add_E(v,u);
    }
}G,T;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
void Tarjan(int x)
{
    dfn[x]=low[x]=++Index, sk[++top]=x, val[x]=-1;
    for(int v,i=G.H[x]; i; i=G.nxt[i])
        if(!dfn[v=G.to[i]])
        {
            fa[v]=x, Tarjan(v), low[x]=std::min(low[x],low[v]);
            if(dfn[x]<=low[v])
            {
                T.Add_E(x,++tot), val[tot]=1;
                do{
                    T.Add_E(tot,sk[top--]), ++val[tot];
                }while(sk[top+1]!=v);//别在这把x弹掉。。x可能是多个环的根节点。
            }
        }
        else low[x]=std::min(low[x],dfn[v]);
}
void pre_DFS(int x,int f)
{
    if(x<=n) sz[x]=1;
    for(int i=T.H[x]; i; i=T.nxt[i])
        if(T.to[i]!=f) pre_DFS(T.to[i],x), sz[x]+=sz[T.to[i]];
}
void Solve(int x,int f,int tot)
{
    if(x<=n) Ans+=1ll*(tot-1)*val[x];//以x为起点的路径数 
    Ans+=1ll*(tot-sz[x])*sz[x]*val[x];//起点在x到根方向的一侧,终点在另一侧 
    for(int i=T.H[x]; i; i=T.nxt[i])
        Ans+=1ll*val[x]*(tot-sz[T.to[i]])*sz[T.to[i]], Solve(T.to[i],x,tot);//起点在子树方向,终点到其它地方去(包括x)
        //注意刚开始算了个以x为起点的数,直接交换起点终点(*2)是不对的!
}
//void Solve(int x,int f,int tot)
//{
//  if(x<=n) Ans+=2ll*(tot-1)*val[x];//以x为起点&终点的路径数 
//  int sum=tot-sz[x];
//  for(int i=T.H[x]; i; i=T.nxt[i])
//      Ans+=2ll*val[x]*sum*sz[T.to[i]], sum+=sz[T.to[i]], Solve(T.to[i],x,tot);
//}

int main()
{
    tot=/*!*/n=read(),m=read();
    while(m--) G.AddEdge(read(),read());
    for(int i=1; i<=n; ++i)//不一定连通!这个很mmp。
        if(!dfn[i]) Tarjan(i), pre_DFS(i,i), Solve(i,i,sz[i]);
    printf("%lld",Ans);

    return 0;
}

猜你喜欢

转载自www.cnblogs.com/SovietPower/p/9121501.html
今日推荐