dfs求树的直径JAVA_树形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<