【题意】给了一个有n条边关系的一个图,然后要求以每个点出发,经过路径长度为k的长度之后,问在这个路径上经过的最大边权以及经过的边权和。
【解题方法】树上路径倍增,rmq在树上的应用。
【AC 代码】
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x7F7F7F7F7F7F
const int maxn=100010;
ll dp1[35][maxn],dp2[35][maxn],fa[35][maxn];
ll n,k;
int main()
{
cin>>n>>k;
for(int i=0; i<n; i++) cin>>fa[0][i];
for(int i=0; i<n; i++){
cin>>dp1[0][i];//和
dp2[0][i]=dp1[0][i];//最大值
}
for(int i=1; 1LL<<i<=k; i++){
for(int j=0; j<n; j++){
dp1[i][j]=dp1[i-1][j]+dp1[i-1][fa[i-1][j]];
dp2[i][j]=min(dp2[i-1][j],dp2[i-1][fa[i-1][j]]);
fa[i][j]=fa[i-1][fa[i-1][j]];
}
}
for(int i=0; i<n; i++)
{
int u=i;
ll sum=0,minn=inf;
for(int j=0; j<35; j++){
if(k>>j&1){
sum+=dp1[j][u];
minn=min(minn,dp2[j][u]);
u=fa[j][u];
}
}
cout<<sum<<" "<<minn<<endl;
}
}