結果

問題 No.1765 While Shining
コンテスト
ユーザー monnu
提出日時 2021-11-26 22:32:15
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 32 ms / 2,000 ms
コード長 557 bytes
コンパイル時間 3,618 ms
コンパイル使用メモリ 230,380 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-29 18:12:23
合計ジャッジ時間 4,497 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define INF 1000000000000000000
#define MOD 998244353
#define MAX 1000000

int main(){
  int N;
  cin>>N;
  vector<int> A(N);
  for(int i=0;i<N;i++){
    cin>>A[i];
  }
  vector<int> sum0(N,0),sum1(N,0);
  sum0[N-1]=0;
  sum1[N-1]=0;
  ll ans=0;
  for(int i=N-2;i>=0;i--){
    if(A[i]==1){
      sum1[i]=sum0[i+1]+1;
      ans+=(ll)sum1[i];
    }else{
      sum0[i]=sum1[i+1]+1;
    }
  }
  cout<<ans<<'\n';
}
0