HDU 2196 -Computer【树形DP】（树的直径+动态规划）（dfs）（bfs）

一所学校不久前买了第一台电脑（所以这台电脑的ID是1）。近年来，学校购买了N-1新电脑。每台新电脑都连接到一台先前安装的电脑上。学校的管理人员担心网络运行缓慢，希望知道第i台计算机需要发送信号的最大距离si（即到最远计算机的电缆长度）。您需要提供此信息。

提示：示例输入与此图对应。从图中，你可以看到计算机4离1最远，所以s1=3。计算机4和5是距离2最远的，所以s2=2。计算机5是离3最远的，所以s3=3。我们也得到了s4=4，s5=4。

输入

输入文件包含多组测试样例。在每组样例中，第一行中都有自然数n（n<=10000），然后是（n-1）行，其中包含对计算机的描述。第i行包含两个自然数-第i计算机所连接的计算机数量和用于连接的电缆长度。电缆总长度不超过10^9。输入行中的数字用空格分隔。

输出
对于每组样例，输出n行。第i行必须包含第i台计算机的数字Si（1<=i<=n）。

思路：
怎么说呢 这道题我前前后后wa了8次，本以为是一道简单的树的直径题，…由于数据范围太大，所以直接找每个点到末端的最长长度会超时；仔细读题会发现，实际上就是求每个点到两端的距离，即为最大值；所以我们第一次dfs找到一个端点，再dfs找到另外一个端点即可；在进行一次dfs更新一下dp；
贴代码：

AC代码：

DFS

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
const int MAX=1e9+7;
int vist[10020];
int dp[10020];
vector<pair<int,int> > v[10020];
int m,n,ans,point;
void dfs(int x,int fx,int sum)
{
    if(ans<=sum)
    {
        ans=sum;
        point=x;
    }
    pair<int,int> t;
    for(int i=0;i<v[x].size();i++)
    {
        t=v[x][i];
        if(t.first==fx) continue;
        dfs(t.first,x,sum+t.second);
        dp[t.first]=max(dp[t.first],sum+t.second);
    }
}
int main()
{
    int n;
    int x,y,z;
    while(~scanf("%d",&n)&&n)
    {
        for(int i=1;i<n;i++)
        {
            scanf("%d %d",&x,&z);
            y=i+1;
            v[x].push_back(make_pair(y,z));//我采用邻接表来储存点和权值；
            v[y].push_back(make_pair(x,z));
        }
        memset(dp,0,sizeof dp);
        ans=0,point=0;
        dfs(1,-1,0);  //第一次从1开始，找到距离最远的点；
        dfs(point,-1,0);//第二次从上一次找到的端点开始dfs，目的找到每台电脑离这个端点的距离最大值；
        dfs(point,-1,0);//上一步又找到了这条线的另一端，这次从这个端点开始dfs，目的为找到每台电脑离这个端点的最大值
        for(int i = 1; i <= n; ++i)
        {
            printf("%d\n", dp[i]);
        }
        for(int i=0;i<=n;i++)
            v[i].clear();
    }
    return 0;
}

BFS：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
const int MAX=1e9+7;
int dis[10020];
int vist[10020];
int dp[10020];
vector<pair<int,int> > v[10020];
int ans;
int bfs(int x)
{
    memset(vist,0,sizeof vist);
    memset(dis,0,sizeof dis);
    vist[x]=1;
    queue<int> q;
    q.push(x);
    int point=0;
    ans=0;
    int F=x;
    while(!q.empty())
    {

        F=q.front();
        q.pop();
        if(dis[F]>=ans)
        {
            ans=dis[F];
            point=F;
        }
        pair<int,int> t;
        for(int i=0;i<v[F].size();i++)
        {
            t=v[F][i];
            if(!vist[t.first])
            {
                vist[t.first]=1;
                dis[t.first]=t.second+dis[F];
                dp[t.first]=max(dp[t.first],dis[t.first]);
                q.push(t.first);
            }
        }
    }
    return point;
}
int main()
{
    int n;
    int x,y,z;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<n;i++)
        {
            scanf("%d %d",&x,&z);
            y=i+1;
            v[x].push_back(make_pair(y,z));
            v[y].push_back(make_pair(x,z));
        }
        memset(dp,0,sizeof dp);
        int p=bfs(1);
        int pp=bfs(p);
        bfs(pp);
        for(int i=1;i<=n;i++)
            printf("%d\n",dp[i]);
        for(int i=0;i<=n;i++)
            v[i].clear();
    }
    return 0;
}