結果

問題 No.366 ロボットソート
コンテスト
ユーザー 鬼丸制作所プライベートアカウント
提出日時 2026-04-20 11:11:49
言語 C++14
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++14 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 1,069 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 568 ms
コンパイル使用メモリ 97,324 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2026-04-20 11:11:53
合計ジャッジ時間 1,464 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>

using namespace std;

bool solve() {
    int N, K;
    if (!(cin >> N >> K)) return false;

    vector<int> A(N);
    vector<int> S(N);
    for (int i = 0; i < N; ++i) {
        cin >> A[i];
        S[i] = A[i];
    }

    // 理想の並び（昇順）を作成
    sort(S.begin(), S.end());

    // 各グループ (余り 0 ~ K-1) ごとに要素を抽出して比較
    for (int r = 0; r < K; ++r) {
        vector<int> groupA, groupS;
        
        for (int i = r; i < N; i += K) {
            groupA.push_back(A[i]);
            groupS.push_back(S[i]);
        }

        // グループ内の数字をソートして比較
        sort(groupA.begin(), groupA.end());
        sort(groupS.begin(), groupS.end());

        if (groupA != groupS) {
            return false; // 構成要素が違うので並び替え不可
        }
    }

    return true;
}

int main() {
    if (solve()) {
        cout << "Yes" << endl;
    } else {
        cout << "No" << endl;
    }
    return 0;
}
0