結果

問題 No.940 ワープ　ε=ε=ε=ε=ε=│;p>д＜│
コンテスト
ユーザー potoooooooo
提出日時 2019-12-11 23:02:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 120 ms / 5,000 ms
コード長 1,610 bytes
コンパイル時間 1,720 ms
コンパイル使用メモリ 170,128 KB
実行使用メモリ 80,512 KB
最終ジャッジ日時 2024-06-24 08:06:44
合計ジャッジ時間 4,789 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define mod 1000000007
#define MAX_N 2500000
long long inv[MAX_N];
long long factorial[MAX_N];
long long inv_factorial[MAX_N];

void init(){
    for (int i = 1; i < MAX_N; i++) {
        if (i == 1) inv[i] = 1;
        else {
            inv[i] = (mod - (mod / i) * inv[mod % i]) % mod;
            if (inv[i] < 0) inv[i] += mod;
        }
    }
    factorial[0] = 1; inv_factorial[0] = 1;
    for (int i = 1; i < MAX_N; i++) {
        factorial[i] = factorial[i-1] * i;
        factorial[i] %= mod;
        inv_factorial[i] = inv_factorial[i-1] * inv[i];
        inv_factorial[i] %= mod;
    }
}

long long combination(int n, int r) {
    long long ret = factorial[n] * inv_factorial[r];
    ret %= mod;
    ret *= inv_factorial[n-r];
    return ret % mod;
}

int if0() {
    cout << 1 << endl;
    return 0;
}

int main() {
    int x, y, z; cin >> x >> y >> z;
    int n = x + y + z;
    if (n == 0) return if0();
    init();

    vector<long long> w(n+1, 0);
    for (int i = 1; i <= n; i++) {
        w[i] = combination(x+i-1, x) * combination(y+i-1, y) % mod * combination(z+i-1, z) % mod;
    }

    // c = {1 - (x-1)^(n+1)} / 2(1 - x/2)     
    vector<long long> c(n+1, 0); c[0] = 1; 
    for (int i = 0; i <= n; i++) {
        if ((n + 1 - i) & 1) c[i] += combination(n+1, i);
        else c[i] += mod - combination(n+1, i);
        if (i) c[i] += c[i-1];
        c[i] = c[i] % mod * inv[2] % mod;
    }
    long long ans = 0;
    for (int i = 1; i <= n; i++) {
        ans = (ans + c[i] * w[i] % mod) % mod;
    }
    cout << ans << endl;
    return 0;
}
0