結果

問題 No.115 遠足のおやつ
コンテスト
ユーザー kroton
提出日時 2014-12-11 13:46:38
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,323 bytes
コンパイル時間 1,524 ms
コンパイル使用メモリ 156,604 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2025-01-03 00:58:08
合計ジャッジ時間 2,951 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int N, D, K;
int res[11];

bool dfs(int k, int acc, int curr){
    if(k >= K){
        if(acc == D){
            for(int i=0;i<K;i++){
                if(i > 0){
                    cout << " ";
                }
                cout << res[i];
            }
            cout << endl;

            return true;
        } else {
            return false;
        }
    }
    if(acc >= D || curr > N)return false;
    if(N - curr + 1 < K - k)return false;
    {
        int rest_d = D - acc;
        int rest_k = K - k;

        int mini = (curr + rest_k - 1) * (curr + rest_k) / 2 - (curr - 1) * curr / 2;
        int maxi = N * (N + 1) / 2 - (N - rest_k) * (N - rest_k + 1) / 2;

        if(rest_d < mini || maxi < rest_d)return false;
    }
    
    res[k] = curr;
    if(dfs(k + 1, acc + curr, curr + 1))return true;
    if(dfs(k, acc, curr + 1))return true;

    return false;
}

void dfs_solver(int N_, int D_, int K_){
    N = N_;
    D = D_;
    K = K_;
    if(!dfs(0, 0, 1)){
        cout << -1 << endl;
    }
}

int main(){
    int N, D, K;
    cin >> N >> D >> K;
 
    assert(1 <= N && N <= 100);
    assert(1 <= D && D <= 1000);
    assert(1 <= K && K <= 10);
    
    dfs_solver(N, D, K);
    return 0;
}
0