結果

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

ソースコード

diff # 

#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;
        }
        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