Submission #166018

#	Time	Username	Problem	Language	Result	Execution time	Memory
166018		theStaticMind	Doktor (COCI17_doktor)	C++14	100 / 100	264 ms	45708 KiB

doktor

#include<bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define ii pair<int,int>
#define all(x) (x).begin(),(x).end()
#define INF 100000000000000000
#define modulo 1000000007
#define mod 998244353
#define int long long int
using namespace std;

int32_t main(){
      ios_base::sync_with_stdio(false);
      cin.tie(NULL);
 //     freopen("q.gir","r",stdin);
 //     freopen("q.cik","w",stdout);
      int n;
      cin>>n;
      vector<int> arr(n+1),pre(n+1,0);
      for(int i=1;i<=n;i++)cin>>arr[i];
      vector<vector<int> >data(2*n+1);
      for(int i=1;i<=n;i++){
            data[arr[i]+i].pb(abs(arr[i]-i));
      }
      for(int i=1;i<=n;i++){
            if(arr[i]==i)pre[i]++;
      }
      for(int i=1;i<=n;i++)pre[i]+=pre[i-1];
      int ans=0;
      int L=0,R=0;
      for(int i=0;i<=2*n;i++){
            sort(all(data[i]));
            for(int j=0;j<data[i].size();j++){
                  int l=(i-data[i][j])/2;
                  int r=(i+data[i][j])/2;
                  int cnt=j+1+pre[l-1]+pre[n]-pre[r];
                  if(cnt>ans){
                        ans=cnt;
                        L=l;
                        R=r;
                  }
            }
      }
      cout<<arr[L]<<" "<<arr[R];
}

Compilation message (stderr)

doktor.cpp: In function 'int32_t main()':
doktor.cpp:33:26: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             for(int j=0;j<data[i].size();j++){
                         ~^~~~~~~~~~~~~~~

• 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