codeforces 702E Analysis of Pathes in Functional Graph 倍增

题目链接

给一个图， 然后给出每条边的权值和一个k值。 让你求出从每个点出发， 走k次能获得的边权的和以及边权的最小值。

用倍增的思想， 求出每个点走一次能到达的点， 权值和以及最小值， 走两次..四次..八次。 这个很容易计算。然后枚举一下所有点就可以了。

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <complex>
#include <cmath>
#include <map>
#include <set>
#include <string>
#include <queue>
#include <stack>
#include <bitset>
using namespace std;
#define pb(x) push_back(x)
#define ll long long
#define mk(x, y) make_pair(x, y)
#define lson l, m, rt<<1
#define mem(a) memset(a, 0, sizeof(a))
#define rson m+1, r, rt<<1|1
#define mem1(a) memset(a, -1, sizeof(a))
#define mem2(a) memset(a, 0x3f, sizeof(a))
#define rep(i, n, a) for(int i = a; i<n; i++)
#define fi first
#define se second
typedef complex <double> cmx;
typedef pair<int, int> pll;
const double PI = acos(-1.0);
const double eps = 1e-;
const int mod = 1e9+;
const int inf = ;
const int dir[][] = { {-, }, {, }, {, -}, {, } };
int nxt[][], mn[][];
ll sum[][];
int main()
{
    int n;
    ll k;
    cin>>n>>k;
    for(int i = ; i < n; i++) {
        scanf("%d", &nxt[i][]);
    }
    for(int i = ; i < n; i++) {
        scanf("%d", &mn[i][]);
        sum[i][] = mn[i][];
    }
    for(int i = ; i < ; i++) {
        for(int j = ; j < n; j++) {
            nxt[j][i] = nxt[nxt[j][i-]][i-];
            mn[j][i] = min(mn[j][i-], mn[nxt[j][i-]][i-]);
            sum[j][i] = sum[j][i-] + sum[nxt[j][i-]][i-];
        }
    }
    for(int j = ; j < n; j++) {
        ll ansSum = , tmpk = k;
        int ansMin = inf, pos = j;
        for(int i = ; i >= ; i--) {
            ll tmp = 1LL<<i;
            if(tmpk >= tmp) {
                tmpk -= tmp;
                ansSum += sum[pos][i];
                ansMin = min(ansMin, mn[pos][i]);
                pos = nxt[pos][i];
            }
        }
        printf("%I64d %d\n", ansSum, ansMin);
    }
    return ;
}