結果

問題 No.942 プレゼント配り
コンテスト
ユーザー otamay6
提出日時 2019-12-05 10:38:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,588 bytes
コンパイル時間 1,914 ms
コンパイル使用メモリ 173,848 KB
実行使用メモリ 12,800 KB
最終ジャッジ日時 2024-12-17 14:03:46
合計ジャッジ時間 4,434 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 13 WA * 5
権限があれば一括ダウンロードができます

ソースコード

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;
    set<int> st;
    REP(i,N) st.insert(i+1);
    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+=N+i+1;
                    cout<<(cnt/2+1)*i+j+1<<" "<<N-(cnt/2)*i-j;
                    if(j==cnt/2-1){
                        cout<<" "<<cost-sum<<endl;
                        st.erase(cost-sum);
                    }
                    else cout<<" ";
                    st.erase((cnt/2+1)*i+j+1);
                    st.erase(N-(cnt/2)*i-j);
                }
                else{
                    auto itb=st.begin(),ite=st.end();
                    ite--;
                    sum+=(*itb)+(*ite);
                    cout<<*itb<<" "<<*ite;
                    if(j==cnt/2-1){
                        cout<<" "<<cost-sum<<endl;
                        st.erase(cost-sum);
                    }
                    else cout<<" ";
                    st.erase(ite);
                    st.erase(itb);
                }
            }
        }
    }
}
0