Envío 5155

Código

/// Write by Daniel Perez .PERAPRO
#include<bits/stdc++.h>

using namespace std;
#define fast_io ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define pb push_back
#define ff first
#define ss second
#define all(s) s.begin(), s.end()
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
using vi=vector<int>;
using vl=vector<ll>;
using pii=pair<int,int>;
char el = '\n';
char esp = ' ';

template<class T> bool ckmin(T& a, const T& b) { return b < a ? a = b, 1 : 0; }
template<class T> bool ckmax(T& a, const T& b) { return a < b ? a = b, 1 : 0; }

ostream& operator<<(ostream& os, const vector<int> &v){
    for(auto const &i: v){
        os<<i<<" ";
    }
    os<<'\n';
    return os;
}
string yes="YES";
string no="NO";

const int Log2 = 62;
const int N = 1e5+5;

int lca[N][Log2];

int solve(int idx, ll R){
    int ans = idx;
    for(int i = 0;R;R >>= 1, i++){
        if(R & 1){
            ans = lca[ans][i];
        }
    }
    return ans;
}

int main(){
    fast_io;
    /*
     freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    */
    ll n, R, child;

    cin>>n>>R;

    vi v(n);
    iota(all(v), 1);

    vi p(n);

    for(int i=0;i<n;i++){
        cin>>child;
        lca[i][0] = child;
    }

    for(int j=0;j + 1 < Log2;j++){
        for(int i=0;i<n;i++){
            lca[i][j + 1] = lca[ lca[i][j] ][j];
        }
    }

    vi ans(n);

    for(int i=0;i<n;i++){
        ans[solve(i, R)] = v[i];
    }
    for(int e:ans) cout<<e<<esp;
}

/*

*/