Submission #426392

#	Time	Username	Problem	Language	Result	Execution time	Memory
426392		timmyfeng	Shortcut (IOI16_shortcut)	C++17	97 / 100	2070 ms	60456 KiB

shortcut

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

const int N = 1000001;

int sum_order[N], diff_order[N], n;
long long sum[N], diff[N], x[N];

long long find_shortcut(int n, vector<int> l, vector<int> d, int c) {
    ::n = n;

    for (int i = 0; i < n - 1; ++i) {
        x[i + 1] = x[i] + l[i];
    }

    for (int i = 0; i < n; ++i) {
        sum[i] = x[i] + d[i];
        diff[i] = x[i] - d[i];
    }

    sort(d.rbegin(), d.rend());

    iota(sum_order, sum_order + n, 0);
    sort(sum_order, sum_order + n, [&](int a, int b) {
        return sum[a] < sum[b];
    });

    iota(diff_order, diff_order + n, 0);
    sort(diff_order, diff_order + n, [&](int a, int b) {
        return diff[a] < diff[b];
    });

    long long low = d[0] + d[1], high = sum[sum_order[n - 1]] - diff[diff_order[0]];
    while (low < high) {
        long long mid = (low + high) / 2;

        long long min_sum = LLONG_MIN, max_sum = LLONG_MAX;
        long long min_diff = LLONG_MIN, max_diff = LLONG_MAX;
        long long s = LLONG_MIN, d = LLONG_MAX;
        vector<int> ext;

        for (int i = 0, j = 0; j < n; ++j) {
            while (i < n && sum[sum_order[j]] - diff[diff_order[i]] > mid) {
                s = max(s, sum[diff_order[i]]);
                d = min(d, diff[diff_order[i]]);
                ++i;
            }

            if (sum[sum_order[j]] - diff[sum_order[j]] > mid) {
                ext.push_back(sum_order[j]);
            } else if (s > LLONG_MIN) {
                min_sum = max(min_sum, -mid + c + sum[sum_order[j]] + s);
                max_sum = min(max_sum, mid - c + diff[sum_order[j]] + d);
                min_diff = max(min_diff, -mid + c + sum[sum_order[j]] - d);
                max_diff = min(max_diff, mid - c + diff[sum_order[j]] - s);
            }
        }

        for (auto j : ext) {
            for (int i = 0; i < j; ++i) {
                if (sum[j] - diff[i] > mid) {
                    min_sum = max(min_sum, -mid + c + sum[j] + sum[i]);
                    max_sum = min(max_sum, mid - c + diff[j] + diff[i]);
                    min_diff = max(min_diff, -mid + c + sum[j] - diff[i]);
                    max_diff = min(max_diff, mid - c + diff[j] - sum[i]);
                }
            }
        }

        bool ok = false;
        int l1 = n - 1, l2 = 0, r1 = n - 1, r2 = 0;
        for (int i = 0; i < n; ++i) {
            while (l1 >= 0 && x[i] + x[l1] >= min_sum) {
                --l1;
            }
            while (r1 >= 0 && x[i] + x[r1] > max_sum) {
                --r1;
            }
            while (l2 < n && x[i] - x[l2] > max_diff) {
                ++l2;
            }
            while (r2 < n && x[i] - x[r2] >= min_diff) {
                ++r2;
            }
            ok |= max(l1 + 1, l2) < min(r1 + 1, r2);
        }

        if (ok) {
            high = mid;
        } else {
            low = mid + 1;
        }
    }

    return low;
}

• Subtask 1: 9 / 9
• Subtask 2: 14 / 14
• Subtask 3: 8 / 8
• Subtask 4: 7 / 7
• Subtask 5: 33 / 33
• Subtask 6: 22 / 22
• Subtask 7: 4 / 4
• Subtask 8: 0 / 3