Description
有一个n个点n条边的有向图,每条边为<i,f(i),w(i)>,意思是i指向f(i)的边权为w(i)的边,现在小A想知道,对于每个点的si和mi。
si:由i出发经过k条边,这k条边的权值和。
mi:由i出发经过k条边,这k条边的权值最小值。
Input
第一行两个数n和k
第二行n个数f(i)
第三行n个数w(i)
Output
每行两个数si和mi
Sample Input
7 3
1 2 3 4 3 2 6
6 3 1 4 2 2 3
Sample Output
10 1
8 1
7 1
10 2
8 2
7 1
9 3
Data Constraint
30%的数据:n,k<=1000。
100%的数据:N<=105,k<=1010,0<=f(i)<n,w(i)<=10^8。
思路
其实这是个RMQ题,只是看不看的出来
我们维护三个值:
F[r][u].to 表示从 u 出发经过 2^r 条边到达的顶点。
F[r][u].len 表示从 u 出发经过 2^r 条边的权值和。
F[r][u].min 表示从 u 出发经过 2^r 条边的最小值。
用RMQ即可
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define ll long long
using namespace std;
const int maxn=1e5+77;
int n;
ll k;
struct F
{
ll to,m,s;
}f[50][maxn];
int main()
{
freopen("data.in","r",stdin); freopen("data.out","w",stdout);
scanf("%d%lld",&n,&k);
for(int i=0; i<=45; i++) for(int j=0; j<=n; j++) f[i][j].m=0x3f3f3f3f3f3f3f3f;
for(int i=0; i<n; i++) scanf("%d",&f[0][i].to);
for(int i=0; i<n; i++) scanf("%d",&f[0][i].s),f[0][i].m=f[0][i].s;
for(int i=1; i<=log2(k)+2; i++)
{
for(int j=0; j<n; j++) f[i][j].to=f[i-1][f[i-1][j].to].to,f[i][j].s=f[i-1][f[i-1][j].to].s+f[i-1][j].s,f[i][j].m=min(f[i-1][j].m,f[i-1][f[i-1][j].to].m);
}
for(int i=0; i<n; i++)
{
ll m=0x3f3f3f3f3f3f3f3f,s=0,t=k; int x=i;
for(int j=0; t; j++,t>>=1) if(t&1) m=min(m,f[j][x].m),s+=f[j][x].s,x=f[j][x].to;
printf("%lld %lld\n",s,m);
}
}