結果

問題 No.1505 Zero-Product Ranges
ユーザー abc3
提出日時 2021-05-14 22:17:50
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 60 ms / 2,000 ms
コード長 887 bytes
コンパイル時間 3,620 ms
コンパイル使用メモリ 192,824 KB
最終ジャッジ日時 2025-01-21 11:23:32
ジャッジサーバーID
(参考情報)
judge1 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #

    #include <bits/stdc++.h>
    using namespace std;
    #include <math.h>
    #include <iomanip>
    #include <cstdint>
    #include <string>
    #include <sstream>
    template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
    template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
    #define rep(i,n) for (int i = 0; i < (n); ++i)
    typedef long long ll;
    using P=pair<ll,ll>;
    const int INF=1001001001;
    const int mod=1e9+7;
    
    void solve(){
      ll n;
      cin>>n;
      vector<int>a(n);
      rep(i,n){cin>>a[i];}
      ll ans=n*(n+1)/2;
      int i=0;
      while(i<n){
        while(i<n&&a[i]==0){i++;}
        ll cnt=0;
        while(i<n&&a[i]==1){i++;cnt++;}
        ans-=cnt*(cnt+1)/2;
      }
      cout<<ans<<endl;
    }
    int main(){
        solve();
        return 0;
    }
0