#990843 (C++23) No.2792 Security Cameras on Young Diagram

• 問題: No.2792 Security Cameras on Young Diagram
• ユーザー: たたき@競プロ
• 提出日時: 2024-06-26 09:48:30
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 36 ms / 2,000 ms
• コード長: 1,526 bytes
• コンパイル時間: 5,456 ms
• コンパイル使用メモリ: 311,544 KB
• 実行使用メモリ: 5,720 KB
• 最終ジャッジ日時: 2024-06-26 09:48:44
• 合計ジャッジ時間: 6,529 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	3 ms; 5,376 KB
testcase_06	AC	3 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	22 ms; 5,544 KB
testcase_13	AC	16 ms; 5,376 KB
testcase_14	AC	22 ms; 5,404 KB
testcase_15	AC	12 ms; 5,408 KB
testcase_16	AC	24 ms; 5,548 KB
testcase_17	AC	29 ms; 5,624 KB
testcase_18	AC	10 ms; 5,388 KB
testcase_19	AC	10 ms; 5,384 KB
testcase_20	AC	15 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB
testcase_22	AC	2 ms; 5,376 KB
testcase_23	AC	36 ms; 5,720 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint=modint998244353; //1000000007;
using ll=long long;
using pp=pair<int,int>;
#define sr string 
#define vc vector
#define fi first
#define se second
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define pb push_back
#define all(v) v.begin(),v.end()
#define pque priority_queue
#define bpc(a) __builtin_popcount(a)
struct modinv {
  int n; vector<mint> d;
  modinv(): n(2), d({0,1}) {}
  mint operator()(int i) {
    while (n <= i) d.push_back(-d[mint::mod()%n]*(mint::mod()/n)), ++n;
    return d[i];
  }
  mint operator[](int i) const { return d[i];}
} invs;
struct modfact {
  int n; vector<mint> d;
  modfact(): n(2), d({1,1}) {}
  mint operator()(int i) {
    while (n <= i) d.push_back(d.back()*n), ++n;
    return d[i];
  }
  mint operator[](int i) const { return d[i];}
} facts;
struct modfactinv {
  int n; vector<mint> d;
  modfactinv(): n(2), d({1,1}) {}
  mint operator()(int i) {
    while (n <= i) d.push_back(d.back()*invs(n)), ++n;
    return d[i];
  }
  mint operator[](int i) const { return d[i];}
} ifacts;
mint comb(int n, int k) {
  if (n < k || k < 0) return 0;
  return facts(n)*ifacts(k)*ifacts(n-k);
} 
int main(){
  int n;cin>>n;
  vc<int>v(n); rep(i,n)cin>>v[i];
  mint ans=0;
  rep(i,n){
    ans+=comb(i+v[i]-1,i);
  }
  int now=n-1,t=0;
  vc<int>x;
  while(now>=0){
    if(t==v[now])now--;
    else {
      x.pb(now);
      t++;
    }
  }
 
  rep(i,x.size())ans+=comb(i+x[i],i);
  cout<<ans.val();
}