結果

問題 No.2230 Good Omen of White Lotus
ユーザー wolgnikwolgnik
提出日時 2023-02-24 22:48:03
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,823 bytes
コンパイル時間 144 ms
コンパイル使用メモリ 82,792 KB
実行使用メモリ 106,608 KB
最終ジャッジ日時 2024-09-13 05:53:27
合計ジャッジ時間 11,264 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 WA * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
H, W, N, P = map(int, input().split())
mod = 998244353
a = [[] for _ in range(H + 1)]

for _ in range(N):
  x, y = map(int, input().split())
  a[x].append(y)

class SegTree:
  def segfunc(self, x, y):
    return max(x, y)
  def __init__(self, n, ide_ele, init_val):
    #####単位元######
    self.ide_ele = ide_ele
    #num:n以上の最小の2のべき乗
    self.num = 2 ** (n - 1).bit_length()
    self.seg = [self.ide_ele] * 2 * self.num
    #set_val
    for i in range(n):
      self.seg[i + self.num - 1] = init_val[i]    
    #built
    for i in range(self.num - 2, -1, -1) :
      self.seg[i] = self.segfunc(self.seg[2 * i + 1], self.seg[2 * i + 2]) 
  def update(self, k, x):
    k += self.num - 1
    self.seg[k] = x
    while k + 1:
      k = (k - 1) // 2
      self.seg[k] = self.segfunc(self.seg[k * 2 + 1], self.seg[k * 2 + 2]) 
  def query(self, p, q):
    if q <= p:
      return self.ide_ele
    p += self.num - 1
    q += self.num - 2
    res = self.ide_ele
    while q - p > 1:
      if p & 1 == 0:
        res = self.segfunc(res, self.seg[p])
      if q & 1 == 1:
        res = self.segfunc(res, self.seg[q])
        q -= 1
      p = p // 2
      q = (q - 1) // 2
    if p == q:
      res = self.segfunc(res, self.seg[p])
    else:
      res = self.segfunc(self.segfunc(res, self.seg[p]), self.seg[q])
    return res

seg = SegTree(W + 1, 0, [0] * (W + 1))

for x in range(H, -1, -1):
  s = []
  a[x].sort(reverse = True)
  for y in a[x]:
    q = seg.query(y, W + 1)
    s.append((y, q + 1))
  for y, upd in s: seg.update(y, upd)

  #print(s)
q = seg.query(0, W + 1)
res = pow(P - 2, q, mod) * pow(P, (mod - 2) * q, mod)
res %= mod
res *= pow(P - 1, (H + W - 3 - q), mod) * pow(P, (mod - 2) * (H + W - 3 - q), mod)
res %= mod
res = 1 - res
res %= mod
print(res)
0