結果

問題 No.366 ロボットソート
ユーザー 🍡yurahuna🍡yurahuna
提出日時 2016-04-22 10:31:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,944 bytes
コンパイル時間 3,150 ms
コンパイル使用メモリ 180,368 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-29 17:16:02
合計ジャッジ時間 3,108 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define int long long   // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

#define rep(i,n) for (int i=0;i<(n);i++)
#define rep2(i,a,b) for (int i=(a);i<(b);i++)
#define rrep(i,n) for (int i=(n)-1;i>=0;i--)
#define rrep2(i,a,b) for (int i=(b)-1;i>=(a);i--)
#define all(a) (a).begin(),(a).end()

typedef long long ll;
typedef pair<int, int> P;

const int MAX_N = 1000;
const int MAX_K = 1000;
const int MAX_A = 1e9;

/************** assertつけたやつ ***************/

vector<int> input(){
    string line;
    getline(cin, line);
    stringstream ss(line);

    vector<int> in;
    int x;
    while (ss >> x) {
        in.emplace_back(x);
    }

    return in;
}

signed main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);

    auto in = input();
    assert(in.size() == 2);
    int N = in[0], K = in[1];
    assert(1 <= N && N <= MAX_N);
    assert(1 <= K && K <= MAX_K);

    in.clear();
    in = input();
    assert(in.size() == N);

    vector<int> a(N);
    vector<int> b(N);
    set<int> s;
    rep(i, N) {
        a[i] = in[i];
        assert(1 <= a[i] && a[i] <= MAX_A);
        assert(s.count(a[i]) == 0);
        s.insert(a[i]);
        b[i] = a[i];
    }

    // 各modグループについて、実際にバブルソートして回数を数える
    int ans = 0;
    rep(k, K) {
        for (int j = k; j < N; j += K) {
            for (int i = k; i < j; i += K) {
                if (a[i] > a[j]) {
                    swap(a[i], a[j]);
                    ans++;
                }
            }
        }
    }

    sort(all(b));
    rep(i, N) {
        // ロボットソートの結果が、もとの数列をソートしたものと異なれば実は不可能だったということ
        if (a[i] != b[i]) {
            cout << -1 << endl;
            return 0;
        }
    }

    // ちゃんとソートできていれば回数を出力
    cout << ans << endl;

}
0