結果

問題 No.2956 Substitute with Average
ユーザー 沙耶花沙耶花
提出日時 2024-11-08 21:44:35
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,210 ms / 3,000 ms
コード長 758 bytes
コンパイル時間 4,199 ms
コンパイル使用メモリ 267,468 KB
実行使用メモリ 10,368 KB
最終ジャッジ日時 2024-11-08 21:44:59
合計ジャッジ時間 20,437 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 726 ms
9,984 KB
testcase_11 AC 733 ms
9,984 KB
testcase_12 AC 713 ms
9,984 KB
testcase_13 AC 755 ms
9,984 KB
testcase_14 AC 746 ms
9,984 KB
testcase_15 AC 737 ms
10,112 KB
testcase_16 AC 759 ms
10,112 KB
testcase_17 AC 740 ms
10,112 KB
testcase_18 AC 1,156 ms
10,240 KB
testcase_19 AC 916 ms
10,240 KB
testcase_20 AC 865 ms
10,368 KB
testcase_21 AC 870 ms
10,240 KB
testcase_22 AC 1,210 ms
10,240 KB
testcase_23 AC 1,153 ms
10,368 KB
testcase_24 AC 812 ms
10,368 KB
testcase_25 AC 827 ms
10,240 KB
testcase_26 AC 765 ms
10,240 KB
testcase_27 AC 808 ms
10,240 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000000LL


int main(){
	
	int n;
	cin>>n;
	vector<long long> a(n);
	rep(i,n){
		cin>>a[i];
	}
	
	long long ans = n;
	ans *= n+1;
	ans /= 2;
	ans++;
	
	for(int i=1;i<=30;i++){
		
		rep(j,2){
			map<int,long long> mp;
			if(a[0]!=i || j==0){
				mp[0]++;
			}
			long long cur = 0;
			rep(k,n){
				cur += a[k]-i;
				if(a[k]==i){
					ans -= mp[cur];
				}
				if(k!=n-1 && a[k+1]==i){
					if(j==0)mp[cur]++;
				}
				else mp[cur]++;
			}
			reverse(a.begin(),a.end());
		}
		
	}
	cout<<ans<<endl;
	
	
	return 0;
}
0