Submission #404132

#	Time	Username	Problem	Language	Result	Execution time	Memory
404132		penguinhacker	Doktor (COCI17_doktor)	C++14	100 / 100	193 ms	36060 KiB

doktor

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define ar array

const int mxN=5e5;
int n, a[mxN], p[mxN], s[mxN];
vector<int> oc[mxN];
ar<int, 3> ans;

int pre(int i) {return i>=0?p[i]:0;}
int suf(int i) {return i<n?s[i]:0;}

void solve(int x) {
	for (int i=0; i<n; ++i) {
		int j=i+a[i];
		if (j%2==x)
			oc[j/2].push_back(abs(j/2-min(i, a[i])));
	}
	for (int i=0; i<n; ++i) {
		sort(oc[i].begin(), oc[i].end());
		int cur=0;
		for (int j : oc[i]) {
			++cur;
			int cur2=cur+pre(i-j-1)+suf(i+j+1+x);
			ans=max(ans, {cur2, i-j, i+j+x});
		}
		oc[i].clear();
	}
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> n;
	for (int i=0; i<n; ++i) {
		cin >> a[i], --a[i];
		p[i]=(i?p[i-1]:0)+(i==a[i]);
	}
	for (int i=n-1; ~i; --i)
		s[i]=(i+1<n?s[i+1]:0)+(i==a[i]);
	ans={s[0], 0, 0};
	solve(0), solve(1);
	cout << a[ans[1]]+1 << " " << a[ans[2]]+1;
	return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10