結果
問題 | No.2792 Security Cameras on Young Diagram |
ユーザー |
|
提出日時 | 2024-06-21 22:57:40 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 66 ms / 2,000 ms |
コード長 | 658 bytes |
コンパイル時間 | 2,272 ms |
コンパイル使用メモリ | 199,180 KB |
最終ジャッジ日時 | 2025-02-22 00:02:23 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h> #define rep(i,n) for(int i=0;i<(n);++i) using namespace std; using ll = long long; #include <atcoder/modint> using mint = atcoder::modint998244353; int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int n; cin >> n; vector<int> a(n); rep(i,n) cin >> a[i]; vector<mint> p(101010,1); rep(i,101010-1) p[i+1] *= p[i]*(i+1); mint ans = 1; rep(i,n){ if(a[i]==1){ ans += 1; continue; } int m = i+a[i]+1; int k = i+2; ans += p[m]/p[m-k]/p[k]; ans -= p[m-1]/p[m-1-k]/p[k]; } cout << ans.val() << endl; }