結果

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

ソースコード

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;
    vector<bool> used(N,false);
    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){
                int l=(cnt/2+1)*i+j+1,r=N-(cnt/2)*i-j;
                while(used[l]) l++;
                while(used[r]) r--;
                sum+=l+r;
                cout<<l<<" "<<r;
                if(j==cnt/2-1){
                    cout<<" "<<cost-sum<<endl;
                    used[cost-sum]=true;
                }
                else cout<<" ";
                used[l]=true;
                used[r]=true;
            }
        }
    }
}
0