洛谷题解 P5503 【[JSOI2016]灯塔】

来一篇优先队列的~

观察到 $p\ge h_j+\sqrt{|i-j|}-a_i$ ,对于第 $i$ 个询问, $a_i$ 是一定的，所以我们只需要维护 $h_j+\sqrt{|i-j|}$ 最大就 OK 了。对于第 $i$ 次询问，不难发现是可以由第 $i-1$ 次转移过来的，且每次转移最多只会更改 $2\times \sqrt{n}$ 个数，下标分别为 $i、i+1、i+4\dots \dots$ 和 $i-1、i-2、i-5\dots \dots$。左边 $ + 1$ 右边 $-1$ 。

但是单纯存储最大值及其数量会挂掉，不容易转移（其实是笔者太菜了）。容易想到直接用优先队列维护当前最大值和最大值代表的下标，如果下标的数被修改则 $pop$ 。

但是单纯有限队列还是会挂，错误结果是 MLE 。如果每次都压入 $\sqrt{n}$ ，总共会有超过 $6\times 10^7$ 个 int 。我们继续思考，发现 $h_j + \sqrt{|i - j|} $ 最多只会被减去 $\sqrt{n}$ 次，因此我们只需要将 $ max_{inqueue} - (h_j + \sqrt{|i - j|}) \le 2 \times \sqrt{n}$ （既有加又有减）的放入就好了（其实可以再小一点，估计小于 100 也可以过），其余的无论再怎么放也不会大于目前最大值。

复杂度为 $O(n\times \sqrt{n}\ast lo{g}_{?})$，但其实跑的比一般的 $O(n\times \sqrt{n})$ 要快好几倍。

Code

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#include <queue> 
#include <algorithm>
#define ll long long
#define inf 0x3f3f3f3f

using namespace std;

inline int read(){
	int x=0,w=0;char ch=getchar();
	while (!isdigit(ch))w|=ch=='-',ch=getchar();
	while (isdigit(ch))x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
	return w?-x:x;
}

priority_queue <pair<int, int>, vector<pair<int, int> >, less<pair<int, int> > > que;
int n, a[1000006], ans[1000006], dis[1000006], calc[1000006], id[1000006], tot, sum;

int main(){
//	freopen("gemo.in","r",stdin);
//	freopen("gemo.out","w",stdout);
	n = read();
	for(int i = 1; i <= n; i++){
		a[i] = read();
	}
	for(int i = 1; i <= n; i++){
		int w = sqrt(i - 1);
		if(w * w != i - 1) w++;
		dis[i] = w + a[i];
		if(que.empty() || dis[i] + 319 >= que.top().first )
		que.push(make_pair(dis[i], i));
	}
	ans[1] = que.top().first;
	for(int i = 2; i <= n; i++){
		for(int j = 0; j * j + i <= n; j++){
			int k = j * j + i;
			dis[k]--;
			if(que.empty() || dis[k] + 319 >= que.top().first )
			que.push(make_pair(dis[k], k));
		}
		for(int j = 0; i - 1 - j * j >= 1; j++){
			int k = i - 1 - j * j;
			dis[k]++;
			if(que.empty() || dis[k] + 319 >= que.top().first )
			que.push(make_pair(dis[k], k));
		}
		pair <int, int> u = que.top();
		while(dis[u.second] != u.first){
			que.pop();
			if(que.empty() ) break;
			u = que.top();
		}
//		u = que.top() ;
		ans[i] = u.first ;
	}
	for(int i = 1; i <= n; i++){
		ans[i] = max(0, ans[i] - a[i]);
		printf("%d\n",ans[i]);
	}
    return 0;
}

完结撒花✿✿ヽ(°▽°)ノ✿