結果

問題 No.940 ワープ　ε=ε=ε=ε=ε=│;p>д＜│
ユーザー potoooooooopotoooooooo
提出日時 2019-12-11 23:02:40
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 75 ms
61,988 KB
testcase_01 AC 73 ms
62,124 KB
testcase_02 AC 75 ms
62,120 KB
testcase_03 AC 71 ms
61,988 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 65 ms
62,116 KB
testcase_06 AC 68 ms
61,992 KB
testcase_07 AC 69 ms
62,120 KB
testcase_08 AC 68 ms
61,992 KB
testcase_09 AC 66 ms
62,124 KB
testcase_10 AC 70 ms
62,124 KB
testcase_11 AC 75 ms
62,120 KB
testcase_12 AC 72 ms
61,992 KB
testcase_13 AC 70 ms
61,992 KB
testcase_14 AC 70 ms
62,120 KB
testcase_15 AC 77 ms
63,144 KB
testcase_16 AC 83 ms
65,960 KB
testcase_17 AC 92 ms
70,260 KB
testcase_18 AC 102 ms
72,748 KB
testcase_19 AC 93 ms
70,112 KB
testcase_20 AC 109 ms
75,948 KB
testcase_21 AC 85 ms
67,236 KB
testcase_22 AC 108 ms
75,368 KB
testcase_23 AC 85 ms
66,092 KB
testcase_24 AC 108 ms
75,656 KB
testcase_25 AC 118 ms
78,580 KB
testcase_26 AC 120 ms
80,512 KB
権限があれば一括ダウンロードができます

ソースコード

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