介绍
Zkw线段是是一种非递归式线段树,Zkw线段树有很多优点:1. 代码简短;2. 纯天然非递归;3. 常数小(尤其在差分区间更新时)。
它采用堆结构,构造一颗满二叉树(也可以说是完全二叉树),而二叉树的最后一层则是各个节点
注意:zkw线段树必须是点树,即完全闭区间
普通线段树中的修改需要去查询节点,并分为三类:
完全覆盖,在左区间,在右区间
那么换个方法,自底向上更新呢?
这就是zkw的想法
具体的介绍还请看下面的参考博客:
非递归式(zkw)线段树详解(一)
zkw线段树详解
[数据结构]走近Zkw线段树(一)
例子
单点修改,区间求和
以HDU1166 敌兵布阵为例子:
#include<cstdio>
#include<cstring>
#include<string>
#include<set>
#include<iostream>
#include<stack>
#include<queue>
#include<vector>
#include<algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define mod 10000007
#define debug() puts("what the fuck!!!")
#define ll long long
using namespace std;
const int N=50000+20;
int M,n,tree[N<<2];
void pushup(int i)//向上更新
{
tree[i]=tree[i<<1]+tree[i<<1|1];
}
void update(int x,int v)//单点修改,把x的值增加v
{
for(tree[x+=M]+=v,x>>=1; x; x>>=1)
pushup(x);
}
int query(int l,int r)//求[l~r]的区间和
{
int ans=0;
//l^r^1判断是否是兄弟节点,>0时是兄弟节点
//是兄弟是退出循环
for(l=l+M-1,r=r+M+1; l^r^1; l>>=1,r>>=1)
{
//l&1^1,r&1分别判断偶数和奇数
if(l&1^1) ans+=tree[l^1];//l+1
if(r&1) ans+=tree[r^1];//r-1
}
return ans;
}
void build()
{
for(M=1; M<n; M<<=1);
for(int i=M+1; i<=M+n; i++)
scanf("%d",&tree[i]);
for(int i=M; i>=1; i--)
pushup(i);
}
int main()
{
int t,q=1,a,b;
char op[10];
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
mem(tree,0);
printf("Case %d:\n",q++);
build();
while(scanf("%s",op)&&strcmp(op,"End")!=0)
{
scanf("%d%d",&a,&b);
if(op[0]=='Q')
printf("%d\n",query(a,b));
else if(op[0]=='S')
update(a,-b);
else
update(a,b);
}
}
return 0;
}
改前:
#include<cstdio>
#include<cstring>
#include<string>
#include<set>
#include<iostream>
#include<stack>
#include<queue>
#include<vector>
#include<algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define mod 10000007
#define debug() puts("what the fuck!!!")
#define ll long long
using namespace std;
#define MAXN (50000+10)
#define MAXAi (50+10)
int n,m,k,M,tt,t[MAXN*5];
char s[50];
int main()
{
scanf("%d",&tt);
for(int k=1; k<=tt; k++)
{
printf("Case %d:\n",k);
scanf("%d",&n);
memset(t,0,sizeof(t));
for(M=1; M<n; M<<=1);
printf("M=%d\n",M);
for(int i=M+1; i<=M+n; i++)
{
scanf("%d",&t[i]);
printf("t[%d]=%d\n",i,t[i]);
}
printf("*******\n");
for(int i=M; i>=1; i--)
{
t[i]=t[i<<1]+t[(i<<1)^1];
printf("t[%d]=%d\n",i,t[i]);
}
while(scanf("%s",s)!=EOF)
{
if(!strcmp(s,"End")) break;
int p1,p2;
scanf("%d%d",&p1,&p2);
if(s[0]=='Q')
{
int ans=0;
p1--;
p2++;
p1+=M;
p2+=M;
while(p1^p2^1>0)
{
if(~p1&1) ans+=t[p1+1];
if(p2&1) ans+=t[p2-1];
p1>>=1;
p2>>=1;
}
cout<<ans<<endl;
}
else
{
p1+=M;
if(s[0]=='A') t[p1]+=p2;
else t[p1]-=p2;
for(p1/=2; p1; p1/=2)
{
t[p1]=t[p1<<1]+t[(p1<<1)^1];
}
}
}
}
return 0;
}
单点修改,区间求最值
以HDU1754为例
#include<cstdio>
#include<cstring>
#include<string>
#include<set>
#include<iostream>
#include<stack>
#include<queue>
#include<vector>
#include<algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define mod 10000007
#define debug() puts("what the fuck!!!")
#define ll long long
using namespace std;
const int N=1000000+20;
int M,n,m,tree[N<<2];
void pushup(int i)//向上更新
{
tree[i]=max(tree[i<<1],tree[i<<1|1]);
}
void update(int x,int v)//单点修改,把x的值变成v
{
for(tree[x+=M]=v,x>>=1; x; x>>=1)
pushup(x);
}
int query(int l,int r)//求[l~r]的最大值
{
int ans=0;
//l^r^1判断是否是兄弟节点,>0时是兄弟节点
//是兄弟是退出循环
for(l=l+M-1,r=r+M+1; l^r^1; l>>=1,r>>=1)
{
//l&1^1,r&1分别判断偶数和奇数
if(l&1^1) ans=max(ans,tree[l^1]);//l+1
if(r&1) ans=max(ans,tree[r^1]);//r-1
}
return ans;
}
void build()
{
for(M=1; M<n; M<<=1);
for(int i=M+1; i<=M+n; i++)
scanf("%d",&tree[i]);
for(int i=M; i>=1; i--)
pushup(i);
}
int main()
{
char op[10];
int a,b;
while(~scanf("%d%d",&n,&m))
{
build();
while(m--)
{
scanf("%s%d%d",op,&a,&b);
if(op[0]=='Q')
printf("%d\n",query(a,b));
else
update(a,b);
}
}
return 0;
}