結果

問題 No.1683 Robot Guidance
ユーザー kai4096_donkai4096_don
提出日時 2021-09-17 22:34:28
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 704 ms / 2,000 ms
コード長 2,210 bytes
コンパイル時間 3,411 ms
コンパイル使用メモリ 231,720 KB
実行使用メモリ 34,644 KB
最終ジャッジ日時 2024-06-29 21:19:56
合計ジャッジ時間 29,801 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 589 ms
34,560 KB
testcase_01 AC 588 ms
34,456 KB
testcase_02 AC 595 ms
34,576 KB
testcase_03 AC 600 ms
34,416 KB
testcase_04 AC 588 ms
34,412 KB
testcase_05 AC 592 ms
34,588 KB
testcase_06 AC 608 ms
34,516 KB
testcase_07 AC 594 ms
34,620 KB
testcase_08 AC 647 ms
34,552 KB
testcase_09 AC 602 ms
34,572 KB
testcase_10 AC 634 ms
34,564 KB
testcase_11 AC 622 ms
34,524 KB
testcase_12 AC 643 ms
34,420 KB
testcase_13 AC 586 ms
34,560 KB
testcase_14 AC 704 ms
34,480 KB
testcase_15 AC 591 ms
34,452 KB
testcase_16 AC 585 ms
34,560 KB
testcase_17 AC 592 ms
34,420 KB
testcase_18 AC 589 ms
34,560 KB
testcase_19 AC 586 ms
34,476 KB
testcase_20 AC 588 ms
34,484 KB
testcase_21 AC 586 ms
34,448 KB
testcase_22 AC 584 ms
34,488 KB
testcase_23 AC 587 ms
34,560 KB
testcase_24 AC 655 ms
34,620 KB
testcase_25 AC 589 ms
34,420 KB
testcase_26 AC 595 ms
34,524 KB
testcase_27 AC 586 ms
34,620 KB
testcase_28 AC 587 ms
34,416 KB
testcase_29 AC 587 ms
34,560 KB
testcase_30 AC 586 ms
34,420 KB
testcase_31 AC 588 ms
34,512 KB
testcase_32 AC 591 ms
34,560 KB
testcase_33 AC 586 ms
34,420 KB
testcase_34 AC 600 ms
34,560 KB
testcase_35 AC 594 ms
34,484 KB
testcase_36 AC 586 ms
34,644 KB
testcase_37 AC 583 ms
34,416 KB
testcase_38 AC 597 ms
34,560 KB
testcase_39 AC 585 ms
34,616 KB
testcase_40 AC 585 ms
34,560 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
const long nPrime = 1000000007;
//const long nPrime = 998244353;
typedef long long ll;
long Inverse(const long &n){
    if (n % nPrime == 0){
        return 0;
    }
    long a = n, b = nPrime, u = 1, v = 0;
    while (b > 0) {
        long t = a / b;
        a -= t * b;
        swap(a, b);
        u -= t * v;
        swap(u, v);
    }
    u %= nPrime;
    if (u < 0){
        u += nPrime;
    }
    return u;    
}
int main() {
    vector<long> viTable(2000001,1), viInv(2000001,1);
    for(long i = 1; i <= 2000000; i++){
        viTable[i] = (viTable[i-1] * i) % nPrime;
        viInv[i] = Inverse(viTable[i]);
    }
    
    long a,b,x,y;
    cin >> a >> b >> x >> y;
    if(((a+x+y)&1) == 1){
        cout << 0 << endl;
        return 0;
    }
    
    long i1 = (b+4)/4;
    long i2 = (b+2)/4;
    long i3 = (b+3)/4;
    long i4 = (b+1)/4;
    
    long nAns = 0;
    for(long s = 0; s <= a; s++){
        long t = s-x;
        long u = (a-2*s+x+y)/2;
        long v = u-y;
        if(t < 0 || u < 0 || v < 0){
            continue;
        }
        if((i2 == 0 && t > 0) ||(i3 == 0 && u > 0) ||(i4 == 0 && v > 0) ){
            continue;
        }
        
        long iThis = 1;
        iThis *= viTable[s+i1-1];
        iThis %= nPrime;
        iThis *= viInv[s];
        iThis %= nPrime;
        iThis *= viInv[i1-1];
        iThis %= nPrime;
        if(i2 > 0){
            iThis *= viTable[t+i2-1];
            iThis %= nPrime;
            iThis *= viInv[t];
            iThis %= nPrime;
            iThis *= viInv[i2-1];
            iThis %= nPrime;
        }
        if(i3 > 0){
            iThis *= viTable[u+i3-1];
            iThis %= nPrime;
            iThis *= viInv[u];
            iThis %= nPrime;
            iThis *= viInv[i3-1];
            iThis %= nPrime;
        }
        if(i4 > 0){
            iThis *= viTable[v+i4-1];
            iThis %= nPrime;
            iThis *= viInv[v];
            iThis %= nPrime;
            iThis *= viInv[i4-1];
            iThis %= nPrime;
        }
        nAns += iThis;
        nAns %= nPrime;
    }
    
    cout << nAns << endl;
    return 0;
}
0