#1016070 (C++23(gcc13)) No.1505 Zero-Product Ranges

• 問題: No.1505 Zero-Product Ranges
• ユーザー: kazuppa
• 提出日時: 2024-10-07 23:16:22
• 言語: C++23(gcc13) (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 16 ms / 2,000 ms
• コード長: 5,357 bytes
• コンパイル時間: 6,631 ms
• コンパイル使用メモリ: 338,656 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-07 23:16:32
• 合計ジャッジ時間: 8,392 ms
• ジャッジサーバーID （参考情報）: judge4 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 KB
testcase_03	AC	13 ms; 5,248 KB
testcase_04	AC	15 ms; 5,248 KB
testcase_05	AC	6 ms; 5,248 KB
testcase_06	AC	6 ms; 5,248 KB
testcase_07	AC	5 ms; 5,248 KB
testcase_08	AC	6 ms; 5,248 KB
testcase_09	AC	5 ms; 5,248 KB
testcase_10	AC	6 ms; 5,248 KB
testcase_11	AC	6 ms; 5,248 KB
testcase_12	AC	5 ms; 5,248 KB
testcase_13	AC	8 ms; 5,248 KB
testcase_14	AC	8 ms; 5,248 KB
testcase_15	AC	13 ms; 5,248 KB
testcase_16	AC	15 ms; 5,248 KB
testcase_17	AC	15 ms; 5,248 KB
testcase_18	AC	15 ms; 5,248 KB
testcase_19	AC	2 ms; 5,248 KB
testcase_20	AC	2 ms; 5,248 KB
testcase_21	AC	2 ms; 5,248 KB
testcase_22	AC	16 ms; 5,248 KB
testcase_23	AC	15 ms; 5,248 KB
testcase_24	AC	14 ms; 5,248 KB
testcase_25	AC	14 ms; 5,248 KB
testcase_26	AC	14 ms; 5,248 KB
testcase_27	AC	14 ms; 5,248 KB
testcase_28	AC	14 ms; 5,248 KB
testcase_29	AC	15 ms; 5,248 KB
testcase_30	AC	15 ms; 5,248 KB
testcase_31	AC	14 ms; 5,248 KB
testcase_32	AC	8 ms; 5,248 KB
testcase_33	AC	8 ms; 5,248 KB
testcase_34	AC	9 ms; 5,248 KB
testcase_35	AC	7 ms; 5,248 KB
testcase_36	AC	8 ms; 5,248 KB
testcase_37	AC	8 ms; 5,248 KB
testcase_38	AC	8 ms; 5,248 KB
testcase_39	AC	8 ms; 5,248 KB
testcase_40	AC	9 ms; 5,248 KB
testcase_41	AC	10 ms; 5,248 KB
testcase_42	AC	3 ms; 5,248 KB
testcase_43	AC	3 ms; 5,248 KB
testcase_44	AC	2 ms; 5,248 KB
testcase_45	AC	2 ms; 5,248 KB
testcase_46	AC	3 ms; 5,248 KB
testcase_47	AC	3 ms; 5,248 KB
testcase_48	AC	2 ms; 5,248 KB
testcase_49	AC	2 ms; 5,248 KB
testcase_50	AC	2 ms; 5,248 KB
testcase_51	AC	3 ms; 5,248 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
#include <atcoder/all>
using namespace atcoder;
using mints=modint998244353;
#pragma GCC target ("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define rep(i,a,b) for(it i=(it)(a);i<=(it)b;i++)
#define nrep(i,a,b) for(it i=(it)(a);i>=(it)b;i--)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define moda 998244353LL
#define modb 1000000007LL
#define gyaku 166374059LL
#define dai 2500000000000000000LL
#define sho -dai
#define aoi 1e+18
#define tyu 2500000000
#define giri 1000000000
#define en 3.14159265358979
#define eps 1e-14
#define endl '\n'
#define yn(x) cout<<(x?"Yes\n":"No\n");
#define YN(x) cout<<(x?"YES\n":"NO\n");
#define ci(x) cin>>x;
#define fi(x) cout<<fixed<<setprecision(x);
using it=long long;
using itn=int;
using un=unsigned long long;
using db=long double;
using st=string;
using ch=char;
using bo=bool;
using P=pair<it,it>;
using ip=pair<int,int>;
using mint=modint;
using minto=modint1000000007;
using vi=vector<it>;
using ivi=vector<int>;
using vd=vector<db>;
using vs=vector<st>;
using vc=vector<ch>;
using vb=vector<bo>;
using vp=vector<P>;
using ivp=vector<ip>;
using vm=vector<mint>;
using vms=vector<mints>;
using vmo=vector<minto>;
using sp=set<P>;
using isp=set<ip>;
using ss=set<st>;
//using sc=set<ch>;
using si=set<it>;
using isi=set<int>;
using svi=set<vi>;
using vvi=vector<vi>;
using ivvi=vector<ivi>;
using vvd=vector<vd>;
using vvs=vector<vs>;
using vvb=vector<vb>;
using vvc=vector<vc>;
using vvp=vector<vp>;
using ivvp=vector<ivp>;
using vvm=vector<vm>;
using vvms=vector<vms>;
using vvmo=vector<vmo>;
using vsi=vector<si>;
using ivsi=vector<isi>;
using vsp=vector<sp>;
using ivsp=vector<isp>;
using vvsi=vector<vsi>;
using ivvsi=vector<ivsi>;
using vvsp=vector<vsp>;
using ivvsp=vector<ivsp>;
using vvvi=vector<vvi>;
using ivvvi=vector<ivvi>;
using vvvd=vector<vvd>;
using vvvb=vector<vvb>;
using vvvm=vector<vvm>;
using vvvms=vector<vvms>;
using vvvmo=vector<vvmo>;
using vvvvi=vector<vvvi>;
using ivvvvi=vector<ivvvi>;
const it dx[4]={0,1,0,-1};
const it dy[4]={1,0,-1,0};

template<typename T>
void scan(vector<T> &a){
  rep(i,0,a.size()-1)cin>>a[i];
}

template<typename T>
void scan(vector<T> &a,vector<T>&b){
  rep(i,0,a.size()-1)cin>>a[i]>>b[i];
}

template<typename T>
void scan(vector<pair<T,T>> &a){
  rep(i,0,a.size()-1)cin>>a[i].first>>a[i].second;
}

template<typename T>
void dec(vector<T> &a){
  rep(i,0,a.size()-1)a[i]--;
}

it gcda(it a,it b){
  if(!a||!b){
    return max(a,b);
  }
  while(a%b&&b%a){
    if(a>b)a%=b;
    else b%=a;
  }
  return min(a,b);
}

it lcma(it a,it b){
  return a/gcda(a,b)*b;
}

bo su(ch c){return ('0'<=c&&c<='9');}

const int MOD = modb;
vi fact, fact_inv, inv;
/*  init_nCk :二項係数のための前処理
    計算量:O(n)
*/
void init_nCk(int SIZE){
  fact.resize(SIZE+5);
  fact_inv.resize(SIZE+5);
  inv.resize(SIZE+5);
  fact[0]=fact[1]=1;
  fact_inv[0]=fact_inv[1]=1;
  inv[1]=1;
  rep(i,2,SIZE+4){
    fact[i]=fact[i-1]*i%MOD;
    inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD;
    fact_inv[i]=fact_inv[i-1]*inv[i]%MOD;
  }
}
/*  nCk :MODでの二項係数を求める(前処理 int_nCk が必要)
    計算量:O(1)
*/
int nCk(int n, int k){
  assert(!(n<k));
  assert(!(n<0||k<0));
  return fact[n]*(fact_inv[k]*fact_inv[n-k]%MOD)%MOD;
}

struct dsu1{
  ivi par,siz;
  void init(int n){
    rep(i,0,n-1)par.emplace_back(i);
    rep(i,0,n-1)siz.emplace_back(1);
  }
  int leader(int u){
    if(par[u]==u)return u;
    return par[u]=leader(par[u]);
  }
  void merge(int u,int v){
    int ru=leader(u),rv=leader(v);
    if(ru==rv)return;
    if(ru>rv)swap(ru,rv);
    siz[ru]+=siz[rv];
    par[rv]=ru;
  }
  bool same(int u,int v){
    return leader(u)==leader(v);
  }
  int size(int u){
    return siz[leader(u)];
  }
};

struct dsu2{
  ivvi par,siz;int h,w;
  void init(int x,int y){
    h=x,w=y;par.resize(h),siz.resize(h);
    rep(i,0,h-1)
      rep(j,0,w-1)par[i].emplace_back(i*w+j);
    rep(i,0,h-1)
      rep(j,0,w-1)siz[i].emplace_back(1);
  }
  int leader(int x,int y){
    if(par[x][y]==x*par[0].size()+y)return x*w+y;
    return par[x][y]=leader(par[x][y]/w,par[x][y]%w);
  }
  void merge(int a,int b,int c,int d){
    int ru=leader(a,b),rv=leader(c,d);
    if(ru==rv)return;
    if(size(ru/w,ru%w)<size(rv/w,ru%w))swap(ru,rv);
    siz[ru/w][ru%w]+=siz[rv/w][rv%w];
    par[rv/w][rv%w]=ru;
  }
  bool same(int a,int b,int c,int d){
    return leader(a,b)==leader(c,d);
  }
  int size(int a,int b){
    int ru=leader(a,b);
    return siz[ru/w][ru%w];
  }
};

/*総和をもとめるセグ木
struct nod{
  mints val;
  int siz;
  nod(mints v=0,int s=0):val(v),siz(s){}
};

nod op(nod a,nod b){return nod(a.val+b.val,a.siz+b.siz);}
nod e(){return nod(0,0);}

struct act{
  mints a;
  act(mints e=0):a(e){}
};

nod mapping(act f,nod x){return nod(f.a+x.val,x.siz);}
act comp(act f,act g){return act(f.a+g.a);}
act id(){return act(0);}
*/

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  it n;cin>>n;
  ivi a(n);scan(a);
  ivi zero;zero.emplace_back(-1);
  rep(i,0,n-1)
    if(!a[i])zero.emplace_back(i);
  zero.emplace_back(n);
  it ans=0;
  rep(i,0,zero.size()-2){
    it e=zero[i+1]-zero[i];
    ans+=e*(e-1)/2;
  }
  cout<<n*(n+1)/2-ans<<endl;
}