結果

問題 No.2956 Substitute with Average
ユーザー keita-sugiurakeita-sugiura
提出日時 2024-12-24 15:10:27
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 856 bytes
コンパイル時間 6,313 ms
コンパイル使用メモリ 317,664 KB
実行使用メモリ 275,584 KB
最終ジャッジ日時 2024-12-24 15:11:26
合計ジャッジ時間 52,641 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,820 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 2 ms
6,820 KB
testcase_06 AC 1 ms
6,816 KB
testcase_07 AC 2 ms
6,816 KB
testcase_08 AC 2 ms
6,820 KB
testcase_09 AC 2 ms
6,820 KB
testcase_10 AC 2,455 ms
244,608 KB
testcase_11 AC 2,464 ms
243,200 KB
testcase_12 AC 2,482 ms
243,584 KB
testcase_13 AC 2,465 ms
244,736 KB
testcase_14 AC 2,461 ms
242,688 KB
testcase_15 AC 2,472 ms
243,812 KB
testcase_16 AC 2,534 ms
244,224 KB
testcase_17 AC 2,537 ms
243,072 KB
testcase_18 AC 2,726 ms
275,584 KB
testcase_19 AC 2,500 ms
254,080 KB
testcase_20 AC 2,164 ms
259,968 KB
testcase_21 AC 2,526 ms
247,936 KB
testcase_22 TLE -
testcase_23 AC 2,735 ms
275,456 KB
testcase_24 AC 2,674 ms
251,776 KB
testcase_25 AC 2,625 ms
251,648 KB
testcase_26 AC 2,642 ms
251,392 KB
testcase_27 AC 2,603 ms
251,520 KB
権限があれば一括ダウンロードができます

ソースコード

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;
  vc<map<int,int>>v(t);
  vc<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