序列查询新解（第24次CCF - B）（双指针）

传送门

题目

思路

双指针。
分别分析$f(x)和g(x)$的性质

• $对于f(x)$
  $$\begin{gathered} 可以将序列a看作n个不相重叠且连续的区间。第i个区间为[a[i],a[i+1]) \\ 第i个区间中所有的数的函数值f(x)=\mathbf{i} \end{gathered}$$
• $对于g(x)$
  $定义：b[i]=i\times r,且b[i]<=m$
  $$\begin{gathered} b[0]=0 \\ b[1]=r \\ b[2]=2r \\ \cdots \cdots \\ b[k-1]=(k-1)r \\ b[k]=m \end{gathered}$$
  $$\begin{gathered} 同理，b序列也可以看作k个不同的区间 \\ 第j个区间对应的函数值g(x)=\mathbf{j} \end{gathered}$$

每次移动时，将当前指针的公共部分累加入ans中即可。具体细节看代码部分

$第一次求得[0,a[i])的贡献$
$第二次求得[a[1],b[1])的贡献$
$第三次求得[b[1],a[2])的贡献$
$\cdots \cdots$
$最终i，j两个指针均指向两个数组各自的末尾(a[n+1]与b[k])$

CODE

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5 + 7;
ll a[N], b[N];

int main()
{
	IOS;
	ll n, m; cin >> n >> m;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	a[n + 1] = m;

	ll r = m / (n + 1);
	ll k = 1;
	for (; k * r < m; k++)
		b[k] = k * r;
	b[k] = m;

	ll ans = 0;
	int i = 1, j = 1;
	while (i < n + 1 || j < k)//注意是“或”
	{
		if (a[i] < b[j])
		{
			ans += (min(a[i + 1], b[j]) - a[i]) * abs(j - 1 - i);
			i++;
		}
		else if (a[i] > b[j])
		{
			ans += (min(a[i], b[j + 1]) - b[j]) * abs(i - 1 - j);
			j++;
		}
		else
		{
			ans += (ll)(min(a[i + 1], b[j + 1]) - a[i]) * abs(i - j);
			i++;
			j++;
		}
	}

	cout << ans << endl;

	return 0;
}