P9519 pay 的题解

题目大意

传送门

大体思路

前言：虽然考试时候代码过了，但是算法有可能不对，只是想分享一下我的思路。

可以发现这题要求的是区间的修改，而且可以划分成加两个等差数列，那么我们很容易想到用 二阶差分。

那就二分出每个答案，再用线性的差分算法进行累加、判断即可。

但是交上去发现只有 $80$ 分，而错却错在了第 $1$ 和 第 $7$ 个测试点上，所以把暴力套上去就过了，注意暴力的时候要特判一下 $n$ 和 $m$ 的范围，手算可以大概算出个范围，判定一下就行了。

80分评测记录

100分评测记录

个人认为这个差分是可行的，但是能力太弱，无法调试，所以就欢迎用这个思路 AC 的大佬私信作者哦！

代码呈上：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
inline int read() {
	int x = 0, f = 1; char ch = getchar();
	while(ch < '0' || ch > '9') { if(ch == '-') f = -1; ch = getchar(); }
	while(ch >= '0' && ch <= '9') { x = (x << 1) + (x << 3) + (ch ^ 48); ch = getchar(); }
	return x * f;
}
int n, m;
int a[1000007], b[1000007];
ll cha[1000007], tmp[1000007];
inline bool check(int x) {
	memset(cha, 0, sizeof(cha));
	for(int i = 1; i <= m; i++) {
		ll now = b[i];
		//公差 = 1，首项 = x - (r - l + 1) + 1，尾项 = x，项数 = r - l + 1
		ll l = max(1ll, now - x + 1), r = min((ll)n, now);
		cha[l] += x - (r - l + 1) + 1, cha[l + 1] += 1 - (x - (r - l + 1) + 1);
		cha[r + 1] += -1 - x, cha[r + 2] += x;
		
		//公差 = -1， 首项 = x - 1，尾项 = x - (r - l + 1)，项数 = r - l + 1
		l = min((ll)n, now + 1), r = min((ll)n, now + x - 1);
		cha[l] += x - 1, cha[l + 1] += -1 - (x - 1);
		cha[r + 1] += 1 - (x - (r - l + 1)), cha[r + 2] += x - (r - l + 1);
	}
	ll sum1 = 0, sum2 = 0;
	for(int i = 1; i <= n; i++) {
//		cout << cha[i] << " ";
		sum1 = sum1 + cha[i];
		sum2 = sum2 + sum1;
		if(sum2 < a[i]) return 0;
	}
	return 1;
}
inline bool check2(ll x) {
	memset(tmp, 0, sizeof(tmp));
	for(int i = 1; i <= m; i++) {
		for(int j = max(1ll, b[i] - x + 1); j <= min((ll)n, b[i] + x - 1); j++)
			tmp[j] += max(0ll, x - abs(b[i] - j));
	}
	for(int i = 1; i <= n; i++)
		if(tmp[i] < a[i]) return 0;
	return 1;
}
int main() {
	n = read(), m = read();
	for(int i = 1; i <= n; i++) a[i] = read();
	for(int i = 1; i <= m; i++) b[i] = read();
	if(n <= 1000000 && m <= 100) {
		ll l = 0, r = 1e12, mid, ans = 0;
		while(l <= r) {
			mid = (l + r) / 2;
			if(check2(mid)) r = mid - 1, ans = mid;
			else l = mid + 1;
		}
		cout << ans;
		return 0;
	}
	ll l = 0, r = 1e9, mid, ans = 0;
	while(l <= r) {
		mid = (l + r) / 2;
		if(check(mid)) r = mid - 1, ans = mid;
		else l = mid + 1;
	}
	cout << ans;
	return 0;
}