제출 #1300908

#제출 시각아이디문제언어결과실행 시간메모리
1300908m.zeeshanrashidIzbori (COCI22_izbori)C++20
40 / 110
76 ms14524 KiB
// #ifdef __AVX2__
// #pragma GCC target "avx2"
// #endif
// #pragma GCC optimize "O3"
// #pragma GCC optimize "unroll-loops"
#include <bits/stdc++.h>
// #include <ext/pb_ds/assoc_container.hpp> 
// #include <ext/pb_ds/tree_policy.hpp> 
// using namespace __gnu_pbds; 
using namespace std;
#define int long long
#define elif else if
#define all(l) begin(l),end(l)
#define rall(l) rbegin(l),rend(l)
#define append push_back
#define print(l) for(auto i:l) cout<<i<<' '; cout<<endl;
#define pprint(a,b) cout<<a<<' '<<b<<endl;
#define inp(l) for(auto &i:l) cin>>i;
// #define ordered_set tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
#define pai make_pair
#define endl "\n"
#define pii pair<int,int>
#define fi first
#define se second
#define vec vector
#define ins(a,b,c) a.insert(begin(a)+b,c);

// const int mod=998244353;
const int mod1=998244353;
const int mod=1e9+7;
const int N=2e5+5;

int iter=1,itera=1;
void solve(){
	int n;
	cin>>n;
	vec<int>a(n);
	inp(a)
	ins(a,0,0);
	if(n<=2000){
		int ans=0;
		for(int l=1;l<=n;l++){
			int ma=0;
			map<int,int>d;
			for(int r=l;r<=n;r++){
				d[a[r]]++;
				ma=max(ma,d[a[r]]);
				if(ma>(r-l+1)/2) ans++;
				// cout<<l<<' '<<r<<' '<<ma<<endl;
			}
		}
		cout<<ans<<endl;
		return;
	}
	map<int,int>d;
	d[0]=1;
	int ans=(n*(n+1))/2;
	int c=0;
	for(int i=1;i<=n;i++){
		if(a[i]==a[1]) c++;
		else c--;
		ans-=d[c];
		// cout<<d[c]<<' ';
		// cout<<c<<' ';
		d[c]++;
	}
	cout<<ans<<endl;
}
signed main(){
	// freopen("","r",stdin);
	// freopen("","w",stdout);
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); 
	cout.tie(NULL);
	cout<<fixed<<setprecision(20);
	// cin>>itera;
	for(iter=1;iter<=itera;iter++) solve();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...