結果

問題 No.2206 Popcount Sum 2
ユーザー とりゐ
提出日時 2023-01-17 02:53:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,565 ms / 4,000 ms
コード長 843 bytes
コンパイル時間 340 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 461,312 KB
最終ジャッジ日時 2024-07-02 14:39:39
合計ジャッジ時間 30,523 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import stdin
input=lambda :stdin.readline()[:-1]

mod=998244353
M=(10**5)*3+1 
fac=[1]*M
ninv=[1]*M
finv=[1]*M
for i in range(2,M):
  fac[i]=fac[i-1]*i%mod
  ninv[i]=(-(mod//i)*ninv[mod%i])%mod
  finv[i]=finv[i-1]*ninv[i]%mod

def binom(n,k):
  if n<0 or k<0:
    return 0
  if k>n:
    return 0
  return (fac[n]*finv[k]%mod)*finv[n-k]%mod

N=2*10**5+10
B=450

memo=[]
for i in range(B):
  a=[0]*(B*i+1)
  res=0
  for j in range(B*i+1):
    res+=binom(B*i,j)
    res%=mod
    a[j]=res
  memo.append(a)

inv2=pow(2,mod-2,mod)
pow2=[1]
for _ in range(M):
  pow2.append(pow2[-1]*2%mod)

T=int(input())
ans=[0]*T
for i in range(T):
  n,m=map(lambda x:int(x)-1,input().split())
  n0=(n+B-1)//B*B
  res=memo[n0//B][m]
  while n0>n:
    res=inv2*(res+binom(n0-1,m))
    res%=mod
    n0-=1
  ans[i]=res*(pow2[n+1]-1)%mod

print(*ans,sep='\n')
0