結果

問題 No.1419 Power Moves
ユーザー maspymaspy
提出日時 2020-12-27 02:17:46
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 558 ms / 2,000 ms
コード長 718 bytes
コンパイル時間 94 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 52,744 KB
最終ジャッジ日時 2024-10-02 09:12:15
合計ジャッジ時間 21,416 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import numpy as np

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

MOD = 1_000_000_007

def main(N, K):
    NN = N * 2
    A = np.zeros(NN, np.int64)
    # 全部マイナスに進んだ場合の場所
    shift = (1 - pow(2, K, NN)) % NN
    # そこから、0, 2, 4, ..., 2^{K+1}-2 まで足せる
    # q, r = divmod(pow(2,K), NN)
    q, r = divmod(pow(2, K, N * MOD), N)
    A[:2 * r:2] += 1
    A[::2] += q
    A = np.roll(A, shift)
    # 全確率でわる
    x = np.sum(A) % MOD
    A *= pow(int(x), -1, MOD)
    A = A[:N] + A[N:]
    return A % MOD

N, K = map(int, read().split())
ans = main(N, K)
print('\n'.join(map(str, ans.tolist())))
0