結果

問題 No.1043 直列大学
ユーザー htensai
提出日時 2020-05-07 10:45:14
言語 Java
(openjdk 23)
結果
AC  
実行時間 1,823 ms / 2,000 ms
コード長 2,241 bytes
コンパイル時間 2,430 ms
コンパイル使用メモリ 79,740 KB
実行使用メモリ 88,332 KB
最終ジャッジ日時 2024-12-22 20:07:14
合計ジャッジ時間 29,243 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.*;

public class Main {
    static final int MOD = 1000000007;
	public static void main (String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int m = sc.nextInt();
		HashMap<Integer, Integer> vMap = new HashMap<>();
		vMap.put(0, 1);
		for (int i = 0; i < n; i++) {
		    int x = sc.nextInt();
		    HashMap<Integer, Integer> tmp = new HashMap<>();
		    for (Map.Entry<Integer, Integer> entry : vMap.entrySet()) {
		        tmp.put(entry.getKey() + x, entry.getValue());
		    }
		    for (Map.Entry<Integer, Integer> entry : tmp.entrySet()) {
		        if (vMap.containsKey(entry.getKey())) {
		            vMap.put(entry.getKey(), (vMap.get(entry.getKey()) + entry.getValue()) % MOD);
		        } else {
		            vMap.put(entry.getKey(), entry.getValue());
		        }
		    }
		}
		HashMap<Integer, Integer> rMap = new HashMap<>();
		rMap.put(0, 1);
		for (int i = 0; i < m; i++) {
		    int x = sc.nextInt();
		    HashMap<Integer, Integer> tmp = new HashMap<>();
		    for (Map.Entry<Integer, Integer> entry : rMap.entrySet()) {
		        tmp.put(entry.getKey() + x, entry.getValue());
		    }
		    for (Map.Entry<Integer, Integer> entry : tmp.entrySet()) {
		        if (rMap.containsKey(entry.getKey())) {
		            rMap.put(entry.getKey(), (rMap.get(entry.getKey()) + entry.getValue()) % MOD);
		        } else {
		            rMap.put(entry.getKey(), entry.getValue());
		        }
		    }
		}
		int left = sc.nextInt();
		int right = sc.nextInt();
		int[] vsums = new int[n * 1000 + 1];
		for (int i = 1; i < vsums.length; i++) {
		    vsums[i] = vsums[i - 1];
		    if (vMap.containsKey(i)) {
		        vsums[i] += vMap.get(i);
		        vsums[i] %= MOD;
		    }
		}
		long total = 0;
	    for (Map.Entry<Integer, Integer> entry : rMap.entrySet()) {
	        int key = entry.getKey();
	        if ((long)key * left >= vsums.length) {
	            continue;
	        }
	        if (key == 0) {
	            continue;
	        }
	        int min = key * left - 1;
	        int max = (int)Math.min((long)key * right, vsums.length - 1);
	        total += (long)(vsums[max] - vsums[min] + MOD) * entry.getValue();
	        total %= MOD;
	    }
		System.out.println(total);
	}
}
0