結果
問題 | No.2792 Security Cameras on Young Diagram |
ユーザー | mattu34 |
提出日時 | 2024-06-21 23:49:08 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 2,805 bytes |
コンパイル時間 | 302 ms |
コンパイル使用メモリ | 82,608 KB |
実行使用メモリ | 122,924 KB |
最終ジャッジ日時 | 2024-06-21 23:49:17 |
合計ジャッジ時間 | 8,613 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 311 ms
101,820 KB |
testcase_01 | AC | 277 ms
101,500 KB |
testcase_02 | AC | 262 ms
101,712 KB |
testcase_03 | AC | 267 ms
101,568 KB |
testcase_04 | AC | 267 ms
101,740 KB |
testcase_05 | AC | 261 ms
101,804 KB |
testcase_06 | AC | 260 ms
101,628 KB |
testcase_07 | AC | 258 ms
101,704 KB |
testcase_08 | AC | 259 ms
101,776 KB |
testcase_09 | AC | 290 ms
101,604 KB |
testcase_10 | AC | 258 ms
101,768 KB |
testcase_11 | AC | 260 ms
101,800 KB |
testcase_12 | AC | 303 ms
122,212 KB |
testcase_13 | AC | 321 ms
118,772 KB |
testcase_14 | RE | - |
testcase_15 | AC | 289 ms
116,520 KB |
testcase_16 | AC | 298 ms
122,924 KB |
testcase_17 | RE | - |
testcase_18 | AC | 285 ms
116,532 KB |
testcase_19 | AC | 283 ms
116,468 KB |
testcase_20 | AC | 291 ms
118,064 KB |
testcase_21 | AC | 253 ms
101,908 KB |
testcase_22 | AC | 256 ms
101,752 KB |
ソースコード
from collections import * import sys import heapq import bisect import itertools from functools import lru_cache from types import GeneratorType from fractions import Fraction import math import copy import random # sys.setrecursionlimit(int(1e7)) # @lru_cache(maxsize=None) # CPython特化 # @bootstrap # PyPy特化(こっちのほうが速い) yield dfs(), yield Noneを忘れずに def bootstrap(f, stack=[]): # yield def wrappedfunc(*args, **kwargs): if stack: return f(*args, **kwargs) else: to = f(*args, **kwargs) while True: if type(to) is GeneratorType: stack.append(to) to = next(to) else: stack.pop() if not stack: break to = stack[-1].send(to) return to return wrappedfunc dxdy1 = ((0, 1), (0, -1), (1, 0), (-1, 0)) # 上下右左 dxdy2 = ( (0, 1), (0, -1), (1, 0), (-1, 0), (1, 1), (-1, -1), (1, -1), (-1, 1), ) # 8方向すべて dxdy3 = ((0, 1), (1, 0)) # 右 or 下 dxdy4 = ((1, 1), (1, -1), (-1, 1), (-1, -1)) # 斜め INF = float("inf") _INF = 1 << 60 MOD = 998244353 mod = 998244353 MOD2 = 10**9 + 7 mod2 = 10**9 + 7 # memo : len([a,b,...,z])==26 # memo : 2^20 >= 10^6 # 小数の計算を避ける : x/y -> (x*big)//y ex:big=10**9 # @:小さい文字, ~:大きい文字,None: 空の文字列 # ユークリッドの互除法:gcd(x,y)=gcd(x,y-x) # memo : d 桁以下の p 進表記を用いると p^d-1 以下のすべての # 非負整数を表現することができる # memo : (X,Y) -> (X+Y,X−Y) <=> 点を原点を中心に45度回転し、√2倍に拡大 # memo : (x,y)のx正から見た偏角をラジアンで(-πからπ]: math.atan2(y, x) # memo : a < bのとき ⌊a⌋ ≦ ⌊b⌋ input = lambda: sys.stdin.readline().rstrip() mi = lambda: map(int, input().split()) li = lambda: list(mi()) ii = lambda: int(input()) py = lambda: print("Yes") pn = lambda: print("No") pf = lambda: print("First") ps = lambda: print("Second") # 階乗 & 逆元計算 factorial = [1] inverse = [1] for i in range(1, 10**5 + 1): factorial.append(factorial[-1] * i % MOD) inverse.append(pow(factorial[-1], MOD - 2, MOD)) # 組み合わせ計算 def nCr(n, r): if n < r or r < 0: return 0 elif r == 0: return 1 return factorial[n] * inverse[r] % MOD * inverse[n - r] % MOD N = ii() A = li() cnt = defaultdict(int) for a in A: cnt[a] += 1 ans = 0 for i in range(N): ans += nCr(i + A[i] - 1, i) ans %= MOD tmp = N for x in range(1, 10**5 + 1): ans += nCr(tmp + x - 2, x - 1) ans %= MOD tmp -= cnt[x] if tmp == 0: break print(ans)