結果
問題 | No.584 赤、緑、青の色塗り |
ユーザー | mamekin |
提出日時 | 2017-11-12 21:17:18 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 23 ms / 2,000 ms |
コード長 | 2,885 bytes |
コンパイル時間 | 1,071 ms |
コンパイル使用メモリ | 106,896 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-05-03 19:34:30 |
合計ジャッジ時間 | 1,706 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,376 KB |
testcase_02 | AC | 1 ms
5,376 KB |
testcase_03 | AC | 1 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 1 ms
5,376 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 2 ms
5,376 KB |
testcase_08 | AC | 1 ms
5,376 KB |
testcase_09 | AC | 1 ms
5,376 KB |
testcase_10 | AC | 1 ms
5,376 KB |
testcase_11 | AC | 1 ms
5,376 KB |
testcase_12 | AC | 2 ms
5,376 KB |
testcase_13 | AC | 1 ms
5,376 KB |
testcase_14 | AC | 3 ms
5,376 KB |
testcase_15 | AC | 23 ms
5,376 KB |
testcase_16 | AC | 2 ms
5,376 KB |
testcase_17 | AC | 2 ms
5,376 KB |
testcase_18 | AC | 2 ms
5,376 KB |
testcase_19 | AC | 16 ms
5,376 KB |
ソースコード
#define _USE_MATH_DEFINES #include <cstdio> #include <iostream> #include <sstream> #include <fstream> #include <iomanip> #include <algorithm> #include <cmath> #include <complex> #include <string> #include <vector> #include <list> #include <queue> #include <stack> #include <set> #include <map> #include <bitset> #include <numeric> #include <limits> #include <climits> #include <cfloat> #include <functional> #include <iterator> using namespace std; void mod_inverse(int n, vector<long long>& inv, int mod) { inv.assign(n+1, -1); inv[1] = 1; for(int i=2; i<=n; ++i) inv[i] = inv[mod % i] * (mod - mod / i) % mod; } class FactorialCalculation { private: const int mod; vector<long long> factorial; vector<long long> invFactorial; public: FactorialCalculation(int n, int mod) : mod(mod) { factorial.resize(n+1, 1); invFactorial.resize(n+1, 1); vector<long long> inv; mod_inverse(n, inv, mod); for(int i=1; i<=n; ++i){ factorial[i] = factorial[i-1] * i % mod; invFactorial[i] = invFactorial[i-1] * inv[i] % mod; } } long long getFactorial(int n){ return factorial[n]; } long long getInvFactorial(int n){ return invFactorial[n]; } long long getPermutation(int n, int r){ if(n < r) return 0; return factorial[n] * invFactorial[n-r] % mod; } long long getCombination(int n, int r){ if(n < r) return 0; return getPermutation(n, r) * invFactorial[r] % mod; } long long getHomogeneous(int n, int r){ return getCombination(n+r-1, r); } }; const int MOD = 1000000007; long long power(int a, int b) { long long ret = 1; long long tmp = a; while(b > 0){ if(b & 1){ ret *= tmp; ret %= MOD; } tmp *= tmp; tmp %= MOD; b >>= 1; } return ret; } int main() { int n, r, g, b; cin >> n >> r >> g >> b; int sum = r + g + b; int white = n - sum; FactorialCalculation fc(2*n, MOD); long long ans = 0; for(int p=0; p<=sum/2; ++p){ long long tmp = fc.getCombination(white+1, sum-p); tmp *= fc.getCombination(sum-p, p); tmp %= MOD; tmp *= power(2, p); tmp %= MOD; for(int i=0; i<=p; ++i){ int r2 = r - i; int g2 = g - (p - i); int b2 = b - (p - i); if(r2 < 0 || g2 < 0 || b2 < 0) continue; long long tmp2 = tmp; tmp2 *= fc.getCombination(p, i); tmp2 %= MOD; tmp2 *= fc.getCombination(sum - 2 * p, r2); tmp2 %= MOD; tmp2 *= fc.getCombination(g2 + b2, g2); tmp2 %= MOD; ans += tmp2; ans %= MOD; } } cout << ans << endl; return 0; }