結果

問題 No.1765 While Shining
ユーザー suzuken_w
提出日時 2021-11-26 23:29:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 1,292 bytes
コンパイル時間 2,278 ms
コンパイル使用メモリ 213,384 KB
最終ジャッジ日時 2025-01-26 01:54:21
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC optimize("O3")
#include<bits/stdc++.h> 
using namespace std;
using ll=long long;
using P=pair<ll,ll>;
template<class T> using V=vector<T>; 
#define fi first
#define se second
#define all(v) (v).begin(),(v).end()
const ll inf=(1e18);
//const ll mod=998244353;
const ll mod=1000000007;
const vector<int> dy={-1,0,1,0},dx={0,-1,0,1};
ll GCD(ll a,ll b) {return b ? GCD(b,a%b):a;}
ll LCM(ll c,ll d){return c/GCD(c,d)*d;}
struct __INIT{__INIT(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(20);}} __init;
template<class T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a=b; return 1; } return 0; }
template<class T>void debag(const vector<T> &a){cerr<<"debag :";for(auto v:a)cerr<<v<<" ";cerr<<"\n";}
template<class T>void print(const vector<T> &a){for(auto v:a)cout<<v<<" ";cout<<"\n";}
int main(){
      int n;
      cin>>n;
      V<int> a(n);
      for(int i=0;i<n;i++){
            cin>>a[i];
      }
      int c=a.back()^1;
      ll ans=0;
      ll sum=0;
      for(int i=n-2;i>=0;i--){
            if(c!=a[i]){
                  sum=0;
                  c=a[i];
            }
            sum++;
            if(a[i])ans+=sum;
            c^=1;
      }
      cout<<ans<<"\n";
}
0