結果

問題 No.584 赤、緑、青の色塗り
コンテスト
ユーザー TangentDay
提出日時 2017-10-27 23:22:17
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 1,567 ms / 2,000 ms
コード長 2,254 bytes
コンパイル時間 1,161 ms
コンパイル使用メモリ 83,004 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-21 23:16:58
合計ジャッジ時間 3,542 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <fstream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <algorithm>
using namespace std;

#define REP(i,n) for(int i=0; i<n; ++i)
#define FOR(i,a,b) for(int i=a; i<=b; ++i)
#define FORR(i,a,b) for (int i=a; i>=b; --i)
#define ALL(c) (c).begin(), (c).end()

typedef long long ll;
typedef vector<int> VI;
typedef vector<ll> VL;
typedef vector<VL> VVL;
typedef vector<VI> VVI;
typedef pair<int,int> P;
typedef pair<ll,ll> PL;

const ll mod = 1000000007;
const int N = 3000;
ll fact[N], invf[N], tw[N];

ll mul(ll x, ll y){
    return (x%mod)*(y%mod)%mod;
}

ll powll(ll x, ll y){
    ll res = 1LL;
    while(y){
        if (y & 1LL)
            res *= x;
        res %= mod;
        x = (x*x) % mod;
        y >>= 1LL;
    }
    return res;
}

ll divll(ll x, ll y){
    return (x * powll(y,mod-2)) % mod;
}

ll nPr(ll n, ll r){
    if (n < r || r < 0) return 0;
    return mul(fact[n], invf[n-r]);
}

ll nCr(ll n, ll r){
    if (n < r || r < 0) return 0;
    return mul(mul(fact[n], invf[r]), invf[n-r]);
}

int main() {
    fact[0] = invf[0] = tw[0] = 1;
    FOR(i,1,N-1){
        fact[i] = (fact[i-1] * i) % mod;
        invf[i] = divll(invf[i-1], i);
        tw[i] = (tw[i-1] * 2) % mod;
    }

    ll n, r, g, b;
    cin >> n >> r >> g >> b;

    ll sum = r + g + b;
    ll ans = 0;

    REP(x,n/3+2){
        int y = sum - 2*x;
        if (y < 0) continue;
        ll di = (nCr(n-2*x-y+1, x+y) * nCr(x+y, x)) % mod;
        if (di == 0) continue;
        ll tmp = 0;
        REP(rg,x+1) REP(gb,x-rg+1){
            int br = x - rg - gb;
            int rr = r - rg - br;
            int gg = g - rg - gb;
            int bb = b - gb - br;
            if (rr < 0 || gg < 0 || bb < 0) continue;
            ll saku = ((nCr(x, rg) * nCr(x-rg, gb) % mod) * tw[rg + gb + br]) % mod;
            ll hima = (nCr(y, rr) * nCr(y-rr, gg)) % mod;
            tmp = (tmp + (saku * hima) % mod) % mod;
        }
        // printf("x = %d, y = %d, di = %lld, tmp = %lld\n", x, y, di, tmp);
        tmp = (tmp * di) % mod;
        ans = (ans + tmp) % mod;
    }
    cout << ans << endl;

    return 0;
}
0