제출 #1018757

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1018757		vjudge1	Izbori (COCI22_izbori)	C++17	25 / 110	3043 ms	600 KiB

Main

#include <bits/stdc++.h>
#define fast cin.tie(0)->sync_with_stdio(0);
#define inf ((int)1e9)
using namespace std;

void solve1(int n) {
	vector <int> arr(n);
	for(int i = 0; i < n; i++) {
		cin >> arr[i];
	}
	auto compress = [&](vector <int> &a) {
		vector <pair<int, int> > v(a.size());
		for(int i = 0; i < n; i++) {
			v[i] = {a[i], i};
		}
		sort(v.begin(), v.end());
		for(int i = 0, ind = 0; i < n; i++) {
			if(i and v[i].first != v[i - 1].first) ind++;
			a[v[i].second] = ind;
		}
	};
	compress(arr);
	int ans = 0;
	for(int l = 0; l < n; l++) {
		vector <int> cnt(n);
		int mx = 0;
		for(int r = l; r < n; r++) {
			cnt[arr[r]]++;
			mx = max(mx, cnt[arr[r]]);
			if(mx > (r - l + 1) / 2) ans++;
		}
	}
	cout << ans << "\n";
}

bitset <200000> st;

void solve2(int n) {
	for(int i = 0; i < n; i++) {
		int in;
		cin >> in;
		st[i] = in - 1;
	}
	int ans = 0;
	for(int l = 0; l < n; l++) {
		int cnt[2] = {0};
		for(int r = l; r < n; r++) {
			cnt[st[r]]++;
			ans += cnt[0] != cnt[1];
		}
	}
	cout << ans << "\n";
}

int32_t main(){
	fast
	int n;
	cin >> n;
	if(n <= 2000) solve1(n);
	else solve2(n);
}

• Subtask 1: 10 / 10
• Subtask 2: 15 / 15
• Subtask 3: 0 / 15
• Subtask 4: 0 / 70