結果

問題 No.2956 Substitute with Average
ユーザー keita-sugiurakeita-sugiura
提出日時 2024-12-24 15:08:39
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 895 bytes
コンパイル時間 6,461 ms
コンパイル使用メモリ 314,448 KB
実行使用メモリ 510,436 KB
最終ジャッジ日時 2024-12-24 15:09:57
合計ジャッジ時間 77,615 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint=modint998244353; //1000000007;
using ll=long long;
using pp=pair<int,int>;
#define sr string 
#define vc vector
#define fi first
#define se second
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define pb push_back
#define all(v) v.begin(),v.end()
#define pque priority_queue
#define bpc(a) __builtin_popcount(a)
int main(){
  int n;cin>>n;
  ll ans=(ll)n*(n+1)/2;
  int t=30;
  int mx=2*n*t+10;
  vc<unordered_map<int,int>>v(t);
  vc<unordered_map<int,int>>x(t);
  vc<int>vv(n); rep(i,n)cin>>vv[i],vv[i]--;
  rep(i,t)v[i][0]=1;
  rep(i,t)if(vv[0]!=i)x[i][0]=1;
  vc<int>s(t,0);
  rep(j,n){
    int a=vv[j];
    rep(i,t){
      s[i]+=a-i;
      ans-=v[i][s[i]];
      if(a!=i)ans+=x[i][s[i]];
      v[i][s[i]]++;
      if(j+1<n&&vv[j+1]!=i)x[i][s[i]]++;
    }
  }
  ans++;
  cout<<ans;
}
0