BZOJ2989 数列/BZOJ4170 极光

首先修改操作是一个比较难实现的点，因此我们通过增加一个时间维度将修改操作变成增加操作。

曼哈顿距离直接跑 CDQ 分治，貌似不太可做，考虑转化为切比雪夫距离。

如何转化看这：

距离 - OI Wiki

转换完之后我们会发现他的查询操作实际上就是查询一个边长位2*k的正方形范围内的点的数量，

这一步我们可以用类似[BalkanOI2007] Mokia 摩基亚 - 洛谷的方法（双倍经验）。

类似的题还有[Violet] 天使玩偶/SJY摆棋子 - 洛谷（三倍经验）！！！

代码如下：

#include<bits/stdc++.h>
using namespace std;

#define ull unsigned long long
#define ll long long
#define edl '\n'

const int N = 1e6 + 10;
const int M = 1e3 + 10;

const int mod = 1e9 + 7;
ll n,m,q,x,y;
string s;

template <typename T>
struct Fenwick {
	int n;
	std::vector<T> a;
	
	Fenwick(int n_ = 0) {
		init(n_);
	}
	
	void init(int n_) {
		n = n_;
		a.assign(n, T{});
	}
	
	void add(int x, const T &v) {
		for (int i = x + 1; i <= n; i += i & -i) {
			a[i - 1] = a[i - 1] + v;
		}
	}
	//[0,x)
	T sum(int x) {
		T ans{};
		for (int i = x; i > 0; i -= i & -i) {
			ans = ans + a[i - 1];
		}
		return ans;
	}
	//左闭右开
	T rangeSum(int l, int r) {
		return sum(r) - sum(l);
	}
	
	//return 第一个sum(x) > k的x
	int select(const T &k) {
		int x = 0;
		T cur{};
		for (int i = 1 << __lg(n); i; i /= 2) {
			if (x + i <= n && cur + a[x + i - 1] <= k) {
				x += i;
				cur = cur + a[x - 1];
			}
		}
		return x;
	}
};

Fenwick<ll> f(N);

struct QWQ
{
	ll x,y,op,t;
}a[N];

bool cmp(QWQ A,QWQ B)
{
	if(A.x != B.x) return A.x < B.x;
	if(A.y != B.y) return A.y < B.y;
	if(A.t != B.t) return A.t < B.t;
}

bool cmp1(QWQ A,QWQ B)
{
	if(A.y != B.y) return A.y < B.y;
	if(A.t != B.t) return A.t < B.t;
}

bool cmp2(QWQ A,QWQ B)
{
	if(A.t != B.t) return A.t < B.t;
}
int idx;
ll AN[N];
ll b[N];

void cdq(int l,int r)
{
	if(l == r) return;
	ll mid = (l + r) >> 1;
	cdq(l,mid);
	cdq(mid + 1,r);
	sort(a + l,a + mid + 1,cmp1);
	sort(a + mid + 1,a + r + 1,cmp1);
	int i = l;
	int j = mid + 1;
	while(j <= r)
	{
		while(i <= mid && a[i].y <= a[j].y)
		{
			f.add(a[i].t,1 - a[i].op);
			i ++;
		}
		AN[a[j].t] += f.sum(a[j].t + 1);
		j ++;
	}
	for(j = l; j < i; j ++ ) f.add(a[j].t,a[j].op - 1);
}

void solve()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i ++ )
	{
		cin >> x;
		b[i] = x;
		a[++idx].x = x + i;
		a[idx].y = i - x;
		a[idx].op = 0;
		a[idx].t = idx;
	}
	for(int i = 1; i <= m; i ++ )
	{
		cin >> s;
		if(s == "Query")
		{
			cin >> x >> y;
			a[++idx] = {x + b[x] + y,x - b[x] + y,1,idx};
			a[++idx] = {x + b[x] - y - 1,x - b[x] - y - 1,1,idx};
			a[++idx] = {x + b[x] - y - 1,x - b[x] + y,1,idx};
			a[++idx] = {x + b[x] + y,x - b[x] - y - 1,1,idx};
		}
		else
		{
			cin >> x >> y;
			b[x] = y;
			a[++idx].x = x + y;
			a[idx].y = x - y;
			a[idx].op = 0;
			a[idx].t = idx;
		}
	}
	sort(a + 1, a + idx + 1,cmp);
	cdq(1,idx);
	sort(a + 1, a + idx + 1,cmp2);
	for(int i = 1; i <= idx; i ++ )
	{
		if(a[i].op)
		{
			cout << AN[a[i].t] + AN[a[i + 1].t] - AN[a[i + 2].t] - AN[a[i + 3].t] << edl;
			i += 3;
		}
	}
}

int main()
{
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	int t = 1;
//	cin >> t;
	while(t -- ) solve();
	return 0;
}