前言
之前学LCA就学过了 知道了DFS方法
这次为了了解树DP,滚过来复习
思路
DFS :
我们只需要 从根节点开始跑到能跑到的最远点,再从该点枚举到最远点那么最后一次便是我们的直径
树形DP :
我们记录当
1
1
1为树的根时,每个节点作为子树的根向下,所能延伸的最远距离
d
1
d1
d1,和次远距离
d
2
d2
d2,那么直径就是所有
d
1
+
d
2
d1+d2
d1+d2的最大值。
CODE
DFS
#include <bits/stdc++.h>
using namespace std;
const int N = 10000 + 10;
int n, c, d[N];
int h[N],w[N*2],e[N*2],ne[N*2],idx;
void add(int a,int b,int c)
{
e[idx] = b,ne[idx] = h[a],w[idx] =c,h[a]=idx++;
}
void dfs(int u, int fa)
{
for(int i = h[u];~i;i=ne[i])
{
int v = e[i];
if (v == fa) continue;
d[v] = d[u] + w[i];
if (d[v] > d[c]) c = v;
dfs(v, u);
}
}
void solve()
{
cin>>n;
memset(h,-1,sizeof h);
for(int i=1; i<n; i++)
{
int u,v,c;
cin>>u>>v>>c;
add(u,v,c);
add(v,u,c);
}
dfs(1,0);
d[c] = 0 ;
dfs(c,0);
cout<<d[c]<<endl;
return;
}
int main()
{
ios::sync_with_stdio(false);
solve();
return 0;
}
树形DP :
#include <bits/stdc++.h>
using namespace std;
const int N = 10000 + 10;
int n, c, d1[N],d2[N];
int h[N],w[N*2],e[N*2],ne[N*2],idx;
int d;
void add(int a,int b,int c)
{
e[idx] = b,ne[idx] = h[a],w[idx] =c,h[a]=idx++;
}
void dfs(int u, int fa)
{
d1[u] = d2[u] = 0 ;
for(int i = h[u];~i;i=ne[i])
{
int v = e[i];
if (v == fa) continue;
dfs(v, u);
int t = d1[v] + w[i];
if(t>d1[u])
d2[u] =d1[u],d1[u] = t;
else if(t>d2[u])
d2[u] = t;
//if (d[v] > d[c]) c = v;
}
d =max(d,d1[u]+d2[u]);
}
void solve()
{
cin>>n;
memset(h,-1,sizeof h);
for(int i=1; i<n; i++)
{
int u,v,c;
cin>>u>>v>>c;
add(u,v,c);
add(v,u,c);
}
dfs(1,0);
cout<<d<<endl;
return;
}
int main()
{
ios::sync_with_stdio(false);
solve();
return 0;
}