結果

問題 No.115 遠足のおやつ
ユーザー ry0u_ydry0u_yd
提出日時 2015-09-02 14:25:38
言語 C++11
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,949 bytes
コンパイル時間 549 ms
コンパイル使用メモリ 73,344 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-18 17:50:36
合計ジャッジ時間 1,476 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <cstring>
#include <algorithm>
#include <sstream>
#include <cassert>
#include <map>
#include <set>

#define REP(i,k,n) for(int i=k;i<n;i++)
#define rep(i,n) for(int i=0;i<n;i++)
#define INF 1<<30
#define pb push_back
#define mp make_pair

using namespace std;
typedef long long ll;
typedef pair<int,int> P;

int main() {
    int n,d,k;
    cin >> n >> d >> k;

    int sum = 0, t = n;
    rep(i,k) {
        sum += t;
        t--;
    }

    if(d < (k*(k+1))/2 || d > sum) cout << -1 << endl;
    else {
        vector<int> v;
        rep(i,k) v.push_back(d/k);

        int cnt = d % k;
        rep(i,cnt) {
            v[i]++;
        }

        sort(v.begin(),v.end());
        int s = 0, t = v.size()-1;

        vector<int> down(n), up(n);
        rep(i,n) {
            down[i] = i+1;
            up[i] = n-k+1+i;
        }

        while(true) {
            if(s == t) break;

            if(v[s] == down[s] || v[s] == up[s]) {
                s++;
                continue;
            }

            if(v[t] == up[t]) {
                t--;
                continue;
            }


            int diff;
            if(v[s] >= up[s]) {
                diff = v[s] - up[s];
            } else {
                diff = v[s] - down[s];
            }

            while(diff) {
                if(v[t] + diff <= up[t]) {
                    v[s] -= diff;
                    v[t] += diff;
                    diff = 0;
                }
                else {
                    v[s] -= up[t] - v[t];
                    diff -= up[t] - v[t];
                    v[t] = up[t];
                    t--;
                }

                if(s == t) break;
            }

        }

        sort(v.begin(),v.end());
        rep(i,v.size()) {
            cout << v[i];

            if(i == v.size()-1) cout << endl;
            else cout << " ";
        }
    }


    return 0;
}
0