首先修改操作是一个比较难实现的点,因此我们通过增加一个时间维度将修改操作变成增加操作。
曼哈顿距离直接跑 CDQ 分治,貌似不太可做,考虑转化为切比雪夫距离。
如何转化看这:
转换完之后我们会发现他的查询操作实际上就是查询一个边长位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;
}