結果

問題 No.1470 Mex Sum
ユーザー kaede2020
提出日時 2021-04-09 21:50:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 951 bytes
コンパイル時間 4,000 ms
コンパイル使用メモリ 228,892 KB
実行使用メモリ 6,656 KB
最終ジャッジ日時 2024-06-25 05:04:08
合計ジャッジ時間 6,974 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 WA * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
//using mint = modint1000000007;
using mint = modint998244353;
template <typename T> bool chmax(T &u, const T z) { if (u < z) {u = z; return true;} else return false; }
template <typename T> bool chmin(T &u, const T z) { if (u > z) {u = z; return true;} else return false; }
#define ll long long
#define rep(i, n) for (ll i = 0; i < (ll)(n); i++)
typedef pair<ll, int> P;
ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }
ll lcm(ll a, ll b){return a * b / gcd(a, b);}

int main(){
  int n;
  cin>>n;
  vector<ll>a(n),b(n-1);
  rep(i,n)cin>>a[i];
  rep(i,n-1){
    if(a[i]+a[i+1]==2)b[i]=2;
    else if(a[i]+a[i+1]==3)b[i]=3;
    else if(a[i]!=1&&a[i+1]!=1)b[i]=1;
    else if(a[i]!=2&&a[i+1]!=2)b[i]=2;
    else b[i]=min(a[i],a[i+1])+1LL;
  }
  //rep(i,n-1)cout<<b[i];
  ll ans=0;
  rep(i,n-1){
    ans+=b[i]*(n-i-1);
  }
  cout<<ans<<endl;
  return 0;
}
0