Submission #992517

#	Time	Username	Problem	Language	Result	Execution time	Memory
992517		LOLOLO	Doktor (COCI17_doktor)	C++14	100 / 100	104 ms	46724 KiB

doktor

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
 
#define           f     first
#define           s     second
#define           pb    push_back
#define           ep    emplace
#define           eb    emplace_back
#define           lb    lower_bound
#define           ub    upper_bound
#define       all(x)    x.begin(), x.end()
#define      rall(x)    x.rbegin(), x.rend()
#define   uniquev(v)    sort(all(v)), (v).resize(unique(all(v)) - (v).begin())
#define     mem(f,x)    memset(f , x , sizeof(f))
#define        sz(x)    (ll)(x).size()
#define  __lcm(a, b)    (1ll * ((a) / __gcd((a), (b))) * (b))
#define          mxx    *max_element
#define          mnn    *min_element
#define    cntbit(x)    __builtin_popcountll(x)
#define       len(x)    (int)(x.length())

const int N = 1e6 + 100;
const int M = 5e5 + 10;
int a[N], S[N], P[N];
vector <int> pos[N];

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int n;
    cin >> n;

    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        pos[a[i] + i].pb(i);
    }

    for (int i = 1; i <= n; i++) {
        if (i == a[i]) 
            S[i]++;

        S[i] += S[i - 1];
    }

    for (int i = n; i >= 1; i--) {
        if (i == a[i])
            P[i]++;

        P[i] += P[i + 1];
    }

    int x = 1, y = 1, ans = S[n];
    for (int i = 1; i <= n; i++) {
        int l = min(i, a[i]), r = max(a[i], i), t = a[i] + i, add = P[r + 1] + S[l - 1];
        int R = lower_bound(all(pos[t]), l) - pos[t].begin();
        int L = upper_bound(all(pos[t]), r) - pos[t].begin();
        if (ans < L - R + add) {
            ans = L - R + add;
            x = a[l], y = a[r];
        }
    }

    cout << x << " " << y << '\n';
    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