結果

問題 No.1521 Playing Musical Chairs Alone
ユーザー kozykozy
提出日時 2021-05-28 22:15:34
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 512 ms / 2,000 ms
コード長 2,291 bytes
コンパイル時間 328 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 44,696 KB
最終ジャッジ日時 2024-04-25 00:20:08
合計ジャッジ時間 16,779 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 494 ms
44,324 KB
testcase_01 AC 501 ms
44,568 KB
testcase_02 AC 491 ms
44,696 KB
testcase_03 AC 494 ms
44,568 KB
testcase_04 AC 492 ms
44,320 KB
testcase_05 AC 500 ms
44,180 KB
testcase_06 AC 498 ms
44,696 KB
testcase_07 AC 502 ms
44,436 KB
testcase_08 AC 486 ms
44,576 KB
testcase_09 AC 494 ms
44,688 KB
testcase_10 AC 488 ms
44,568 KB
testcase_11 AC 495 ms
44,184 KB
testcase_12 AC 494 ms
44,308 KB
testcase_13 AC 503 ms
44,184 KB
testcase_14 AC 500 ms
44,184 KB
testcase_15 AC 508 ms
44,184 KB
testcase_16 AC 502 ms
44,188 KB
testcase_17 AC 510 ms
44,564 KB
testcase_18 AC 504 ms
44,184 KB
testcase_19 AC 512 ms
44,564 KB
testcase_20 AC 511 ms
44,184 KB
testcase_21 AC 501 ms
44,312 KB
testcase_22 AC 500 ms
44,580 KB
testcase_23 AC 502 ms
44,612 KB
testcase_24 AC 502 ms
44,188 KB
testcase_25 AC 493 ms
44,192 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
mod=10**9+7
inv2=499122177
def fft_len(N):
    fftlen=1
    beki=0
    while fftlen<N:
      fftlen*=2
      beki+=1
    return fftlen,beki
def FPStime(L,R):
  fftlen,beki=fft_len(len(L)+len(R))
  FS=np.fft.rfft(L, fftlen)
  FT=np.fft.rfft(R, fftlen)
  Fh=FS*FT
  h = np.fft.irfft(Fh, fftlen)
  h = np.rint(h).astype(np.int64)
  return h[:len(L) + len(R) - 1]
def FPStime2(L,R,mod):
  f1,f2=np.divmod(L,1<<15)
  g1,g2=np.divmod(R,1<<15)
  a=FPStime(f1,g1)%mod
  c=FPStime(f2,g2)%mod
  b=(FPStime(f1+f2,g1+g2)-(a+c))%mod
  h=(a<<30)+(b<<15)+c
  return h%mod
def FPSinv(f):
  """
  1/f=gをreturn
  """
  f=np.array(f, np.int64)
  this_len,this_roop=fft_len(len(f))
  c=pow(int(f[0]),-1,mod)
  g=np.array([c], np.int64)
  a=1
  for i in range(this_roop):
      a*=2
      S=FPStime2(g,f,mod)[:a]
      S*=-1
      S[0]+=2
      S=[i%mod for i in S]
      g=FPStime2(g,S,mod)
      g=g[:a]
      g=[i%mod for i in g]
  return g[:len(f)+1]
def FPSsqrt(f):
  """
  sqrt(f)=gをreturn
  """
  f=np.array(f, np.float64)
  g=np.array([f[0]**0.5], np.float64)
  this_len,this_roop=fft_len(len(f))
  a=1
  for i in range(this_roop):
      a*=2
      g=FPSplus(FPStime(f,FPSinv(g)),g)
      g=g*inv2
      g=g[:a]
      g=[i%mod for i in g]
  return g[:len(f)]
def keisu(P,Q,N):#P(x)/Q(x)のx^Nの係数
  Q2=list()
  for i in range(len(Q)):
    if N==0:
      return P[0]
    if i%2==0:
      Q2.append(Q[i])
    else:
      Q2.append(-Q[i])
  S=FPStime2(Q,Q2,mod)
  V=list()
  for i in range(0,len(S),2):
    V.append(S[i])
  Ue=list()
  Uo=list()
  P=FPStime2(P,Q2,mod)
  for i in range(len(P)):
    if i%2==0:
      Ue.append(P[i])
    else:
      Uo.append(P[i])
  if N%2==0:
    return keisu(Ue,V,N//2)
  else:
    return keisu(Uo,V,N//2)
from functools import lru_cache

@lru_cache(maxsize=1000000)
def beki(a):
  if a==1:
    return f
  if a%2==0:
    L=FPStime2(beki(a//2),beki(a//2),mod)
  else:
    L=FPStime2(FPStime2(beki(a//2),beki(a//2),mod),f,mod)
  C=[0]*(N)
  for i in range(len(L)):
    C[i%(N)]+=L[i]
    C[i%N]%=mod
  return C
N,K,L=map(int,input().split())
f=[0]+[1]*L
f=f+[0]*(N-len(f))
if N==L:
  f=[1]*N
ansl=beki(K)
for i in ansl:
  print(i)
0