poj 3684 Destroy

題目鏈接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4947

題目大意:

給定一棵樹,每條邊有兩種權值(lenth:邊的長度,power:去掉這條邊需要的花費).

有且僅有一個結點是中心:所有結點到該中心的距離中最大的最小.

power=x的炸彈可以去掉所有power小於或等於x的邊.

問使得所有葉子結點與中心不相連的最小x值.

題目思路:

思路很明顯,找出中心,再dfs求出中心到各個葉子結點的路的最小power值中的最大值.

難點在於找中心.

一次dfs,我們可以求出當前結點下的所有結點到當前結點的最遠距離.(更新父節點的最遠距離)

有了第一次dfs的數據,再一次dfs,我們就可以求出所有結點到其他結點的最遠距離了.(更新子節點的最遠距離)

但是再第一次dfs時我們還要保存兩個值,who[]該最遠距離來自哪個子結點,sec[]第二遠距離.

因爲再第二次dfs時,如果更新子節點時,父節點的最遠距離來自該子節點時,用父節點的最遠距離去更新,顯然是錯誤的,應該用第二遠距離去更新.

貌似這也算是一種樹形DP來着...,那就算是我做的第一道樹形DP吧..

代碼:

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

#define ll __int64
#define ls rt<<1
#define rs ls|1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define eps (1e-8)
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define PI (acos(-1.0))
#pragma comment(linker, "/STACK:102400000,102400000")
template <class T> T _max(T x,T y){return x>y? x:y;}
template <class T> T _min(T x,T y){return x<y? x:y;}
template <class T> T _abs(T x){return (x < 0)? -x:x;}
template <class T> T _mod(T x,T y){return (x > 0)? x%y:((x%y)+y)%y;}
template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}
template <class T> void getmax(T& x,T y){x=(y > x)? y:x;}
template <class T> void getmin(T& x,T y){x=(x<0 || y<x)? y:x;}
int TS,cas=1;
const int M=10000+5;
int n;
int fir[M],who[M],sec[M],ret;
struct edge{
	int v,w,p;
	edge(){}
	edge(int _v,int _w,int _p){v=_v,w=_w,p=_p;}
}edg[M<<1];
vector<int>init[M];
bool vis[M];

void update(int u,int v,int len){
	if(fir[u] < len){
		sec[u] = fir[u];
		fir[u] = len;
		who[u] = v;
	}else getmax(sec[u],len);
}

void dfs1(int u){
	vis[u]=1;
	for(int i=0;i<init[u].size();i++){
		edge& e=edg[init[u][i]];
		if(vis[e.v]) continue;
		vis[e.v]=1;
		dfs1(e.v);
		update(u,e.v,fir[e.v]+e.w);
	}
}

void dfs2(int u){
	vis[u]=1;
	for(int i=0;i<init[u].size();i++){
		edge& e=edg[init[u][i]];
		if(vis[e.v]) continue;
		update(e.v,u,((who[u]!=e.v)? fir[u]:sec[u])+e.w);
		dfs2(e.v);
	}
}

void dfs(int u,int power){
	vis[u]=1;
	if(init[u].size() == 1){
		getmax(ret,power);
		return;
	}
	for(int i=0;i<init[u].size();i++){
		edge& e=edg[init[u][i]];
		if(vis[e.v]) continue;
		dfs(e.v,_min(e.p,power));
	}
}

void run(){
	int i,j;
	for(i=1;i<=n;i++)
		init[i].clear();
	for(i=1;i<n;i++){
		int u,v,w,p;
		scanf("%d%d%d%d",&u,&v,&w,&p);
		edg[(i-1)<<1]=edge(v,w,p);
		edg[(i-1)<<1|1]=edge(u,w,p);
		init[u].push_back((i-1)<<1);
		init[v].push_back((i-1)<<1|1);
	}
	clr(fir,0,n+1),clr(who,0,n+1),clr(sec,0,n+1);
	clr(vis,0,n+1),dfs1(1);
	clr(vis,0,n+1),dfs2(1);
	int ct=1;
	for(i=2;i<=n;i++){
		//printf("%d: %d %d %d\n",i,fir[i],who[i],sec[i]);
		if(fir[ct] > fir[i]) ct=i;
	}
	//printf("center = %d\n",ct);
	clr(vis,0,n+1),ret=-INF,dfs(ct,INF);
	printf("%d\n",ret);
}

void preSof(){
}

int main(){
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    preSof();
    //run();
    while((~scanf("%d",&n))) run();
    //for(scanf("%d",&TS);cas<=TS;cas++) run();
    return 0;
}