結果

問題 No.115 遠足のおやつ
コンテスト
ユーザー takeya_okino
提出日時 2017-06-14 13:35:18
言語 Java
(openjdk 23)
結果
AC  
実行時間 174 ms / 5,000 ms
コード長 1,492 bytes
コンパイル時間 2,891 ms
コンパイル使用メモリ 80,756 KB
実行使用メモリ 65,428 KB
最終ジャッジ日時 2025-01-03 01:23:53
合計ジャッジ時間 8,895 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    int N = sc.nextInt();
    int D = sc.nextInt();
    int K = sc.nextInt();
    // dp[i][j][k]は最低金額(i+1)円,総額j円,個数k個の買い方があるかを表す
    long[][][] dp = new long[N][D + 1][K + 1];
    // sum[i][j][k] = dp[i][j][k] +...+dp[N - 1][j][k]
    long[][][] sum = new long[N][D + 1][K + 1];
    dp[N - 1][0][0] = 1;
    if(N < D + 1) dp[N - 1][N][1] = 1;
    sum[N - 1][0][0] = 1;
    if(N < D + 1) sum[N - 1][N][1] = dp[N - 1][N][1];
    for(int i = N - 2; i >= 0; i--) {
      dp[i][0][0] = 1;
      sum[i][0][0] = 1 + sum[i + 1][0][0];
      for(int j = 1; j < D + 1; j++) {
        for(int k = 1; k < K + 1; k++) {
          if(j >= (i + 1)) dp[i][j][k] = sum[i + 1][j - i - 1][k - 1];
          sum[i][j][k] = dp[i][j][k] + sum[i + 1][j][k];
        }
      }
    }
    boolean flg = (sum[0][D][K] > 0);
    if(flg) {
      ArrayList<Integer> ans = new ArrayList<Integer>();
      int cost = D;
      int num = K;
      for(int i = 0; i < N; i++) {
        if(num > 0) {
          if(dp[i][cost][num] > 0) {
            ans.add(i + 1);
            cost -= (i + 1);
            num--;
          }
        } else {
          break;
        }
      }
      for(int i = 0; i < K; i++) {
        System.out.print(ans.get(i));
        if(i < K - 1) System.out.print(" ");
      }
    } else {
      System.out.println(-1);
    }
  }
}
0