Envío 5659

Código

#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; i < n; i++)
#define pb push_back
#define ff first
#define ss second
#define ii pair<int, int>
#define vi vector<int>
#define vii vector<ii>
#define lli long long int
#define fast_io                                                                \
  ios_base::sync_with_stdio(0);                                                \
  cin.tie(0);                                                                  \
  cout.tie(0);
using namespace std;
ostream &operator<<(ostream &os, const vector<int> &v) {
  for (auto const &i : v) {
    os << i << " ";
  }
  os << endl;
  return os;
}
const lli mod = (1e9) + 7;
const int N = 1e5 + 5;
const lli LOG2 = 62;

int jump[N][LOG2];

int go(int u, lli R, vector<vector<lli>> &jump){
    int ans = u;
    for(int i = 0;R;R >>= 1, i++){
        if(R & 1){
            ans = jump[ans][i];
        }
    }
    return ans;
}

int main() {
  fast_io;
  lli n, r;
  cin >> n >> r;
  lli v;
  vector<vector<lli>> jump(n, vector<lli>(LOG2, -1));
  REP(u, n) {
    cin >> v;
    jump[u][0] = v;
  }
  for (lli j = 1; j < LOG2; j++) {
    for (lli u = 0; u < n; u++) {
      jump[u][j] = jump[jump[u][j - 1]][j - 1];
    }
  }
  vi ans(n);
  REP(i, n) {
    lli pos = go(i, r, jump);
    ans[pos] = i + 1;
  }
  cout << ans << endl;
  return 0;
}