結果

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

ソースコード

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(){
      int 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