树形dp求树直径、两次dfs求树直径

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_41661919/article/details/84947796

树形dp求树直径：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include<algorithm>
#include <set>
#include <queue>
#include <stack>
#include<vector>
#include<map>
#include<ctime>
#define ll long long
using namespace std;
const int N=100010,M=1000010;
int head[N];
int ver[M];
int edge[M];
int Next[M];
bool v[N];
int tot;
void add(int x,int y,int z)
{
    ver[++tot]=y;
    edge[tot]=z;
    Next[tot]=head[x];
    head[x]=tot;
}
ll ans;
int d[N];//以N为根节点在其子树的最长路
void dp(int x)//根节点
{
    v[x]=1;
    for(int i=head[x];i;i=Next[i])
    {
        int y=ver[i];
        if(v[y])continue;
        dp(y);
        if(d[x]+d[y]+edge[i]>ans)ans=d[x]+d[y]+edge[i];//这两个if的顺序不要弄错
        if(d[y]+edge[i]>d[x])d[x]=d[y]+edge[i];//
    }
}
int main()
{
    int n;//节点数
    int m;//边数
    while(cin>>n>>m)
    {
        ans=-9999999999;
        for(int i=1;i<=m;++i)
        {
            int x,y,z;
            cin>>x>>y>>z;
            add(x,y,z);
            add(y,x,z);
        }
        dp(1);
        cout<<ans<<endl;
    }
}

两次dfs求树直径：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include<algorithm>
#include <set>
#include <queue>
#include <stack>
#include<vector>
#include<map>
#include<ctime>
#define ll long long
using namespace std;
const int N=100010,M=1000010;
int head[N];
int ver[M];
int edge[M];
int Next[M];
int tot;
priority_queue< pair<int,int> >Q;
void add(int x,int y,int z)
{
    ver[++tot]=y;
    edge[tot]=z;
    Next[tot]=head[x];
    head[x]=tot;
}

ll ans;//记录最深的点的节点标号；
int d[N];//以N为根节点的子树的深度；
bool v[N];//无向图防止子节点回到父节点
void dfs(int x)
{
    v[x]=1;
    for(int i=head[x];i;i=Next[i])
    {
        int y=ver[i];
        if(v[y])continue;
        d[y]=d[x]+1;//从上往下加一即可；
        if(d[y]>d[ans])ans=y;
        dfs(y);
    }
}
int main()
{
    int n;//节点数
    int m;//边数
    while(cin>>n>>m)
    {
        ans=0;
        d[0]=0;
        for(int i=1;i<=m;++i)
        {
            int x,y,z;
            cin>>x>>y>>z;
            add(x,y,z);
            add(y,x,z);
        }
        dfs(1);
        memset(d,0,sizeof(d));//两次dfs注意清空搜索标记数组、子树深度数组d；
        memset(v,0,sizeof(v));
        //cout<<ans<<endl;
        dfs(ans);
        cout<<d[ans]<<endl;//最后答案存储在d中，答案点标号为ans；
    }
    return 0;
}