結果

問題 No.942 プレゼント配り
ユーザー face4
提出日時 2019-12-05 12:59:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,637 bytes
コンパイル時間 675 ms
コンパイル使用メモリ 73,792 KB
実行使用メモリ 14,316 KB
最終ジャッジ日時 2024-12-17 18:53:56
合計ジャッジ時間 2,326 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 17 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

// 15 5 のようなケースでYesであるにもかかわらずNoと出力してしまう
// 1 11 12
// 2 9 13
// 3 7 14
// 4 5 15
// 6 8 10

#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;

void output(vector<vector<int>> &v){
    for(int i = 0; i < v[0].size(); i++){
        for(int j = 0; j < v.size(); j++){
            cout << v[j][i] << " \n"[j==v.size()-1];
        }
    }
}

int main(){
    int n, k;
    cin >> n >> k;
    if(n/k%2 == 0){
        cout << "Yes" << endl;
        vector<vector<int>> ans(n/k, vector<int>(k));
        for(int i = 0; i < n/k; i++){
            int from = i*k+1;
            for(int j = 0; j < k; j++){
                if(i%2) ans[i][j] = from+j;
                else    ans[i][j] = from+(k-1-j);
            }
        }
        output(ans);
    }else{
        int t;
        for(t = 1; t*k <= n/k; t++){
            if((n/k-k*t)%2 == 0)    break;
        }
        // そもそも k > n/kのときどうするかを考えないといけない
        if(t*k > n/k){
            cout << "No" << endl;
            return 0;
        }
        cout << "Yes" << endl;
        vector<vector<int>> ans(n/k, vector<int>(k));
        for(int i = 0; i < t*k; i++){
            int from = i*k+1;
            for(int j = 0; j < k; j++){
                ans[i][(i%k+j)%k] = from+j;
            }
        }
        for(int i = t*k; i < n/k; i++){
            int from = i*k+1;
            for(int j = 0; j < k; j++){
                if(i%2) ans[i][j] = from+j;
                else    ans[i][j] = from+(k-1-j);
            }
        }
        output(ans);
    }
    return 0;
}
0