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

树形dp求树直径:

#include
#include
#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<

 

 

 

两次dfs求树直径:

#include
#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 >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<

 

 

你可能感兴趣的:(模板,树形dp,树的直径与最近公共祖先,dfs,dp)