結果

問題 No.895 MESE
ユーザー aa
提出日時 2019-09-28 16:50:37
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 80 ms / 2,000 ms
コード長 4,238 bytes
コンパイル時間 1,748 ms
コンパイル使用メモリ 172,372 KB
実行使用メモリ 5,632 KB
最終ジャッジ日時 2024-10-02 12:00:45
合計ジャッジ時間 4,044 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 3 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 49 ms
5,248 KB
testcase_14 AC 41 ms
5,248 KB
testcase_15 AC 61 ms
5,248 KB
testcase_16 AC 51 ms
5,248 KB
testcase_17 AC 31 ms
5,248 KB
testcase_18 AC 80 ms
5,632 KB
testcase_19 AC 79 ms
5,504 KB
testcase_20 AC 79 ms
5,504 KB
testcase_21 AC 78 ms
5,504 KB
testcase_22 AC 78 ms
5,504 KB
testcase_23 AC 76 ms
5,504 KB
testcase_24 AC 76 ms
5,632 KB
testcase_25 AC 80 ms
5,504 KB
testcase_26 AC 79 ms
5,504 KB
testcase_27 AC 80 ms
5,504 KB
testcase_28 AC 78 ms
5,632 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

using namespace std;

template <int p>
struct Modint
{
    int value;

    Modint() : value(0) {}
    Modint(long x) : value(x >= 0 ? x % p : (p + x % p) % p) {}

    inline Modint &operator+=(const Modint &b)
    {
        if ((this->value += b.value) >= p)
            this->value -= p;
        return (*this);
    }
    inline Modint &operator-=(const Modint &b)
    {
        if ((this->value += p - b.value) >= p)
            this->value -= p;
        return (*this);
    }
    inline Modint &operator*=(const Modint &b)
    {
        this->value = (int)((1LL * this->value * b.value) % p);
        return (*this);
    }
    inline Modint &operator/=(const Modint &b)
    {
        (*this) *= b.inverse();
        return (*this);
    }

    Modint operator+(const Modint &b) const { return Modint(*this) += b; }
    Modint operator-(const Modint &b) const { return Modint(*this) -= b; }
    Modint operator*(const Modint &b) const { return Modint(*this) *= b; }
    Modint operator/(const Modint &b) const { return Modint(*this) /= b; }

    inline Modint &operator++(int) { return (*this) += 1; }
    inline Modint &operator--(int) { return (*this) -= 1; }

    inline bool operator==(const Modint &b) const
    {
        return this->value == b.value;
    }
    inline bool operator!=(const Modint &b) const
    {
        return this->value != b.value;
    }
    inline bool operator<(const Modint &b) const
    {
        return this->value < b.value;
    }
    inline bool operator<=(const Modint &b) const
    {
        return this->value <= b.value;
    }
    inline bool operator>(const Modint &b) const
    {
        return this->value > b.value;
    }
    inline bool operator>=(const Modint &b) const
    {
        return this->value >= b.value;
    }

    // requires that "this->value and p are co-prime"
    // a_i * v + a_(i+1) * p = r_i
    // r_i = r_(i+1) * q_(i+1) * r_(i+2)
    // q == 1 (i > 1)
    // reference: https://atcoder.jp/contests/agc026/submissions/2845729
    // (line:93)
    inline Modint inverse() const
    {
        assert(this->value != 0);
        int r0 = p, r1 = this->value, a0 = 0, a1 = 1;
        while (r1)
        {
            int q = r0 / r1;
            r0 -= q * r1;
            swap(r0, r1);
            a0 -= q * a1;
            swap(a0, a1);
        }
        return Modint(a0);
    }

    friend istream &operator>>(istream &is, Modint<p> &a)
    {
        long t;
        is >> t;
        a = Modint<p>(t);
        return is;
    }
    friend ostream &operator<<(ostream &os, const Modint<p> &a)
    {
        return os << a.value;
    }
};

/*
verified @ https://atcoder.jp/contests/abc034/submissions/4316971
*/

const int MOD = 1e9 + 7;

using Int = Modint<MOD>;

Int pow(Int e, long x)
{
    Int res = 1, tmp = e;
    while (x > 0)
    {
        if (x & 1)
            res *= tmp;
        tmp *= tmp;
        x >>= 1;
    }
    return res;
}

class Comb
{
public:
    vector<Int> fact, finv;
    Comb(int n) : fact(n + 1), finv(n + 1)
    {
        fact[0] = Int(1);
        for (int i = 1; i <= n; i++)
        {
            fact[i] = fact[i - 1] * Int(i);
        }
        finv[n] = Int(fact[n]).inverse();
        for (int i = n - 1; i >= 0; i--)
        {
            finv[i] = finv[i + 1] * Int(i + 1);
        }
    }

    inline Int nCk(int n, int k)
    {
        if (k < 0 || n < k)
            return Int(0);
        return Int(fact[n] * finv[n - k] * finv[k]);
    }

    inline Int nPk(int n, int k)
    {
        if (k < 0 || n < k)
            return Int(0);
        return Int(fact[n] * finv[n - k]);
    }
};

void solve()
{
    int a, b, c;
    cin >> a >> b >> c;
    int d = a + b + c;
    Comb comb(d + 30);
    Int ans = 0;
    for (int i = 2; i <= d - c; i++)
    {
        int r = d - i - 1, res = r - c + 1;
        if (c > 1 && r)
        {
            ans += ((pow((Int)2, r) - 1) * (Int)(c-1) / Int(r) + pow((Int)2, r)) * comb.nCk(r, c-1) * (comb.nCk(a+b-1, b) - comb.nCk(res, b));
        }
        else
        {
            ans += pow((Int)2, r) * (comb.nCk(a+b-1, b) - comb.nCk(res, b));
        }
        //cout << i << ' ' << ans << endl;
    }
    cout << ans << endl;
}

int main()
{
    solve();
    return 0;
}
0