結果
| 問題 | No.1765 While Shining | 
| コンテスト | |
| ユーザー |  umezo | 
| 提出日時 | 2021-11-26 23:31:35 | 
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) | 
| 結果 | 
                                AC
                                 
                             | 
| 実行時間 | 18 ms / 2,000 ms | 
| コード長 | 518 bytes | 
| コンパイル時間 | 1,972 ms | 
| コンパイル使用メモリ | 193,920 KB | 
| 最終ジャッジ日時 | 2025-01-26 01:54:48 | 
| ジャッジサーバーID (参考情報) | judge5 / judge5 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 22 | 
ソースコード
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;
 
#include<bits/stdc++.h>
using namespace std;
int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  
  ll n;
  cin>>n;
  vector<int> A(n);
  rep(i,n) cin>>A[i];
  
  ll ans=0;
  ll now=0;
  for(int i=n-1;i>=0;i--){
    if(i==n-1) now=1;
    else{
      if(A[i]!=A[i+1]) now++;
      else now=1;
      
      if(A[i]==0) continue;
      ans+=min(now,n-1-i);
    }
  }
  cout<<ans<<endl;
  return 0;
}
            
            
            
        