結果

問題 No.2206 Popcount Sum 2
ユーザー とりゐとりゐ
提出日時 2023-01-16 02:08:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,149 ms / 4,000 ms
コード長 1,126 bytes
コンパイル時間 174 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 124,168 KB
最終ジャッジ日時 2024-07-02 14:38:32
合計ジャッジ時間 15,203 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 75 ms
88,320 KB
testcase_01 AC 71 ms
88,320 KB
testcase_02 AC 166 ms
103,944 KB
testcase_03 AC 166 ms
104,064 KB
testcase_04 AC 174 ms
103,680 KB
testcase_05 AC 1,023 ms
123,132 KB
testcase_06 AC 1,030 ms
122,868 KB
testcase_07 AC 1,026 ms
123,008 KB
testcase_08 AC 1,060 ms
123,128 KB
testcase_09 AC 1,149 ms
123,604 KB
testcase_10 AC 670 ms
122,508 KB
testcase_11 AC 685 ms
123,332 KB
testcase_12 AC 678 ms
122,944 KB
testcase_13 AC 620 ms
123,884 KB
testcase_14 AC 625 ms
124,168 KB
testcase_15 AC 643 ms
123,584 KB
testcase_16 AC 501 ms
123,456 KB
testcase_17 AC 479 ms
123,500 KB
testcase_18 AC 521 ms
123,580 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

mod=998244353
M=(10**5)*3 
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

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


N=2*10**5+10
q=int(input())
M=int(q**0.5)+1
bucket=[[] for i in range(M)]
ans=[0]*q
for i in range(q):
  n,m=map(int,input().split())
  ans[i]=pow2[n]-1
  n-=1
  m-=1
  bucket[m*M//N].append((m,n,i))

for i in range(M):
  if i&1:
    bucket[i].sort(key=lambda x:-x[1])
  else:
    bucket[i].sort(key=lambda x:x[1])

res=1
n=1
m=0
inv2=pow(2,mod-2,mod)

for b in bucket:
  for mi,ni,i in b:
    while m<mi:
      res+=binom(n,m+1)
      res%=mod
      m+=1
    while m>mi:
      res-=binom(n,m)
      res%=mod
      m-=1
    while n<ni:
      res=2*res-binom(n,m)
      res%=mod
      n+=1
    while n>ni:
      res=inv2*(res+binom(n-1,m))
      res%=mod
      n-=1
    ans[i]*=res
    ans[i]%=mod

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