結果

問題 No.261 ぐるぐるぐるぐる！あみだくじ！
ユーザー maine_honzuki
提出日時 2021-05-05 17:13:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 21 ms / 5,000 ms
コード長 1,663 bytes
コンパイル時間 4,381 ms
コンパイル使用メモリ 257,568 KB
最終ジャッジ日時 2025-01-21 07:33:01
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

//https://ncode.syosetu.com/n4830bu/261/
#include <bits/stdc++.h>
using namespace std;
using ll = long long;

#include "atcoder/all"
using namespace atcoder;

int main() {
    int N, K;
    cin >> N >> K;
    vector<int> P(N);
    iota(P.begin(), P.end(), 0);
    for (int i = 0; i < K; i++) {
        int x, y;
        cin >> x >> y;
        x--;
        y--;
        swap(P[x], P[y]);
    }
    dsu uf(N);
    for (int i = 0; i < N; i++) {
        uf.merge(i, P[i]);
    }
    auto groups = uf.groups();
    int Q;
    cin >> Q;
    while (Q--) {
        vector<ll> R, M;
        vector<int> A(N);
        for (auto&& a : A) {
            cin >> a;
            a--;
        }
        vector<ll> maine(N);
        bool book = true;
        for (auto&& group : groups) {
            int r, m = group.size();
            for (auto&& x : group) {
                maine[x] = x;
            }
            for (r = 0; r < m; r++) {
                bool flag = true;
                for (auto&& x : group) {
                    if (maine[x] != A[x]) {
                        flag = false;
                        break;
                    }
                }
                if (flag)
                    break;
                for (auto&& x : group)
                    maine[x] = P[maine[x]];
            }
            if (r == m) {
                book = false;
                break;
            }
            M.emplace_back(m);
            R.emplace_back(r);
        }
        if (!book) {
            cout << -1 << endl;
            continue;
        }
        auto [y, z] = crt(R, M);
        if (y == 0)
            y += z;
        cout << y << endl;
    }
}
0