hdu 2196 computer 求樹上的任意最遠點對 O（n）

標籤：hdu

題意：

給定n個結點，他們之間用n-1條邊連結（這一點說明這個圖的形狀 就是一棵樹 無環），給你一個結點，距離此節點最遠的點與這個節點之間的距離。

解題思路：

經典的樹上最長點對問題。不過帶權，但是解決方案沒有區別

首先找任意一個點，dfs()求出距離這個點的最遠點END1 O(n)

然後從END1出發 再次dfs() 求出距離END1的最遠點 期間經過每一個結點時，更新dist[i] （這個狀態表示距離結點i的最遠點到它的距離） 然後求出END2

從END2 再次出發 再遍曆一遍所有結點 再次更新dist[i];

明確一個性質：可以證明 對於任意一個節點 距離它最遠的結點 一定是END1 或 END2 。

code：

#include<cstdio>#include<cstring>#include<cstring>#include<vector>#include<algorithm>using namespace std;//#pragma comment(linker, "/STACK:102400000,102400000")const int maxn = 10005;vector<int> g[maxn];vector<int> cost[maxn];int dist[maxn];int END,max_;int n;void init(){    for(int i = 0; i <= maxn; i++){        g[i].clear();        cost[i].clear();    }    for(int i = 2; i <= n; i++){        int u,w;        scanf("%d%d",&u, &w);        g[i].push_back(u);        cost[i].push_back(w);        g[u].push_back(i);        cost[u].push_back(w);    }}void dfs(int u, int fa, int len){    dist[u] = max(dist[u], len);    for(int i = 0; i < g[u].size(); i++){        int v = g[u][i];        int w = cost[u][i];        if(v == fa) continue;        dfs(v, u, len + w);    }    if(len >= max_){        END = u;        max_ = len;    }}void solve(){    memset(dist, 0, sizeof(dist));    max_ = 0;    dfs(1, -1, 0);    max_ = 0;    dfs(END, -1, 0);    max_ = 0;    dfs(END, -1, 0);    for(int i = 1; i <= n; i++){        printf("%d\n",dist[i]);    }}int main(){    while(scanf("%d",&n) != EOF){        init();        solve();    }    return 0;}

一開始鬆弛操作的部分，忘記每次更新max_值。。好在很快發現 水...

後來交上題目，返回RE....簡直無情 

臥槽我居然傻逼手動擴棧。。。擴棧。。。棧。。簡直無情 當然是毫無懸念的MT了...

實際上是我沒有每次清空vector 。 忘了愛

hdu 2196 computer 求樹上的任意最遠點對 O（n）