Submission #56827648

---

Source Code Expand

// I AM A MUSLIM

#include "bits/stdc++.h"

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")

#define fastIO std::ios::sync_with_stdio(0);std::cin.tie(0)
#define ll long long int
#define flush fflush(stdout)
#define bl printf("\n")
#define yn(a, b) printf("%s\n", a >= b ? "Yes":"No")
// #define int ll

using pii = std::pair<int,int>;

const int MOD = 1000000007;
// const int MOD = 998244353;
const int mxN = 200100;

int N;
// std::vector<int> g[mxN];
int X[mxN], a[mxN];
const int LIM = 60;
int par[mxN][LIM+5];

int kth_par(int n, ll k) {
    int at = n;
    for (int pow = 0; pow <= LIM; pow++) {
        if ((k & (1ll << pow)) != 0) {
            at = par[at][pow];
            if (at == -1) {
                break;
            }
        }
    }
    return at;
}

signed main() {
    // fastIO;
    int testCases=1;
    // scanf("%d",&testCases);
    // std::cin>>testCases;
    
    for (int TC = 1; TC <= testCases; TC++) {
        ll k;
        scanf("%d%lld",&N,&k);
        for (int i = 1; i <= N; i++) scanf("%d",&X[i]);
        for (int i = 1; i <= N; i++) scanf("%d",&a[i]);
        
        if (k == 0) {
            for (int i = 1; i <= N; i++) printf("%d ",a[i]); bl;
            continue;
        }
        
        memset(par, -1, sizeof par);
        for (int i = 1; i <= N; i++) {
            par[i][0] = X[i];
        }
        
        for (int j = 1; j <= LIM; j++) {
            for (int i = 1; i <= N; i++) {
                int p = par[i][j-1];
                if (p != -1) {
                    par[i][j] = par[p][j-1];
                }
            }
        }
        
        for (int i = 1; i <= N; i++) {
            int at = kth_par(i,k);
            assert(at != -1);
            printf("%d ",a[at]);
        } bl;
        
    }
    
    return 0;
}

/*

*/

Submission Info

Compile Error

./Main.cpp:5:13: warning: unknown pragma ignored [-Wunknown-pragmas]
#pragma GCC optimize("O3,unroll-loops")
            ^
./Main.cpp:6:13: warning: unknown pragma ignored [-Wunknown-pragmas]
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
            ^
./Main.cpp:17:11: warning: unused variable 'MOD' [-Wunused-const-variable]
const int MOD = 1000000007;
          ^
3 warnings generated.

Judge Result