Submission #308551

#	Time	Username	Problem	Language	Result	Execution time	Memory
308551		shivensinha4	Lightning Conductor (POI11_pio)	C++17	18 / 100	52 ms	52088 KiB

pio

#include <bits/stdc++.h> 
using namespace std; 
#define for_(i, s, e) for (int i = s; i < (int) e; i++)
#define for__(i, s, e) for (ll i = s; i < e; i++)
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> ii;
#define endl '\n'

const int MXV = 1e6, INF = 1e9;
int rt[MXV+1];
vi pos[MXV+1];

int main() {
	#ifdef shiven
	freopen("test.in", "r", stdin);
	#endif
	
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	
	int n; cin >> n;
	vi h(n);
	int mx = 0;
	for_(i, 0, n) {
		cin >> h[i];
		mx = max(mx, h[i]);
		if (!pos[h[i]].size()) pos[h[i]] = {i, i};
		else pos[h[i]] = {min(pos[h[i]][0], i), max(pos[h[i]][1], i)};
	}
	
	int r = 0, sq = 0;
	for_(i, 0, mx+1) {
		rt[i] = r;
		if (i == sq) {
			r += 1;
			sq = r*r;
		}
	}
	
	//vi poss;
	//for (int i = mx; i >= max(mx-710, 0); i--) if (pos[i].size()) {
		 //poss.push_back(i);
	//}
	
	for_(i, 0, n) {
		int ans = h[i];
		for (int v = mx; v >= max(mx-710, 0); v--) if (pos[v].size()) ans = max({ans, v + rt[abs(pos[v][0] - i)], v + rt[abs(pos[v][1] - i)]});
		cout << ans-h[i] << endl;
	}
	

	return 0;
}

• Subtask 1: 9 / 9
• Subtask 2: 9 / 9
• Subtask 3: 0 / 9
• Subtask 4: 0 / 9
• Subtask 5: 0 / 9
• Subtask 6: 0 / 9
• Subtask 7: 0 / 9
• Subtask 8: 0 / 9
• Subtask 9: 0 / 9
• Subtask 10: 0 / 9
• Subtask 11: 0 / 10