日常吐槽:考场没加long long结果三十分。呵。
题目
100026. 【NOIP2017提高A组模拟7.7】图
Time Limits: 2000 ms Memory Limits: 524288 KB
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)
第二行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<=10^5,k<=10^10,0<=f(i)<n,w(i)<=10^8。
Hint
题解
倍增。
(这种入度为一的有向图什么的不打倍增真是傻瓜)
代码
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
const int N = 100010;
int n, f[N][40], w[N], num, minn[N][40];
unsigned long long sum[N][40], power[40], k;
int main()
{
scanf("%d%lld", &n, &k);
num = log2(k);
memset(sum, 0, sizeof sum);
memset(minn, 0x3f, sizeof minn);
for (int i = 0; i < n; i++) scanf("%d", &f[i][0]);
for (int i = 0; i < n; i++) scanf("%lld", &sum[i][0]), minn[i][0] = sum[i][0];
power[0] = 1;
for (int j = 1; j <= num; j++)
{
power[j] = power[j - 1] * 2;
for (int i = 0; i < n; i++)
{
f[i][j] = f[f[i][j - 1]][j - 1];
sum[i][j] = sum[f[i][j - 1]][j - 1] + sum[i][j - 1];
minn[i][j] = min(minn[f[i][j - 1]][j - 1], minn[i][j - 1]);
}
}
for (int i = 0; i < n; i++)
{
long long s = 0, k1 = k;
int v = (int)1e8;
int u = i;
for (int j = num; j >= 0; j--)
{
if (power[j] <= k1)
{
k1 -= power[j];
s += sum[u][j];
v = min(v, minn[u][j]);
u = f[u][j];
}
}
printf("%lld %d\n", s, v);
}
return 0;
}