結果

問題 No.2732 Similar Permutations
ユーザー eve__fuyukieve__fuyuki
提出日時 2024-04-19 22:25:38
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 2,360 bytes
コンパイル時間 2,475 ms
コンパイル使用メモリ 213,576 KB
実行使用メモリ 5,760 KB
最終ジャッジ日時 2024-10-11 15:55:49
合計ジャッジ時間 17,311 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 101
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
void fast_io() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
}

int main() {
    fast_io();
    int n;
    cin >> n;
    vector<int> a(n);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    if (n == 1) {
        cout << -1 << endl;
        return 0;
    }

    if (n <= 10) {
        map<int, vector<int>> mp;
        vector<int> p(n);
        iota(p.begin(), p.end(), 1);
        do {
            int x = 0;
            for (int i = 0; i < n; i++) {
                x ^= a[i] + p[i];
            }
            if (mp.find(x) == mp.end()) {
                mp[x] = p;
            } else {
                auto pp = mp[x];
                for (int i = 0; i < n; i++) {
                    cout << pp[i] << " ";
                }
                cout << "\n";
                for (int i = 0; i < n; i++) {
                    cout << p[i] << " ";
                }
                cout << "\n";
                return 0;
            }
        } while (next_permutation(p.begin(), p.end()));
        cout << -1 << endl;
        return 0;
    }
    int s10 = 0;
    for (int i = 10; i < n; i++) {
        s10 ^= a[i] + (i + 1);
    }
    vector<int> p10(10);
    iota(p10.begin(), p10.end(), 1);
    map<int, vector<int>> mp;
    do {
        int x = s10;
        for (int i = 0; i < 10; i++) {
            x ^= a[i] + p10[i];
        }
        if (mp.find(x) == mp.end()) {
            mp[x] = p10;
        } else {
            auto pp = mp[x];
            vector<int> ans1(n);
            vector<int> ans2(n);
            for (int i = 0; i < 10; i++) {
                ans1[i] = pp[i];
                ans2[i] = p10[i];
            }
            for (int i = 10; i < n; i++) {
                ans1[i] = i + 1;
                ans2[i] = i + 1;
            }
            int s1 = 0, s2 = 0;
            for (int i = 0; i < n; i++) {
                s1 ^= a[i] + ans1[i];
                s2 ^= a[i] + ans2[i];
            }
            assert(s1 == s2);
            for (int i = 0; i < n; i++) {
                cout << ans1[i] << " ";
            }
            cout << "\n";
            for (int i = 0; i < n; i++) {
                cout << ans2[i] << " ";
            }
            cout << "\n";
            return 0;
        }
    } while (next_permutation(p10.begin(), p10.end()));
}
0