abc377_E

E - Permute K times 2

思路

這題由於序列P是一個排列，所以將P表示成一個圖的時候，這個圖將由\(m\)個環構成

對於每個環上的點來說，第一回合它會移動到距離它為\(2\)的點上，距離它為\(2\)的點同時也以相同的方式移動，那麼第二回合，它就會移動到距離它為\(4\)的點上，得出規律，一個點移動\(k\)回合會移動到距離它為\(2^k\)的點上，由於是在一個環上移動，所以直接取模環的長度即可

程式碼

這裡直接使用jiangly的程式碼了，很簡潔優美

//來自jiangly
#include <bits/stdc++.h>

using i64 = long long;
using u64 = unsigned long long;
using u32 = unsigned;

int power(int a, i64 b, int p) {
    int res = 1;
    for (; b; b /= 2, a = 1LL * a * a % p) {
        if (b & 1) {
            res = 1LL * res * a % p;
        }
    }
    return res;
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
    int N;
    i64 K;
    std::cin >> N >> K;
    
    std::vector<int> P(N);
    for (int i = 0; i < N; i++) {
        std::cin >> P[i];
        P[i]--;
    }
    
    std::vector<bool> vis(N);
    for (int i = 0; i < N; i++) {
        if (vis[i]) {
            continue;
        }
        
        int j = i;
        std::vector<int> a;
        while (!vis[j]) {
            vis[j] = true;
            a.push_back(j);
            j = P[j];
        }
        
        i64 d = power(2, K, a.size());
        for (int x = 0; x < a.size(); x++) {
            P[a[x]] = a[(x + d) % a.size()];
        }
    }
    for (int i = 0; i < N; i++) {
        std::cout << P[i] + 1 << " \n"[i == N - 1];
    }
    
    return 0;
}