結果

問題 No.942 プレゼント配り
ユーザー otamay6
提出日時 2019-12-05 10:25:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,145 bytes
コンパイル時間 1,447 ms
コンパイル使用メモリ 168,592 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-12-17 13:26:27
合計ジャッジ時間 3,528 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define REP(i,n) for(int i=0,i##_len=(n);i<i##_len;++i)
#define rep(i,a,b) for(int i=int(a);i<int(b);++i)
#define All(x) (x).begin(),(x).end()
using namespace std;
using ll = long long;

int main(){
    ll N,K;cin>>N>>K;
    ll cnt=N/K;
    ll cost=N*(N+1)/2;
    if(cost%K!=0){
        cout<<"No"<<endl;
        return 0;
    }
    cost/=K;
    cout<<"Yes"<<endl;
    REP(i,K){
        if(cnt%2==0) REP(j,cnt/2){
            cout<<(N/K)*i+j+1<<" "<<N-(N/K)*i-j;
            if(j==cnt/2-1) cout<<endl;
            else cout<<" ";
        }
        else{
            ll sum=0;
            REP(j,cnt/2){
                if(i<=K/2){
                    sum+=(cnt/2+1)*i+N-i+1;
                    cout<<(cnt/2+1)*i+j+1<<" "<<N-i-j;
                    if(j==cnt/2-1) cout<<" "<<cost-sum<<endl;
                    else cout<<" ";
                }
                else{
                    sum+=(cnt/2+1)*(i-K/2)+N-i;
                    cout<<(cnt/2+1)*(i-K/2)+j<<" "<<N-i-j;
                    if(j==cnt/2-1) cout<<" "<<cost-sum<<endl;
                    else cout<<" ";
                }
            }
        }
    }
}
0