P4381 [IOI2008]Island（基环树直径，破环成链）

题意：

给定n个点n条边的带权无向图,保证不存在单独的孤立点
对于每个连通块,你可以从任意一个点开始,向没走过的点走,
连通块的权值为走过的路径的可能最大值,
现在要求计算所有连通块的权值并求和输出.

数据范围：n<=1e5

解法：

题目给定n个点n条边的带权无向图,保证不存在单独的孤立点,
那么给定的图就是基环树森林,

根据题意发现每个连通块的权值就是连通块中的最长链,也就是直径,
那么问题就变为计算基环树森林中每棵基环树的直径.

基环树的直径有两种情况:
1.不经过环
2.经过环
对两种情况分别计算取max即可

对于情况1:
找到环,对环上的点的每棵子树进行树形dp计算最长链,
选用树形dp是因为可以顺便计算这个环上点作为起点的最长链(情况2有用)

对于情况2:
设情况1中点x作为起点的最长链长度为d[x],
那么经过环即为找到两个点x,y,满足d[x]+d[y]+dist(x,y)最大,
其中dist(x,y)表示点x,y在环上的距离
环问题可以先破环成链,然后变成序列上dp
令sum[]为环链序列的距离前缀和,
那么式子变为d[x]+d[y]+sum[y]-sum[x]
移项一下变为(d[x]-sum[x])+(d[y]+sum[y])
发现x和y是独立的
枚举y,在[y-cirlen+1,y-1]范围内取个d[x]-sum[x]的最大值就行了

环只有两个点的情况需要特判,
因为这时候x->v有两条权值不同的边

ps:
参考了别人的代码(基环树题debug实在太恶心了)
找环用的是dfn时间戳法

code：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=1e6+5;
int head[maxm],nt[maxm<<1],to[maxm<<1],w[maxm<<1],cnt;
int mark[maxm];
int n;
//
int cir[maxm],cc;
int dfn[maxm],idx;
int pre[maxm];
ll d[maxm],maxd;
ll sum[maxm];
//
void add(int x,int y,int z){
    cnt++;nt[cnt]=head[x];head[x]=cnt;to[cnt]=y;w[cnt]=z;
}
void dfs1(int x){//找环
    dfn[x]=++idx;
    for(int i=head[x];i;i=nt[i]){
        int v=to[i];
        if(v==pre[x])continue;
        if(!dfn[v]){
            pre[v]=x;
            dfs1(v);
        }else if(dfn[v]>dfn[x]){
            cc=0;
            int t=v;
            while(t!=x){
                cir[++cc]=t;
                t=pre[t];
            }
            cir[++cc]=x;
        }
    }
}
void dfs(int x){//直径不在环上的情况,树形dp求直径
    mark[x]=1;
    d[x]=0;
    for(int i=head[x];i;i=nt[i]){
        int v=to[i];
        if(mark[v])continue;
        dfs(v);
        maxd=max(maxd,d[x]+d[v]+w[i]);
        d[x]=max(d[x],d[v]+w[i]);
    }
}
ll solve(int x){
    idx=0;
    dfs1(x);
    for(int i=1;i<=cc;i++)mark[cir[i]]=1;
    ll ma1=0,ma2=0;
    for(int i=1;i<=cc;i++){//第一种情况
        maxd=0;
        dfs(cir[i]);
        ma1=max(ma1,maxd);
    }
    if(cc==2){//特判环只有两个点的情况
        for(int i=head[cir[1]];i;i=nt[i]){
            int v=to[i];
            if(v==cir[2]){//找到他们之间的边,计算出在环上的情况
                ma2=max(ma2,d[cir[1]]+d[cir[2]]+w[i]);//这里要取max,因为有两条边
            }
        }
        return max(ma1,ma2);
    }
    cir[0]=cir[cc];
    for(int k=1;k<=cc;k++){//计算环链距离
        int x=cir[k];
        for(int i=head[x];i;i=nt[i]){
            int v=to[i];
            if(v==cir[k-1]){
                sum[k]=w[i];
                break;
            }
        }
    }
    for(int i=1;i<=cc;i++)sum[i+cc]=sum[i];
    for(int i=1;i<=cc*2;i++)sum[i]+=sum[i-1];//计算环链距离前缀和
    multiset<ll>s;
    for(int i=1;i<=cc*2;i++){
        if(i>cc){
            s.erase(s.find(d[cir[i-cc]]-sum[i-cc]));
        }
        auto it=s.rbegin();
        if(!s.empty()){
            ma2=max(ma2,d[i>cc?cir[i-cc]:cir[i]]+sum[i]+*it);
        }
        s.insert(d[i>cc?cir[i-cc]:cir[i]]-sum[i]);
    }
    return max(ma1,ma2);
}
signed main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x,L;scanf("%d%d",&x,&L);
        add(x,i,L);
        add(i,x,L);
    }
    ll ans=0;
    for(int i=1;i<=n;i++){//染色,同一基环树的点颜色相同
        if(!mark[i]){
            ans+=solve(i);
        }
    }
    cout<<ans<<endl;
    return 0;
}

---