結果

問題 No.269 見栄っ張りの募金活動
ユーザー uafr_cs
提出日時 2015-12-07 01:30:37
言語 Java
(openjdk 23)
結果
TLE  
実行時間 -
コード長 1,942 bytes
コンパイル時間 2,524 ms
コンパイル使用メモリ 79,444 KB
実行使用メモリ 146,064 KB
最終ジャッジ日時 2024-09-14 17:50:27
合計ジャッジ時間 9,608 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 1 -- * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Scanner;
import java.util.Set;

public class Main {
	
	public static final long MOD = 1000000007;
	
	public static long dfs(final int N, final int rest, final int K, final int prev, ArrayList<HashMap<Integer, HashMap<Integer, Long>>> memoized){
		final int use_limit = prev + K;
		
		if(memoized.get(N).containsKey(rest) && memoized.get(N).get(rest).containsKey(prev)){
			return memoized.get(N).get(rest).get(prev);
		}
		
		if(N == 0){ return rest == 0 ? 1 : 0; }
		else if(rest <= 0){ return 0; }
		
		final int lowest = N * use_limit + N * (N - 1) / 2 * K;
		//System.out.println(lowest);
		if(lowest > rest){ 
			if(!memoized.get(N).containsKey(rest)){ memoized.get(N).put(rest, new HashMap<Integer, Long>()); }
			memoized.get(N).get(rest).put(prev, 0l);
			return 0;
		}
		//System.out.println(N + " " + use_limit + " " + rest + " " + lowest + " " + (rest - lowest + N - 1) / N);
		
		long ret = 0;
		for(int use = use_limit + (rest - lowest + N - 1) / N; use >= use_limit; use--){
			//System.out.println(use);
			ret += dfs(N - 1, rest - use, K, use, memoized);
			ret %= MOD;
		}
		
		if(!memoized.get(N).containsKey(rest)){
			memoized.get(N).put(rest, new HashMap<Integer, Long>());
		}
		memoized.get(N).get(rest).put(prev, ret);
		
		return ret;
	}
	
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		
		final int N = sc.nextInt();
		final int S = sc.nextInt();
		final int K = sc.nextInt();
		
		ArrayList<HashMap<Integer, HashMap<Integer, Long>>> memoized = new ArrayList<HashMap<Integer, HashMap<Integer, Long>>>();
		for(int i = 0; i <= N; i++){
			memoized.add(new HashMap<Integer, HashMap<Integer, Long>>());
		}
		
		System.out.println(dfs(N, S, K, -K, memoized));
		
		//System.out.println(memoized);
	}
	
}
0