BZOJ5152 通道

Link

Description

给你3棵带权树，找两个点使得它们在3棵树中最短路径上权值之和最大。

$n\le 10^5,w\le 10^{12}$

Solution

正解的边分+虚树+某神奇dp表示根本看不懂。。。

LCA开场28min过这题，太神啦Orz

LCA的做法就是找50~100个点，然后每个点dfs找最远点（3棵树中的）

更新答案之后将当前点换成最远点。

这么迭代个10次左右，就非常靠谱了，卡卡时就过了。

然而UOJ最后一组数据还是非常强，至今不知道构造原理。。。

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstring>
#define LL long long
using namespace std;
inline int read(){
    int x=0,f=1;char ch=' ';
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return f==1?x:-x;
}
const int N=2e5+5;
int n;LL ans;
double time1;
struct graph{
    int tot,head[N],to[N],Next[N];LL val[N],d[N];
    graph(){
        tot=0;
        memset(head,0,sizeof head);
        memset(to,0,sizeof to);
        memset(Next,0,sizeof Next);
        memset(val,0,sizeof val);
        memset(d,0,sizeof d);
    }
    inline void addedge(int x,int y,LL v){
        to[++tot]=y;
        Next[tot]=head[x];
        head[x]=tot;
        val[tot]=v;
    }
}g[3];
inline void dfs(int x,int fa,int id){
    for(int i=g[id].head[x];i;i=g[id].Next[i]){
        int u=g[id].to[i];
        if(u==fa)continue;
        g[id].d[u]=g[id].d[x]+g[id].val[i];
        dfs(u,x,id);
    }
}
inline void solve1(){
    for(int i=1;i<=n;++i){
        g[0].d[i]=g[1].d[i]=g[2].d[i]=0;
        dfs(i,0,0);
        dfs(i,0,1);
        dfs(i,0,2);
        for(int j=1;j<=n;++j){
            ans=max(ans,g[0].d[j]+g[1].d[j]+g[2].d[j]);
        }
    }
}
int vis[N];
inline void solve2(){
    for(int s=1;s<=50;++s){
        int now=rand()%n+1;
        while(vis[now])now=rand()%n+1;
        for(int i=1;i<=10;++i){
            if(vis[now])break;
            vis[now]=1;
            g[0].d[now]=g[1].d[now]=g[2].d[now]=0;
            dfs(now,0,0);
            dfs(now,0,1);
            dfs(now,0,2);
            int mx=0;LL num=0;
            for(int j=1;j<=n;++j){
                if(g[0].d[j]+g[1].d[j]+g[2].d[j]>num){
                    num=g[0].d[j]+g[1].d[j]+g[2].d[j];
                    mx=j;
                }
            }
            now=mx;
            ans=max(ans,num);
        }
    }
}
int main(){
    n=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        LL v;scanf("%lld",&v);
        g[0].addedge(x,y,v);
        g[0].addedge(y,x,v);
    }
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        LL v;scanf("%lld",&v);
        g[1].addedge(x,y,v);
        g[1].addedge(y,x,v);
    }
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        LL v;scanf("%lld",&v);
        g[2].addedge(x,y,v);
        g[2].addedge(y,x,v);
    }
    if(n<=3000)solve1();
    else solve2();
    printf("%lld\n",ans);
    return 0;
}