ZOJ 3684 ZJU2013年01月月赛I题 Destroy

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3684

题目意思是：

　　给定一个n个节点的树，边有长度和权值。然后求一个最小值值Power，当把权值小于等于Power的边全部破坏时，所有的叶子节点都和中心点不连通。

　　题目保证树的中心唯一。

首先两次dfs求出树的直径，然后在这条路径上找到中点。

从任意点进行第一次dfs求得数的直径的一个端点，从这个端点dfs求得另一个端点

然后遍历直径，找到最接近直径一半的点就是中点。

然后以中点为根，进行树形DP（dfs）

对于一个点point来说，所需要的最小Power为，Min(Max(所有子节点所需要的Power)，父节点的边的权值)

　　当然，根节点没有父节点，叶节点没有子节点，需要特判。

这么考虑，如果断开父节点和该节点，则该点的所有子节点也都断开连接，自然所有叶节点也断开连接。或者选择不断开，则所有的子节点的叶节点都必须断开。

最近刚刚开始写图论的题，代码还写得非常挫，以后完善了模板，会更新这里的代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
#define PB(x) push_back(x)
struct poi{
    int v,l,p;
    void in(){
        scanf("%d%d",&l,&p);
    }
};
typedef vector<poi> dint;
dint s[10001];
int n,l,d[10010],sum[10010];
void dfs(int u,int f,int len){
    sum[u]=len;
    for(int i=s[u].size()-1;i>=0;i--){
        poi v=s[u][i];
        if(v.v==f)continue;
        dfs(v.v,u,len+v.l);
    }
}
int gdfs(int u){
    dfs(u,-1,0);
    int v=1;
    for(int i=2;i<n;i++)if(sum[i]>sum[v])v=i;
    return v;
}
bool dfs2(int u,int v,int f,int id){
    for(int i=s[u].size()-1;i>=0;i--){
        poi tmp=s[u][i];
        if(tmp.v==f)continue;
        if(tmp.v==v || dfs2(tmp.v,v,u,id+1)){
            if(id+1>l)l=id+1;
            d[id]=tmp.v;
            return true;
        }
    }
    return false;
}
void grod(int u,int v){
    l=1;
    d[0]=u;
    dfs2(u,v,-1,1);
}
inline int jl(int s,int t){return abs(s-(t<<1));}

int dfs3(int u,int f){
    int ans=-1;
    int ans2;
    for(int i=s[u].size()-1;i>=0;i--){
        poi tmp=s[u][i];
        if(tmp.v==f)ans2=tmp.p;
        else ans=max(ans,dfs3(tmp.v,u));
    }
    if(ans==-1)return ans2;
    return min(ans,ans2);
}

int gans(int u){
    int ans=0;
    for(int i=s[u].size()-1;i>=0;i--){
        poi v=s[u][i];
        ans=max(ans,dfs3(v.v,u));
    }
    return ans;
}
int main()
{
    while (~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)s[i].clear();
        int u,v;
        poi tmp;
        for(int i=1;i<n;i++){
            scanf("%d%d",&u,&v);
            tmp.in();
            tmp.v=u;
            s[v].PB(tmp);
            tmp.v=v;
            s[u].PB(tmp);
        }
        u=gdfs(1);
        v=gdfs(u);
        grod(u,v);
        int ans=0,sm=sum[v];
        for(int i=1;i<l;i++){
            if(jl(sm,sum[d[ans]])>jl(sm,sum[d[i]]))
                ans=i;
        }
        ans=d[ans];
        printf("%d\n",gans(ans));
    }
    return 0;
}