100026. 【NOIP2017提高A组模拟7.7】图
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。
Hint
分析:倍增向上跳。
代码
#include <cstdio>
#define N 100005
#define ll long long
using namespace std;
struct arr
{
int poi,min;
ll sum;
}f[N][40];
ll bin[40],k;
int n;
int min(int x, int y){return x<y?x:y;}
int main()
{
scanf("%d%lld", &n, &k);
for (int i = 0; i < n; i++)
{
int x;
scanf("%d", &x);
f[i][0].poi = x;
}
for (int i = 0; i < n; i++)
{
int x;
scanf("%d", &x);
f[i][0].sum = f[i][0].min = x;
}
for (int j = 1; j <= 35; j++)
for (int i = 0; i < n; i++)
{
f[i][j].poi = f[f[i][j - 1].poi][j - 1].poi;
f[i][j].sum = f[f[i][j - 1].poi][j - 1].sum + f[i][j - 1].sum;
f[i][j].min = min(f[i][j - 1].min, f[f[i][j - 1].poi][j - 1].min);
}
bin[0] = 1;
for (int i = 1; i <= 35; i++) bin[i] = bin[i - 1] * 2;
for (int i = 0; i < n; i++)
{
ll k1 = k;
ll s = 0;
int ans = 1e9;
int p = i;
for (int j = 35; j >= 0; j--)
if (k1 >= bin[j] && k1)
{
k1-= bin[j];
ans = min(ans, f[p][j].min);
s += f[p][j].sum;
p = f[p][j].poi;
}
printf("%lld %d\n", s, ans);
}
}