Description
给定一个长度为n的正整数数列a[i]。
定义2个位置的f值为两者位置差与数值差的和,即f(x,y)=|x-y|+|a[x]-a[y]|。
你需要写一个程序支持2种操作(k都是正整数):
Modify x k:将第x个数的值修改为k。
Query x k:询问有几个i满足f(x,i)<=k。询问不仅要考虑当前数列,还要考虑任意历史版本,即统计任意位置上出现过的任意数值与当前的a[x]的f值<=k的对数。(某位置多次修改为同样的数值,按多次统计)
Main
令F(x,y)=|x-y|+|a[x]-a[y]|,每次可以将a[x]修改为k,或者查询满足f(x,i)≤k的个数。
Analysis
【二维线段树/树状数组】
看到题目这样的粗俗,笼统,简洁明了,便知道一定要用一个数据结构维护
我们可以将x抽象为x坐标,a[x]抽象为y坐标,那么f(x,y)的意思就显然了:表示x点(x,a[x])和y点(y,a[y])的曼哈顿距离。
但是这个曼哈顿距离比较蛋疼,不着急,画(截)个图看看。
我们发现图片大概是这样的,发现这个可以取的范围是个菱形,旋转90°就变成了正方形。旋转就是使坐标(x,y)变成(x+y,x-y)
那么,问题就转化为每次可以加入一些点,求某个正方形内包含点的个数。
【K-Dtree】
。。。
【cbq分治+主席树】
...
Solution
我们考虑用一个数据结构来维护,如二维线段树/树状数组。
每次相当于插入点(x+a[x],x-a[x])到图中,线段树/树状数组维护左下角为(x1,y1),右上角为(x2,y2)的矩阵信息,查询即可。
对于二维线段树,有两种实现方法。一种是划分为4个区域,第二种是划分为2个,根据长宽的大小来切。
要动态开点。直接做会超时,要加优化:当前子树没有贡献,直接退出。
【K-Dtree】
。。。
【cbq分治+主席树】
...
Code
{$inline on}
var
ch,lala:char;
n,m,i,x,y,tot,ans:longint;
a:array[0..60000] of longint;
tree:array[0..2500000] of longint;
son:array[0..2500000,1..2] of longint;
procedure dispose; inline;
begin
read(ch);
lala:='1';
while lala<>' ' do read(lala);
readln(x,y);
end;
procedure fyj(x:longint); inline;
begin
if son[x,1]=0 then
begin
inc(tot); son[x,1]:=tot;
inc(tot); son[x,2]:=tot;
end;
end;
procedure change(root,x1,y1,x2,y2,fx,fy:longint); inline;
var
mid:longint;
begin
if (x1=x2) and (y1=y2) and (x1=fx) and (y1=fy) then
begin
inc(tree[root]);
exit;
end;
fyj(root);
if x2-x1>=y2-y1 then
begin
mid:=(x1+x2) shr 1;
if fx<=mid then
change(son[root,1],x1,y1,mid,y2,fx,fy)
else
change(son[root,2],mid+1,y1,x2,y2,fx,fy);
end
else
begin
mid:=(y1+y2) shr 1;
if fy<=mid then
change(son[root,1],x1,y1,x2,mid,fx,fy)
else
change(son[root,2],x1,mid+1,x2,y2,fx,fy);
end;
tree[root]:=tree[son[root,1]]+tree[son[root,2]];
end;
procedure find(root,x1,y1,x2,y2,fx1,fy1,fx2,fy2:longint); inline;
var
mid:longint;
begin
if (x1=fx1) and (x2=fx2) and (y1=fy1) and (y2=fy2) then
begin
ans:=ans+tree[root];
exit;
end;
if son[root,1]=0 then
exit;
if x2-x1>=y2-y1 then
begin
mid:=(x1+x2) shr 1;
if fx2<=mid then
find(son[root,1],x1,y1,mid,y2,fx1,fy1,fx2,fy2)
else
if fx1>mid then
find(son[root,2],mid+1,y1,x2,y2,fx1,fy1,fx2,fy2)
else
begin
find(son[root,1],x1,y1,mid,y2,fx1,fy1,mid,fy2);
find(son[root,2],mid+1,y1,x2,y2,mid+1,fy1,fx2,fy2);
end;
end
else
begin
mid:=(y1+y2) shr 1;
if fy2<=mid then
find(son[root,1],x1,y1,x2,mid,fx1,fy1,fx2,fy2)
else
if fy1>mid then
find(son[root,2],x1,mid+1,x2,y2,fx1,fy1,fx2,fy2)
else
begin
find(son[root,1],x1,y1,x2,mid,fx1,fy1,fx2,mid);
find(son[root,2],x1,mid+1,x2,y2,fx1,mid+1,fx2,fy2);
end;
end;
end;
begin
readln(n,m);
tot:=1;
for i:=1 to n do
begin
read(a[i]);
change(1,0,0,320000,280000,i+a[i]+100000,i-a[i]+100000);
end;
readln;
for i:=1 to m do
begin
dispose;
if ch='M' then
begin
change(1,0,0,320000,280000,x+y+100000,x-y+100000);
a[x]:=y;
end
else
begin
ans:=0;
find(1,0,0,320000,280000,x+a[x]-y+100000,x-a[x]-y+100000,x+a[x]+y+100000,x-a[x]+y+100000);
writeln(ans);
end;
end;
end.