序列查询新解 CSP 202112-2

 

 

 

 思路：

这道题搞得我好头大，一开始以为是前缀和，在1e5范围内没有问题，能拿70分，后面的就处理不了了。后来看了y总讲解，才有新的思路。

现在有两个函数fx和gx，要计算他们差值的绝对值，根据我们上一题的思路，我们还是把N加到a[n+1]去，采用遍历每一个区间，算出差值，但在每一个区间里，fx和gx的关系有3种。

1.fx全大于gx

2.gx全大于fx

3.由于gx单调递增，左边小于fx，右边大于fx

根据这三种情况，我们可以分类讨论，由于要遍历每一个区间，我们要分情况把每一个区间里的fx和gx求出来，然后一减得到差值，取绝对值加起来。

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 100010;
int n, m;
int a[N];
int R;

LL get(int l, int r) {//得到g(x)在l到r区间的和
    //如果就只有一段
	if (l / R == r / R)return (LL)(r - l + 1) * (l / R);
    //由于g(x)是一个等差数列,用高斯公式算
	int a = l / R + 1, b = r / R - 1;
	LL res = (a + b) * (LL)(b - a + 1) / 2 * (LL)R;
    //左边小段计算
	res += (l / R) * (LL)(a * R - l);
    //右边小段计算
	res += (b + 1) * (LL)(r - (b * R + R)+1);
	return res;
}

int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++)scanf("%d", &a[i]);
	a[n + 1] = m;
	R = m / (n + 1);
	LL res = 0;
	for (int i = 0; i <= n; i++) {
        //这一区间的左右端点
		int l = a[i], r = a[i + 1] - 1;
        //左右端点对应的g(x)值
		int x = l / R, y = r / R;
		if (y <= i || x >= i) {//全在上或全在下
			res += abs((LL)i * (r - l + 1) - get(l, r));
		}
		else {//分段,l到mid,mid+1到r
			int mid = i * R;//交点
			res += abs((LL)i * (mid - l + 1) - get(l, mid));
			res += abs((LL)i * (r - mid) - get(mid + 1 , r));
		}
	}
	cout << res;
	return 0;
}