bzoj5463:[APIO2018]铁人两项

传送门

考虑建立圆方树,然后将圆点权值定为\(-1\)(考虑到每个圆点都会被两个方点计算进去,要去重),方点权值定为点双内的点数
然后对于每个起点\(s\)和终点\(f\),中间点的选择方案就是路径权值和,然后对于每个点为中间点计算贡献,\(O(n)\)dp就好了
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=2e5+10;
int n,m,dfn[maxn],low[maxn],tmp,st[maxn],top,tot,size[maxn],v[maxn],num;
long long ans;
struct oo
{
    int cnt,pre[maxn*2],nxt[maxn*2],h[maxn];
    void add(int x,int y)
    {
        pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt;
        pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt;
    }
}a,b;
void tarjan(int x,int fa)
{
    dfn[x]=low[x]=++tmp,num++,st[++top]=x,v[x]=-1;
    for(rg int i=a.h[x];i;i=a.nxt[i])
        if(a.pre[i]!=fa)
        {
            if(!dfn[a.pre[i]])
            {
                tarjan(a.pre[i],x);
                low[x]=min(low[x],low[a.pre[i]]);
                if(dfn[x]<=low[a.pre[i]])
                {
                    tot++;int z;
                    do{z=st[top--],b.add(z,tot),v[tot]++;}while(z!=a.pre[i]);
                    b.add(x,tot),v[tot]++;
                }
            }
            else low[x]=min(low[x],dfn[a.pre[i]]);
        }
}
void dp(int x,int fa)
{
    size[x]=x<=n;long long sum=0;
    for(rg int i=b.h[x];i;i=b.nxt[i])
        if(b.pre[i]!=fa)
        {
            dp(b.pre[i],x),sum+=2ll*size[x]*size[b.pre[i]];
            size[x]+=size[b.pre[i]];
        }
    sum+=2ll*(num-size[x])*size[x];
    ans+=sum*v[x];
}
int main()
{
    read(n),read(m);tot=n;
    for(rg int i=1,x,y;i<=m;i++)read(x),read(y),a.add(x,y);
    for(rg int i=1;i<=n;i++)if(!dfn[i])num=0,tarjan(i,0),dp(i,0);
    printf("%lld\n",ans);
}
posted @ 2019-03-28 16:22 蒟蒻--lichenxi 阅读(...) 评论(...) 编辑 收藏