結果

問題 No.916 Encounter On A Tree
ユーザー rpy3cpprpy3cpp
提出日時 2019-11-17 01:48:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 1,830 bytes
コンパイル時間 1,579 ms
コンパイル使用メモリ 169,320 KB
実行使用メモリ 11,776 KB
最終ジャッジ日時 2024-09-25 05:44:53
合計ジャッジ時間 3,669 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 56
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

constexpr long long mod = 1e9+7;
long long fac[1 << 20];

void fill_fac(){
    fac[0] = 1;
    for (int i = 1; i < (1 << 20); ++i){
        fac[i] = fac[i - 1] * i % mod;
    }
}

int find_depth(int x){
    int d = 1;
    while (x > (1 << (d - 1))){
        x -= 1 << (d - 1);
        ++d;
    }
    return d;
}

int solve_core(int d, int dr, int dl, int k){
    if (dl < dr) swap(dl, dr);
    int s = (k - (dl - dr)) / 2;
    long long ans = 0;
    if (s == 0){
        if (dl == dr) return 0;
        ans = (1 << (dl - dr)) % mod;
        ans = ans * fac[(1 << (dl - 1)) - 1] % mod;
        for (int i = 1; i <= d; ++i){
            if (i == dl) continue;
            ans = ans * fac[1 << (i - 1)] % mod;
        }
        return ans;
    }else if (dl == dr){
        ans = 1 << (dl - 1);
        ans = ans * (1 << (s - 1)) % mod;
        ans = ans * fac[(1 << (dl - 1)) - 2] % mod;
        for (int i = 1; i <= d; ++i){
            if (i == dl) continue;
            ans = ans * fac[1 << (i - 1)] % mod;
        }
        return ans;
    }else{
        ans = 1 << (s - 1);
        ans = ans * fac[(1 << (dr - 1)) - 1] % mod;
        for (int i = 1; i <= d; ++i){
            if (i == dr) continue;
            ans = ans * fac[1 << (i - 1)] % mod;
        }
        return ans;
    }
}

int solve(int d, int l, int r, int k){
    int dl = find_depth(l);
    int dr = find_depth(r);
    if (abs(dl - dr) > k){
        return 0;
    }else if (dl + dr < k){
        return 0;
    }else if ((k - abs(dl - dr)) % 2){
        return 0;
    }else if ((k - abs(dl - dr)) / 2 > min(dl, dr) - 1){
        return 0;
    }
    return solve_core(d, dl, dr, k);
}

int main(){
    fill_fac();
    int d, l, r, k;
    cin >> d >> l >> r >> k;
    cout << solve(d, l, r, k) << endl;
    return 0;
}
0