結果

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

ソースコード

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;

ui64 dist(ui64 a, ui64 b) {
    vector<ui64> v1, v2;
    while (a > 0) {
        v1.emplace_back(a);
        a >>= 1;
    }
    while (b > 0) {
        v2.emplace_back(b);
        b >>= 1;
    }
    for (int i = 0; i < v1.size(); ++i) {
        for (int j = 0; j < v2.size(); ++j) {
            if (v1[i] == v2[j]) {
                return i + j;
            }
        }
    }
    return 0;
}

signed main() {

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

    if (K > D * 2) {
        cout << 0 << 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]) %= MOD;
        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;
        {
            int x = 1 << (dl - 1);
            for (int y = (1 << (dr - 1)); y < (1 << dr); ++y) {
                if (dist(x, y) == K) ++res;
            }
        }
        (res *= dep[dl]) %= MOD;
        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