首先我们考虑暴力的作法,就是将一个点走k步,最后所在的那个点即为所求,考虑数据范围显然不可以,因此我们可以采用倍增的写法去实现。每个点的步数转化成二进制形式去求,用st表。
代码:
#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f;
const int N = 2e5+5;
int n,k;
int a[N],b[N];
int f[N][105];
void solve(){
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>f[i][0];
}
for(int i=1;i<=n;i++){
cin>>a[i];
b[i] = i;
}
for(int j=1;j<=63;j++){
for(int i=1;i<=n;i++){
f[i][j] = f[f[i][j-1]][j-1];
}
}
int t = 0;
while(k){
if(k&1){
for(int i=1;i<=n;i++){
b[i] = f[b[i]][t];
}
}
t++;
k>>=1;
}
for(int i=1;i<=n;i++)cout<<a[b[i]]<<" ";
cout<<"\n";
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int T=1;
while(T--){
solve();
}
return 0;
}