結果

問題 No.2898 Update Max
コンテスト
ユーザー convexineq
提出日時 2024-09-23 23:15:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 189 ms / 2,000 ms
コード長 2,559 bytes
コンパイル時間 3,422 ms
コンパイル使用メモリ 81,996 KB
実行使用メモリ 123,488 KB
最終ジャッジ日時 2024-09-23 23:20:40
合計ジャッジ時間 12,451 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

class BIT: #0-indexed
    __slots__ = ["size", "tree","depth","n0"]
    def __init__(self, n):
        self.size = n
        self.tree = [0]*(n+1)
        self.depth = n.bit_length()
        self.n0 = 1<<self.depth

    def get_sum(self, i): #a_0 + ... + a_{i} #閉区間
        s = 0; i += 1
        while i > 0:
            s += self.tree[i]
            i -= i & -i
        return s

    def range_sum(self,l,r): #a_l + ... + a_r 閉区間
        return self.get_sum(r) - self.get_sum(l-1) 

    def range_sum_larger(self,l): #a_l + ... (端まで)
        return self.get_sum(self.size-1) - (self.get_sum(l-1) if l else 0)
    
    def add(self, i, x):
        i += 1
        while i <= self.size:
            self.tree[i] += x
            i += i & -i

SIZE=5*10**5+1; MOD=998244353 #10**9+7 #ここを変更する

inv = [0]*SIZE  # inv[j] = j^{-1} mod MOD
fac = [0]*SIZE  # fac[j] = j! mod MOD
finv = [0]*SIZE # finv[j] = (j!)^{-1} mod MOD
fac[0] = fac[1] = 1
finv[0] = finv[1] = 1
for i in range(2,SIZE):
    fac[i] = fac[i-1]*i%MOD
finv[-1] = pow(fac[-1],MOD-2,MOD)
for i in range(SIZE-1,0,-1):
    finv[i-1] = finv[i]*i%MOD
    inv[i] = finv[i]*fac[i-1]%MOD

def choose(n,r): # nCk mod MOD の計算
    if 0 <= r <= n:
        return (fac[n]*finv[r]%MOD)*finv[n-r]%MOD
    else:
        return 0
def perm(n,r): # nPr mod MOD
    if 0 <= r <= n:
        return fac[n]*finv[n-r]%MOD
    else:
        return 0

import sys
readline = sys.stdin.readline

#n = int(readline())
#n,Q = map(int,readline().split())

n = int(readline())
*a, = map(int,readline().split())

# from itertools import permutations
# cnt = [0]*n
# for lst in permutations(list(range(1,n+1))):
#     for i in range(n):
#         if a[i] != -1 and lst[i] != a[i]: break
#     else:
#         m = 0
#         for i in range(n):
#             m = max(m,lst[i])
#             if m==lst[i]: cnt[i] += 1
# print(cnt)


bit = BIT(n+1)
for i in range(1,n+1):
    bit.add(i,1)
    
for ai in a:
    if ai != -1:
        bit.add(ai,-1)


tot = a.count(-1)
ans = 0
m = 0
cnt = 0
for i,ai in enumerate(a):
    if ai == -1:
        cnt += 1
        c = bit.get_sum(m)
        r = (fac[tot]-perm(c,cnt)*fac[tot-cnt])%MOD*inv[cnt]
        #print((perm(m-i-1+cnt,cnt)*fac[tot-cnt])%MOD,r%MOD)
        #print(r%MOD)
        ans = (ans+r)%MOD
    else:
        if m < ai:
            c = bit.get_sum(ai)
            r = perm(c,cnt)*fac[tot-cnt]
            #print(r)
            ans = (ans+r)%MOD
            m = ai        
        else:
            r = 0
        #print(r%MOD)


print(ans)
0