結果

問題 No.366 ロボットソート
コンテスト
ユーザー Tiến Mạnh Trần
提出日時 2026-02-11 01:47:01
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 1,589 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,619 ms
コンパイル使用メモリ 232,748 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2026-02-11 01:47:09
合計ジャッジ時間 7,480 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 7 WA * 14 TLE * 1 -- * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n;
    cin >> n;
    vector<long long> a(n);
    for (int i = 0; i < n; i++) cin >> a[i];
    int k;
    cin >> k;

    vector<long long> b = a;
    sort(b.begin(), b.end());

    long long ans = 0;

    // Xét từng nhóm modulo k
    for (int start = 0; start < k; start++) {
        vector<long long> va, vb;
        vector<int> pos;

        for (int i = start; i < n; i += k) {
            va.push_back(a[i]);
            vb.push_back(b[i]);
            pos.push_back(i);
        }

        // Kiểm tra multiset có giống nhau không
        auto sva = va, svb = vb;
        sort(sva.begin(), sva.end());
        sort(svb.begin(), svb.end());
        if (sva != svb) {
            cout << -1;
            return 0;
        }

        // Map giá trị -> vị trí trong vb
        unordered_map<long long, queue<int>> mp;
        for (int i = 0; i < (int)vb.size(); i++) {
            mp[vb[i]].push(i);
        }

        int m = va.size();
        vector<int> p(m);
        for (int i = 0; i < m; i++) {
            p[i] = mp[va[i]].front();
            mp[va[i]].pop();
        }

        vector<bool> used(m, false);
        for (int i = 0; i < m; i++) {
            if (used[i]) continue;
            int cur = i, len = 0;
            while (!used[cur]) {
                used[cur] = true;
                cur = p[cur];
                len++;
            }
            if (len > 1) ans += len - 1;
        }
    }

    cout << ans;
    return 0;
}
0