結果

問題 No.2956 Substitute with Average
ユーザー はーチャンネルはーチャンネル
提出日時 2024-12-12 10:54:30
言語 C++23(gcc13)
(gcc 13.2.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,075 bytes
コンパイル時間 3,210 ms
コンパイル使用メモリ 277,904 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-12 10:54:37
合計ジャッジ時間 4,634 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 WA -
testcase_04 AC 1 ms
5,248 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 1 ms
5,248 KB
testcase_09 AC 1 ms
5,248 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 22 ms
5,248 KB
testcase_19 AC 25 ms
5,248 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 AC 25 ms
5,248 KB
testcase_23 AC 20 ms
5,248 KB
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<long long> VL;
typedef vector<VL> VVL;
typedef long long LL;
#define all(a) (a).begin(), (a).end()
#define Yes(n) cout << ((n) ? "Yes" : "No"  ) << endl
#define ALL(a)  (a).begin(),(a).end()
#define pb push_back

int main() {
  int n;cin>>n;
  int AA[n];
  rep(i,n){cin>>AA[i];}
  LL ans=LL(n)*(n+1)/2;
  for(int i=1;i<=30;i++){
    //cout<<i<<endl;
    VI A,B;
    int a=0,b=0,sum=0;
    rep(j,n){
      //cout<<A[j]<<endl;
      if(AA[j]==i){
        if(sum!=0){
          if(b==sum*i){B.pb(1);}
          else{B.pb(0);}
          b=0;sum=0;
        }
        a++;
      }
      else{
        if(a!=0){
          A.pb(a);a=0;
        }
        b+=AA[j];sum++;
      }
      //cout<<b<<endl;
    }
    if(a!=0){A.pb(a);}
    int m=A.size();
    //cout<<m<<' '<<B.size()<<endl;
    rep(i,m){ans-=LL(A[i])*(A[i]+1)/2;}
    rep(i,m-1){
      if(B[i]){ans-=LL(A[i])*A[i+1]-1;}
    }
  }
  ans++;
  cout<<ans<<endl;
}
0