Submission #302456

#TimeUsernameProblemLanguageResultExecution timeMemory
302456JPN20Shortcut (IOI16_shortcut)C++17
0 / 100
236 ms378232 KiB
#include "shortcut.h"
#include <bits/stdc++.h>
using namespace std;

class SparseTable {
	public:
	long long sz[1 << 20];
	long long dat[1 << 20][22];
	
	void init(vector<long long> Arr) {
		for (int i = 0; i < (1 << 20); i++) {
			for (int j = 0; j < 22; j++) dat[i][j] = 0;
		}
		for (int i = 0; i < (int)Arr.size(); i++) dat[i][0] = Arr[i];
		for (int i = 0; i < 20; i++) {
			for (int j = 0; j < (int)Arr.size(); j++) {
				if (j + (1 << i) >= (int)Arr.size()) continue;
				dat[j][i + 1] = max(dat[j][i], dat[j + (1 << i)][i]);
			}
		}
		for (int i = 0; i < 20; i++) {
			for (int j = (1 << i); j < (2 << i); j++) sz[j] = i;
		}
	}
	long long query(int l, int r) {
		if (l >= r) return -(1LL << 60);
		int len = r - l;
		int t = sz[len];
		return max(dat[l][t], dat[r - (1 << t)][t]);
	}
};

// Input
long long N, C;
long long X[1 << 20];
long long Y[1 << 20];

// SparseTable
SparseTable Z1;
SparseTable Z2;

long long calc(int l, int r) {
	long long val1 = 0;
	long long val2 = 0;
	long long val3 = 0;
	long long val4 = 0;
	long long TotalLen = X[r] - X[l] + C;
	
	// VAL1
	int pos1 = lower_bound(X + l, X + r + 1, X[l] + (TotalLen + 1LL) / 2LL) - X;
	long long c1 = Z2.query(0, l) + X[l];
	long long c2 = Z1.query(l, pos1) - X[l];
	long long c3 = Z2.query(pos1, r + 1) + X[r] + C;
	val1 = c1 + max(c2, c3);
	
	// VAL2
	int pos2 = lower_bound(X + l, X + r + 1, X[r] - TotalLen / 2LL) - X;
	long long d1 = Z1.query(r + 1, N) - X[r];
	long long d2 = Z2.query(pos2, r + 1) + X[r];
	long long d3 = Z1.query(l, pos2) - X[l] + C;
	val2 = d1 + max(d2, d3);
	
	// VAL3
	long long e1 = Z2.query(0, l + 1) + X[l];
	long long e2 = Z1.query(r, N) - X[r];
	val3 = e1 + e2 + min(C, X[r] - X[l]);
	
	// VAL4
	for (int i = l; i < pos1; i++) {
		int pos3 = lower_bound(X + l, X + r + 1, X[i] + (TotalLen + 1LL) / 2LL) - X;
		long long d1 = Z2.query(l, i) + X[i];
		long long d2 = Z1.query(i + 1, pos3) - X[i];
		long long d3 = Z2.query(pos3, r + 1) + X[r] + C + (X[i] - X[l]);
		val4 = max(val4, max({d1, d2, d3}) + Y[i]);
	}
	for (int i = pos1; i <= r; i++) {
		int pos3 = lower_bound(X + l, X + r + 1, X[i] - TotalLen / 2LL) - X;
		long long d1 = Z1.query(l, pos3) - X[l] + C + (X[r] - X[i]);
		long long d2 = Z2.query(pos3, i) + X[i];
		long long d3 = Z1.query(i + 1, r + 1) - X[i];
		val4 = max(val4, max({d1, d2, d3}) + Y[i]);
	}
	
	// GETVAL
	//printf("(%d, %d) -> % 4lld, % 4lld, % 4lld, % 4lld\n", l, r, val1, val2, val3, val4);
	return max({val1, val2, val3, val4});
}

long long find_shortcut(int n, vector<int> l, vector<int> d, int c) {
	// INPUT
	N = n; C = c;
	for (int i = 1; i < N; i++) X[i] = X[i - 1] + 1LL * l[i - 1];
	for (int i = 0; i < N; i++) Y[i] = d[i];
	
	// TABLE
	vector<long long> V1; for (int i = 0; i < N; i++) V1.push_back(Y[i] + X[i]);
	vector<long long> V2; for (int i = 0; i < N; i++) V2.push_back(Y[i] - X[i]);
	Z1.init(V1);
	Z2.init(V2);
	
	// TANSAKU
	long long FinalAns = (1LL << 60);
	for (int i = 0; i < N; i++) {
		for (int j = i + 1; j < N; j++) {
			long long eval = calc(i, j);
			FinalAns = min(FinalAns, eval);
		}
	}
    return FinalAns;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...