Icefox的博客

生活不止眼前的苟且,还有诗和远方。

bzoj4890 [Tjoi2017]城市(树的直径+枚举+树形dp)

我们枚举切掉的那条边,分成了两个子树,分别dfs一遍求出直径,再dfs一遍求出中心(即距离最远点最近的点),那么把这两棵子树连起来之后的直径最小就是max(d1,d2,r1+r2+w)。
复杂度O(n2)

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 5010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,h[N],num=1,mx1[N],mx1id[N],mx2[N],mx2id[N],ans=inf,res,r;
struct edge{
    int to,next,val;
}data[N<<1];
inline void dfs1(int x,int Fa){//求直径
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;
        dfs1(y,x);int w=mx1[y]+data[i].val;
        if(w>mx1[x]) mx2[x]=mx1[x],mx2id[x]=mx1id[x],mx1[x]=w,mx1id[x]=y;
        else if(w>mx2[x]) mx2[x]=w,mx2id[x]=y;
    }res=max(res,mx1[x]+mx2[x]);
}
inline void dfs2(int x,int Fa){//求中心
    r=min(r,mx1[x]);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;
        int w=0;if(mx1id[x]==y) w=mx2[x]+data[i].val;
        else w=mx1[x]+data[i].val;
        if(w>mx1[y]) mx2[y]=mx1[y],mx2id[y]=mx1id[y],mx1[y]=w,mx1id[y]=x;
        else if(w>mx2[y]) mx2[y]=w,mx2id[y]=x;dfs2(y,x);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read(),val=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=val;
    }for(int i=1;i<n;++i){
        int x=data[i<<1].to,y=data[i<<1|1].to;memset(mx1,0,sizeof(mx1));
        memset(mx2,0,sizeof(mx2));res=0;dfs1(x,y);dfs1(y,x);r=inf;dfs2(x,y);
        int r1=r;r=inf;dfs2(y,x);res=max(res,r1+r+data[i<<1].val);
        ans=min(ans,res);
    }printf("%d\n",ans);
    return 0;
}
阅读更多
版权声明:转载附上原文地址即可~欢迎各位神犇来虐~ https://blog.csdn.net/Icefox_zhx/article/details/79981603
上一篇bzoj4889 [Tjoi2017]不勤劳的图书管理员(线段树套权值线段树)
下一篇bzoj4888 [Tjoi2017]异或和(按位+树状数组)
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭