题目链接:点击这里
题目大意:
求一棵树中所有节点能达到的最长距离
题目分析:
一个节点能到达的最长距离有两种情况,第一种是到其子树中的点中的最大的距离(记为正向最长路),另一种是其父亲节点的不经过该点的最长距离+该点到其父亲节点的距离(记为反向最长路)
设
d
p
[
u
]
[
0
]
dp[u][0]
dp[u][0] 和
d
p
[
u
]
[
1
]
dp[u][1]
dp[u][1] 分别表示
u
u
u 到
u
u
u 的子树中的最大距离和次大距离,即
u
u
u 的正向最长路和正向次长路,
d
p
[
u
]
[
2
]
dp[u][2]
dp[u][2] 表示其父亲节点的不经过该点的最长距离+该点到其父亲节点的距离,即反向最长路
d
p
[
u
]
[
0
]
dp[u][0]
dp[u][0] 和
d
p
[
u
]
[
1
]
dp[u][1]
dp[u][1] 可以通过简单的树形
d
p
dp
dp 处理出来(代码中的
d
f
s
1
dfs1
dfs1)
记
u
u
u 的父亲节点为
f
a
fa
fa
d
p
[
u
]
[
2
]
dp[u][2]
dp[u][2] 的求解过程有两种可能:
第一种是
u
u
u 不是
f
a
fa
fa 正向最长路上的点,此时
d
p
[
u
]
[
2
]
=
m
a
x
(
d
p
[
f
a
]
[
0
]
,
d
p
[
f
a
]
[
2
]
)
+
d
i
s
(
u
,
v
)
dp[u][2]=max(dp[fa][0],dp[fa][2])+dis(u,v)
dp[u][2]=max(dp[fa][0],dp[fa][2])+dis(u,v)
第二种是
u
u
u 是
f
a
fa
fa 正向最长路上的点,此时
d
p
[
u
]
[
2
]
=
m
a
x
(
d
p
[
f
a
]
[
1
]
,
d
p
[
f
a
]
[
2
]
)
+
d
i
s
(
u
,
v
)
dp[u][2]=max(dp[fa][1],dp[fa][2])+dis(u,v)
dp[u][2]=max(dp[fa][1],dp[fa][2])+dis(u,v)
此过程可通过一次自上而下的dfs递推出来(代码中的
d
f
s
2
dfs2
dfs2)
具体细节见代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
int read()
{
int res = 0,flag = 1;
char ch = getchar();
while(ch<'0' || ch>'9')
{
if(ch == '-') flag = -1;
ch = getchar();
}
while(ch>='0' && ch<='9')
{
res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
ch = getchar();
}
return res*flag;
}
const int maxn = 1e5+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
vector<pair<int,int>> v[maxn];
int dp[maxn][3],son[maxn]; //son[u]表示u的正向最长路要经过的儿子
void dfs1(int u,int fa) //处理dp[u][0],dp[u][1]
{
dp[u][0] = dp[u][1] = dp[u][2] = 0;
for(int i = 0;i < v[u].size();i++)
{
int to = v[u][i].first,dis = v[u][i].second;
if(to == fa) continue;
dfs1(to,u);
if(dp[u][0] < dp[to][0]+dis) //如果发现更优的最大值则更新
{
dp[u][1] = max(dp[u][1],dp[u][0]);
dp[u][0] = dp[to][0]+dis;
son[u] = to; //u的正向最大距离要经过儿子to
}
else dp[u][1] = max(dp[u][1],dp[to][0]+dis); //否则更新次大值
}
}
void dfs2(int u,int fa) //处理得到dp[u][2]
{
for(int i = 0;i < v[u].size();i++)
{
int to = v[u][i].first,dis = v[u][i].second;
if(to == fa) continue;
if(son[u] == to) dp[to][2] = max(dp[u][1],dp[u][2])+dis;
else dp[to][2] = max(dp[u][0],dp[u][2])+dis;
dfs2(to,u);
}
}
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i = 1;i <= n;i++)
v[i].clear();
for(int i = 2;i <= n;i++)
{
int to = read(),dis = read();
v[i].push_back(make_pair(to,dis));
v[to].push_back(make_pair(i,dis));
}
dfs1(1,1);
dfs2(1,1);
for(int i = 1;i <= n;i++)
printf("%d\n",max(dp[i][0],dp[i][2]));
}
return 0;
}
2021/8/21 update代码风格:
//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
#define int ll
#define endl '\n'
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
using namespace std;
int read()
{
int res = 0,flag = 1;
char ch = getchar();
while(ch<'0' || ch>'9')
{
if(ch == '-') flag = -1;
ch = getchar();
}
while(ch>='0' && ch<='9')
{
res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
ch = getchar();
}
return res*flag;
}
const int maxn = 1e4+5;
const int mod = 998244353;
const double pi = acos(-1);
const double eps = 1e-8;
int n,son[maxn],dp[maxn][3];
vector<pair<int,int>>g[maxn];
void dfs1(int u,int fa)
{
dp[u][0] = dp[u][1] = dp[u][2] = son[u] = 0;
for(auto p:g[u])
{
int to = p.first,val = p.second;
if(to == fa) continue;
dfs1(to,u);
if(dp[u][0] < dp[to][0]+val)
{
dp[u][1] = dp[u][0];
dp[u][0] = dp[to][0]+val;
son[u] = to;
}
else dp[u][1] = max(dp[u][1],dp[to][0]+val);
}
}
void dfs2(int u,int fa)
{
for(auto p:g[u])
{
int to = p.first,val = p.second;
if(to == fa) continue;
if(son[u] == to) dp[to][2] = max(dp[u][1],dp[u][2])+val;
else dp[to][2] = max(dp[u][0],dp[u][2])+val;
dfs2(to,u);
}
}
signed main()
{
while(~scanf("%d",&n))
{
for(int i = 1;i <= n;i++)
g[i].clear();
for(int i = 2;i <= n;i++)
{
int to = read(),dis = read();
g[i].push_back(make_pair(to,dis));
g[to].push_back(make_pair(i,dis));
}
dfs1(1,1);
dfs2(1,1);
for(int i = 1;i <= n;i++) printf("%lld\n",max(dp[i][0],dp[i][2]));
}
return 0;
}