码迷,mamicode.com
首页 > 其他好文 > 详细

HihoCoder1050 树中的最长路 树形DP第三题(找不到对象)

时间:2017-11-02 21:59:29      阅读:112      评论:0      收藏:0      [点我收藏+]

标签:targe   找不到   col   iostream   mes   strong   for   oid   ext   

题意:求出的树中距离最远的两个结点之间相隔的距离

水题一道,以前只会用路的直径来解。

代码如下:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<memory>
#include<algorithm>
using namespace std;
const int maxn=200010;
int dis[maxn],Laxt[maxn],Next[maxn],To[maxn];
int cnt,S,Max;
void add(int u,int v)
{
    Next[++cnt]=Laxt[u];
    Laxt[u]=cnt;
    To[cnt]=v;
}
void _dfs(int u)
{
    for(int i=Laxt[u];i;i=Next[i]){
        if(!dis[To[i]]){
            dis[To[i]]=dis[u]+1;
            if(dis[To[i]]>Max){
                Max=dis[To[i]];
                S=To[i];
            }
            _dfs(To[i]);
        }
    }
}
int main()
{
    int n,i,j,u,v;
    scanf("%d",&n);
    for(i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dis[1]=1;S=1;Max=1;
    _dfs(1);
    memset(dis,0,sizeof(dis));
    dis[S]=1;Max=1;
    _dfs(S);
    printf("%d\n",Max-1);
}

 

树形DP:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<memory>
#include<algorithm>
using namespace std;
const int maxn=200010;
int Laxt[maxn],Next[maxn],To[maxn];
int dp[maxn],vis[maxn];
int cnt,ans;
void add(int u,int v)
{
    Next[++cnt]=Laxt[u];
    Laxt[u]=cnt;
    To[cnt]=v;
}
int _dfs(int u)
{
    for(int i=Laxt[u];i;i=Next[i]){
        if(!vis[To[i]]){
            vis[To[i]]=1;
            _dfs(To[i]);
            ans=max(ans,dp[u]+dp[To[i]]+1);
            dp[u]=max(dp[u],dp[To[i]]+1);
        }
    }
    return dp[u];
}
int main()
{
    int n,i,j,u,v;
    scanf("%d",&n);
    for(i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    vis[1]=1;
    _dfs(1);
    printf("%d\n",ans);
    return 0;
}

 

HihoCoder1050 树中的最长路 树形DP第三题(找不到对象)

标签:targe   找不到   col   iostream   mes   strong   for   oid   ext   

原文地址:http://www.cnblogs.com/hua-dong/p/7774647.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!