結果

問題 No.1683 Robot Guidance
ユーザー puzneko
提出日時 2021-11-13 00:11:43
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,927 bytes
コンパイル時間 397 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 115,072 KB
最終ジャッジ日時 2024-11-26 00:12:48
合計ジャッジ時間 23,038 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 37 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

def powmod(n,pow,mod):
    val = 1
    while pow > 0:
        if pow & 1:
            val = (val * n) % mod
        pow = pow >> 1
        n = (n * n) % mod
    return val

n, m, x, y = map(int, input().split())
p = 10 ** 9 + 7
kaijo = [1 for i in range(n+m//2+2)]
invkaijo = [1 for i in range(n+m//2+2)]
for i in range(1,n+m//2+2):
    kaijo[i] = kaijo[i-1] * i % p
    invkaijo[i] = powmod(kaijo[i],p-2,p)

pmnum = [m // 4 for i in range(4)]
for i in range(m%4+1):
    pmnum[i] += 1

xans = [0 for i in range(n+1)]
yans = [0 for i in range(n+1)]

if x >= 0:
    if pmnum[2] >= 1:
        for i in range((n-x)//2+1):
            xans[x+i*2] = kaijo[x+i+pmnum[0]-1] * invkaijo[x+i] % p * invkaijo[pmnum[0]-1] % p
            xans[x+i*2] = xans[x+i*2] * kaijo[i+pmnum[2]-1] % p * invkaijo[i] % p * invkaijo[pmnum[2]-1] % p
    else:
        xans[x] = 1
else:
    x = -x
    if pmnum[2] >= 1:
        for i in range((n-x)//2+1):
            xans[x+i*2] = kaijo[x+i+pmnum[2]-1] * invkaijo[x+i] % p * invkaijo[pmnum[2]-1] % p
            xans[x+i*2] = xans[x+i*2] * kaijo[i+pmnum[0]-1] % p * invkaijo[i] % p * invkaijo[pmnum[0]-1] % p

if pmnum[1] >= 1:
    if y >= 0:
        if pmnum[3] >= 1:
            for i in range((n-y)//2+1):
                yans[y+i*2] = kaijo[y+i+pmnum[1]-1] * invkaijo[y+i] % p * invkaijo[pmnum[1]-1] % p
                yans[y+i*2] = yans[y+i*2] * kaijo[i+pmnum[3]-1] % p * invkaijo[i] % p * invkaijo[pmnum[3]-1] % p
        else:
            yans[y] = 1
    else:
        y = -y
        if pmnum[3] >= 1:
            for i in range((n-y)//2+1):
                yans[y+i*2] = kaijo[y+i+pmnum[3]-1] * invkaijo[y+i] % p * invkaijo[pmnum[3]-1] % p
                yans[y+i*2] = yans[y+i*2] * kaijo[i+pmnum[1]-1] % p * invkaijo[i] % p * invkaijo[pmnum[1]-1] % p
else:
    if y == 0:
        yans[0] = 1

ans = 0
for i in range(n+1):
    ans = (ans + xans[i] * yans[n-i] % p) % p

print("{}".format(ans))

0