Submission #132829

#	Time	Username	Problem	Language	Result	Execution time	Memory
132829		Ort	Doktor (COCI17_doktor)	C++11	100 / 100	289 ms	45788 KiB

doktor

#include<iostream>
#include<algorithm>
#include<vector>
#define MAX 500005

using namespace std;

int n, x, soll, solr, sol;
int arr[MAX], indx[MAX], lef[MAX], righ[MAX];
vector<vector<int> > a;

int main() {
	cin.sync_with_stdio(0); cin.tie(0);
	cin.exceptions(cin.failbit);
	cin >> n; a.resize(n*2+1);
	for(int i=1;i<=n;i++) {
		cin >> arr[i];
		indx[arr[i]] = i;
		a[i+arr[i]].push_back(i);
	}
	for(int i=0;i<a.size();i++) sort(a[i].begin(),a[i].end());
	for(int i=1;i<=n;i++) righ[i] = righ[i-1]+(arr[i]==i);
	for(int i=n;i>0;i--) lef[i] = lef[i+1]+(arr[i]==i);
	for(int i=1;i<=n;i++) {
		int l=i, r=indx[i]; if(r<l) swap(r,l);
		int f = (upper_bound(a[l+r].begin(),a[l+r].end(),r) - lower_bound(a[l+r].begin(),a[l+r].end(),l));
		if(righ[l-1]+lef[r+1]+f > sol) {soll = l; solr = r; sol = righ[l-1]+lef[r+1]+f;}
	}
	cout << arr[soll] << " " << arr[solr];
	return 0;
}

Compilation message (stderr)

doktor.cpp: In function 'int main()':
doktor.cpp:21:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<a.size();i++) sort(a[i].begin(),a[i].end());
              ~^~~~~~~~~

• 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