結果

問題 No.584 赤、緑、青の色塗り
ユーザー 夕叢霧香(ゆうむらきりか)夕叢霧香(ゆうむらきりか)
提出日時 2017-10-28 00:18:19
言語 Java21
(openjdk 21)
結果
AC  
実行時間 881 ms / 2,000 ms
コード長 2,736 bytes
コンパイル時間 2,405 ms
コンパイル使用メモリ 77,424 KB
実行使用メモリ 41,824 KB
最終ジャッジ日時 2024-05-01 18:18:24
合計ジャッジ時間 7,304 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 143 ms
41,224 KB
testcase_01 AC 145 ms
41,812 KB
testcase_02 AC 146 ms
41,380 KB
testcase_03 AC 145 ms
41,316 KB
testcase_04 AC 145 ms
41,428 KB
testcase_05 AC 153 ms
41,232 KB
testcase_06 AC 142 ms
41,352 KB
testcase_07 AC 141 ms
41,512 KB
testcase_08 AC 142 ms
41,636 KB
testcase_09 AC 138 ms
41,372 KB
testcase_10 AC 137 ms
41,224 KB
testcase_11 AC 142 ms
41,668 KB
testcase_12 AC 143 ms
41,488 KB
testcase_13 AC 143 ms
41,384 KB
testcase_14 AC 214 ms
41,824 KB
testcase_15 AC 153 ms
41,532 KB
testcase_16 AC 162 ms
41,372 KB
testcase_17 AC 178 ms
41,588 KB
testcase_18 AC 161 ms
41,352 KB
testcase_19 AC 881 ms
41,736 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

class Main {
    static final long MOD = 1000000007;
    static final int N = 5000;
    static long[] fact;
    static long[] invfact;
    static void initialize() {
	fact = new long[N];
	invfact = new long[N];
	fact[0] = invfact[0] = 1;
	for (int i = 1; i < N; ++i) {
	    fact[i] = (i * fact[i - 1]) % MOD;
	    invfact[i] = powerMod(fact[i], MOD - 2);
	}
    }
    static long powerMod(long x, long exponent) {
	long prod = 1;
	for (int i = 63; i >= 0; --i) {
	    prod = (prod * prod) % MOD;
	    if ((exponent & 1L << i) != 0) {
		prod = (prod * x) % MOD;
	    }
	}
	return prod;
    }
    static long comb(int x, int y) {
	if (x < 0) {
	    return 0;
	}
	if (y < 0 || y > x) {
	    return 0;
	}
	long r= (fact[x] * powerMod((fact[x - y] * fact[y]) % MOD, MOD - 2)) % MOD;
	return r;
    }
    static long f(int x, int y) {
	if (x == 0) {
	    return y == 0 ? 1 : 0;
	}
	return comb(x - 1, y - 1);
    }
    static long sub(int n, int k, int t) {
	long total = 0;
	if (n - k >= k - t + 1) {
	    total += f(n - k, k - t + 1);
	}
	if (n - k >= k - t) {
	    total += 2 * f(n - k, k - t);
	}
	if (n - k >= k - t - 1) {
	    total += f(n - k, k - t - 1);
	}
	total %= MOD;
	return (comb(k - t, t) * total) % MOD;
    }
    static long helper(int t, int u, int r, int g, int b) {
	long total = 0;
	for (int x = 0; x <= t; ++x) {
	    if (r < x || g < x) {
		continue;
	    }
	    for (int y = 0; y <= t - x; ++y) {
		int z = t - x - y;
		int rRest = r - x - z;
		int gRest = g - x - y;
		int bRest = b - y - z;
		if (rRest < 0 || gRest < 0 || bRest < 0) {
		    continue;
		}
		long divisor = (invfact[rRest] * invfact[gRest]) % MOD;
		divisor = (divisor * invfact[bRest]) % MOD;
		divisor = (divisor * invfact[x]) % MOD;
		divisor = (divisor * invfact[y]) % MOD;
		divisor = (divisor * invfact[z]) % MOD;
		long tmp = divisor;
		total = (total + tmp) % MOD;
	    }
	}
	long multiplier = (fact[u] * fact[t]) % MOD;
	total = (total * multiplier) % MOD;
	return (total * powerMod(2, t)) % MOD;
    }
    static long calculate(int n, int r, int g, int b) {
	initialize();
	int count = r + g + b;
	if (count > n) {
	    return 0;
	}
	long total = 0;
	for (int t = 0; t <= count / 2; ++t) {
	    long res = sub(n, count, t);
	    if (res == 0) {
		continue;
	    }
	    long dat = helper(t, count - 2 * t, r, g, b);
	    long mult = (res * dat) % MOD;
	    total += mult;
	    total %= MOD;
	}
	return total;
    }
    public static void main(String[] args) {
	Scanner scan = new Scanner(System.in);
	int n = scan.nextInt();
	int r = scan.nextInt();
	int g = scan.nextInt();
	int b = scan.nextInt();
	int[] rgb = new int[]{r, g, b};
	Arrays.sort(rgb);
	System.out.println(calculate(n, rgb[0], rgb[1], rgb[2]));
    }
}
0