結果

問題 No.916 Encounter On A Tree
ユーザー もりを
提出日時 2019-10-25 23:02:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,939 bytes
コンパイル時間 1,725 ms
コンパイル使用メモリ 172,468 KB
実行使用メモリ 19,780 KB
最終ジャッジ日時 2024-09-13 06:56:00
合計ジャッジ時間 3,989 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 45 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using i64 = int_fast64_t;
using ui64 = uint_fast64_t;
#define REP(i, n) for (i64 (i) = 0; (i) < (n); ++(i))
#define FOR(i, a, b) for (i64 (i) = (a); (i) < (b); ++(i))

constexpr int MOD = (int)1e9 + 7;
ui64 MOP(ui64 x, ui64 n, ui64 m = MOD) {
    ui64 r = 1;
    while (n > 0) {
        if (n & 1) (r *= x) %= m;
        (x *= x) %= m;
        n >>= 1;
    }
    return r;
}

int D, L, R, K;
vector<ui64> fact, dep;

signed main() {

    cin >> D >> L >> R >> K;

    if (K > D * 2) {
        cout << -1 << endl;
        return 0;
    }
    
    dep.resize(D + 1);
    dep[0] = dep[1] = 1;
    REP(i, D - 1) dep[i + 2] = dep[i + 1] * 2;

    fact.resize(2100100);
    fact[0] = fact[1] = 1;
    REP(i, 2100108) fact[i + 2] = (fact[i + 1] * (i + 2)) % MOD;
    
    ui64 dl = 0, dr = 0;
    {
        int tmp = L;
        while (tmp > 0) {
            ++dl;
            tmp >>= 1;
        }
        tmp = R;
        while (tmp > 0) {
            ++dr;
            tmp >>= 1;
        }
    }

    ui64 res = 0;
    if (dl == dr) {
        if (K % 2 == 1) {
            cout << 0 << endl;
            return 0;
        }
        K >>= 1;
        res = dep[K];
        REP(i, K) (res <<= 1) %= MOD;
        res *= dep[dl - K];
        for (int i = 1; i <= D; ++i) {
            if (i == dl) {
                res *= fact[dep[i] - 2];
            } else {
                res *= fact[dep[i]];
            }
            res %= MOD;
        }
    } else {
        int dif = dr - dl;
        if (K - dif % 2 == 1) {
            cout << 0 << endl;
            return 0;
        }
        int k = (K - dif) >> 1;
        res = dep[k] * dep[dr];
        for (int i = 1; i <= D; ++i) {
            if (i == dl or i == dr) {
                res *= fact[dep[i] - 1];
            } else {
                res *= fact[dep[i]];
            }
            res %= MOD;
        }
    }

    cout << res << endl;

}
0