結果

問題 No.916 Encounter On A Tree
ユーザー stng
提出日時 2022-08-14 22:16:20
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,553 bytes
コンパイル時間 442 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 75,136 KB
最終ジャッジ日時 2024-10-01 13:27:37
合計ジャッジ時間 4,949 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 44 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

d,L,R,k = map(int,input().split())

MOD = 10**9+7
N = 2**d

fact = [1]*(N+1)
rfact = [1]*(N+1)
r = 1
for i in range(1, N+1):
    fact[i] = r = r * i % MOD
rfact[N] = r = pow(fact[N], MOD-2, MOD)
for i in range(N, 0, -1):
    rfact[i-1] = r = r * i % MOD

# nPk (mod MOD) を求める
def perm(n, k):
    return fact[n] * rfact[n-k] % MOD

rui = []
val = 1
for i in range(d):
    rui.append(val)
    if L >= val:
        lidx = i
    if R >= val:
        ridx = i
    val *= 2

ans = 1

f = False
if lidx == ridx:
    f = True
    if k % 2 != 0 or k > 2*lidx:
        print(0)
        exit()
else:
    if k % 2 != (ridx-lidx) % 2 or k > lidx+ridx or ridx-lidx > k:
        print(0)
        exit()

if f == True:
    for i in range(d):
        if i == lidx:
            tmp = pow(2,i,MOD)
            if tmp == 2:
                ans *= tmp*1*perm(tmp-2,tmp-2)
            else:
                ans *= tmp*2*perm(tmp-2,tmp-2)
            ans %= MOD
            continue
        tmp = pow(2,i,MOD)
        ans *= perm(tmp,tmp)
        ans %= MOD
    print(ans%MOD)
else:
    for i in range(d):
        if i == lidx:
            tmp = pow(2,i,MOD)
            if tmp != 1:
                aa = k // 2
                if k == ridx-lidx:
                    ans *= perm(tmp-1,tmp-1)
                else:
                    ans *= (2**(aa-1))*perm(tmp-1,tmp-1)
            else:
                ans *= perm(tmp-1,tmp-1)
            ans %= MOD
            continue
        tmp = pow(2,i,MOD)
        ans *= perm(tmp,tmp)
        ans %= MOD
    print(ans%MOD)
0