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

77人阅读 评论(0) 收藏 举报
分类:

我们枚举切掉的那条边,分成了两个子树,分别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;
}
查看评论

4890: [Tjoi2017]城市

这题题面看这里:https://www.luogu.org/problem/show?pid=3761 这里怎么做呢 看了下数据范围。。 感觉n^2能过 那就暴力枚举删去那一条边咯 然后暴力...
  • qq_36797743
  • qq_36797743
  • 2017-07-12 14:52:04
  • 182

bzoj 4890: [Tjoi2017]城市 树形dp

题意从加里敦大学城市规划专业毕业的小明来到了一个地区城市规划局工作。这个地区一共有ri座城市,《-1条高速公路,保证了任意两运城市之间都可以通过高速公路相互可达,但是通过一条高速公路需要收取一定的交通...
  • qq_33229466
  • qq_33229466
  • 2017-12-31 08:50:53
  • 143

树形dp题集之树的直径

【codeforces 592D】 #include using namespace std; #define inf 130000 vectoradj[inf]; int vis[inf]; in...
  • saragrean
  • saragrean
  • 2015-11-06 22:29:44
  • 510

树形DP求树的直径

hdu4607
  • u012859437
  • u012859437
  • 2014-05-10 20:09:59
  • 1169

POJ 1849 Two (树形dp 树的直径 两种方法)

POJ 1849 Two (树形dp 树的直径 两种方法)
  • Tc_To_Top
  • Tc_To_Top
  • 2015-07-22 14:52:05
  • 1438

树形dp求树的重心 poj1655

本来是为了写树分治才写这个题的没想到竟然浪费了好长时间 狂wa不止原来前向星的数组要开两倍。。。第一次用前向星真是光荣牺牲了。。。 dp很简单 先DFS一次,求出以每个节点为根的...
  • u012457935
  • u012457935
  • 2014-03-25 16:33:28
  • 576

树的重心——树形dp

树的重心(质心):对于一颗n个节点的无根树,找到一个点,使得把树变成以该节点为根的有根树时,最大节点数最少。换句话说,删除这个节点后最大连通块(一定是树)的节点数最少。 PS:首先要知道什么是树的重...
  • qq_33850438
  • qq_33850438
  • 2016-02-22 02:37:26
  • 623

#bzoj2404#最长链(树的直径/树形DP)

2404: 最长链 时间限制: 1 Sec  内存限制: 128 MB 题目描述 给定一棵有n个节点的树,求每个节点到其他节点的最大距离 输入 输入第一行是一个自然数n(n≤10000), 接下来 (...
  • its_elaine
  • its_elaine
  • 2017-04-05 21:06:45
  • 455

[树的直径 树形DP] UOJ #11【UTR #1】ydc的大树

题解链接大意是我们求出直径的中心 那么到最远点必然经过中心 把它提到根 就可以在枚举删的白点时很快统计出能够使多少个黑点不高兴#include #include #include using na...
  • u014609452
  • u014609452
  • 2017-04-05 21:05:17
  • 332
    个人资料
    持之以恒
    等级:
    访问量: 9万+
    积分: 1万+
    排名: 1934
    文章分类
    文章存档
    最新评论