結果

問題 No.498 ワープクリスタル (給料日編)
ユーザー Konton7Konton7
提出日時 2020-02-02 14:49:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 2,727 bytes
コンパイル時間 2,257 ms
コンパイル使用メモリ 204,936 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-18 20:45:22
合計ジャッジ時間 3,095 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 3 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 4 ms
6,940 KB
testcase_07 AC 4 ms
6,940 KB
testcase_08 AC 3 ms
6,940 KB
testcase_09 AC 4 ms
6,940 KB
testcase_10 AC 2 ms
6,944 KB
testcase_11 AC 3 ms
6,944 KB
testcase_12 AC 2 ms
6,944 KB
testcase_13 AC 2 ms
6,940 KB
testcase_14 AC 2 ms
6,940 KB
testcase_15 AC 2 ms
6,940 KB
testcase_16 AC 2 ms
6,944 KB
testcase_17 AC 2 ms
6,940 KB
testcase_18 AC 3 ms
6,944 KB
testcase_19 AC 3 ms
6,940 KB
testcase_20 AC 4 ms
6,944 KB
testcase_21 AC 3 ms
6,940 KB
testcase_22 AC 4 ms
6,944 KB
testcase_23 AC 3 ms
6,944 KB
testcase_24 AC 4 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
using ll = long long;
using PII = std::pair<int, int>;
using PLL = std::pair<ll, ll>;

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rep2(i, s, n) for (int i = (s); i < (int)(n); i++)
// #define DEBUG

struct crystal{
    int x;
    int y;
    int n;
};

ll pow_mod(int p, int q, int mod)
{
    ll ret, r;
    ret = 1;
    r = p;

    while (q > 0)
    {
       
        if (q % 2)
        {
            ret *= r;
            ret %= mod;
        }
        
        r = (r * r) % mod;
        q /= 2;        
    }

    return ret % mod;
}

int main()
{

#ifdef DEBUG
    cout << "DEBUG MODE" << endl;
    ifstream in("input.txt"); //for debug
    cin.rdbuf(in.rdbuf());    //for debug
#endif

    const int mod = 1e9+7;
    ll frac[76], frac_inv[16];
    frac[0] = 1, frac_inv[0] = 1;
    rep2(i, 1, 76)
        frac[i] = (frac[i-1] * i) % mod;
    rep2(i, 1, 16)
        frac_inv[i] = ( frac_inv[i-1] * pow_mod(i, mod - 2, mod) ) % mod;



    int gx, gy, k, x, y, n, xsum, ysum;
    cin >> gx >> gy >> k;
    ll ans, s;
    vector<crystal> crystals;
    rep(i, k)
    {
        cin >> x >> y >> n;
        crystals.push_back( (crystal) {x, y, n} );
    }
    rep(i, 5-k)
        crystals.push_back( (crystal) {0, 0, 0} );

    ans = 0;
    rep(c0, crystals[0].n+1)
        rep(c1, crystals[1].n+1)
            rep(c2, crystals[2].n+1)
                rep(c3, crystals[3].n+1)
                    rep(c4, crystals[4].n+1){
                        // cout  << c0 << crystals[1].n+1 << c2 << c3 << c4 << "\n";
                        xsum = c0 * crystals[0].x + c1 * crystals[1].x + c2 * crystals[2].x + c3 * crystals[3].x + c4 * crystals[4].x;
                        ysum = c0 * crystals[0].y + c1 * crystals[1].y + c2 * crystals[2].y + c3 * crystals[3].y + c4 * crystals[4].y;
                        if (xsum == gx && ysum == gy)
                        {
                            s = 1;
                            s *= frac[c0+c1+c2+c3+c4];
                            s *= frac_inv[c0];
                            s %= mod;
                            s *= frac_inv[c1];
                            s %= mod;
                            s *= frac_inv[c2];
                            s %= mod;
                            s *= frac_inv[c3];
                            s %= mod;
                            s *= frac_inv[c4];
                            s %= mod;
                            ans += s;
                            ans %= mod;
                            // cout << s << "\n";
                            // cout << c0 << c1 << c2 << c3 << c4 << "\n";
                        }
                    }
    cout << ans << endl;
    return 0;
}
0