ST表与树状数组

ST表 

 st表可以解决区间最值的问题。可以做到O(nlogn)预处理 ,O(1)查询,但是不支持修改。

  st表的大概思路就是用st[i][j]来表示从i开始的2的j次方个树中的最值,查询时就从左端点开始,找到区间长度是2的多少次方,然后进行查询。然而,很明显,我们要查询的区间长度不一定是2的多少次幂。那怎么做到O(1)查询呢,这就要用到最值的特性。

 

如图,假如我们要查询2到7之间的最大值,但是7-2+1在22与23之间,我们选择22,也就是st[2][2],那剩下的6,7怎么办,我们考虑倒着从7往回算,也就是在st[7-22][2]与st[2][2]取max作为从2到7的最大值。

  首先,进行预处理,st[i][j]表示从i开始的2的j次方,那么st[i][j]就应该是从i开始2的j-1次方与从i+2j-1开始的2的j-1次方中的最大值,那么进行递推就好了。

代码:

 

 1 #include<cstdio>
 2 #include<iostream>
 3 using namespace std;
 4 const int N=100100;
 5 int n,m,a[N],st[N][20],log[N],cf[20];
 6 void pre()
 7 {
 8   log[2]=1;
 9   log[1]=0;
10   for(int i=3;i<=n;++i)
11   {
12     log[i]=log[i/2]+1;
13   }
14   cf[0]=1;
15   cf[1]=2;
16   for(int i=2;i<=log[n]+1;++i)
17   {
18     cf[i]=cf[i-1]*2;
19   }
20 }
21 int read()
22 {
23   int x=0,f=1;char c=getchar();
24   while(c<'0'||c>'9')
25   {
26     if(c=='-') f=-1;
27     c=getchar();
28   }
29   while(c>='0'&&c<='9')
30   {
31     x=x*10+c-'0';
32     c=getchar();
33   }
34   return x;
35 }
36 int ff(int x,int y)
37 {
38   int l=y-x+1,k=log[l];
39   int f=max(st[x][k],st[y-cf[k]+1][k]);
40   return f;
41 }
42 int main()
43 {
44   n=read(),m=read();
45   pre();
46   for(int i=1;i<=n;++i)
47   {
48     st[i][0]=a[i]=read();
49 
50   }
51 
52   for(int j=1;j<=log[n];++j)
53   {
54     for(int i=1;i+cf[j]-1<=n;++i)
55     {
56       st[i][j]=max(st[i][j-1],st[i+cf[j-1]][j-1]);
57     }
58   }
59   for(int i=1,x,y;i<=m;++i)
60   {
61     x=read(),y=read();
62     cout<<ff(x,y)<<"\n";
63   }
64   return 0;
65 }

 

 

 

 树状数组

其实,树状数组的原理我并不是很懂,但是因为其短小精炼的代码,令我非常喜欢。。。。

树状数组不需要预处理,只有修改与查询两种操作。修改可以是加或减一个值,查询的是一个区间和。

首先我们需要一个数组tree。

然后就是修改,只需写一个几行的子函数,然后将修改元素的下标和要加的元素传入函数,然后奇迹就发生了。

查询传入要查询的下标就可以查询从1到改元素之间的区间和,如果查询从l到r,只需分别求出其从以到他们的区间和,然后相减即可,类似于前缀和。具体的都在代码里了。

 1 #include<cstdio>
 2 #include<iostream>
 3 using namespace std;
 4 const int N=500005;
 5 int tree[N],n,m;
 6 void add(int a,int pos)
 7 {
 8     while(pos<=n){
 9         tree[pos]+=a;
10         pos+=pos&-pos;
11     }
12 }
13 int cc(int pos)
14 {
15     int ans=0;
16     while(pos>=1){
17         ans+=tree[pos];
18         pos-=pos&-pos;
19     }
20     return ans;
21 }
22 int main()
23 {
24     scanf("%d%d",&n,&m);
25     for(int i=1,x;i<=n;++i)
26     {
27         scanf("%d",&x);
28         add(x,i);
29     }
30     for(int i=1,bz,x,y;i<=m;++i){
31         scanf("%d%d%d",&bz,&x,&y);
32         if(bz==1)
33             add(y,x);
34         else{
35             int kk=cc(y),zz=cc(x-1);
36             printf("%d\n",kk-zz);
37         }
38     }
39     return 0;
40 }

 上面讲的是树状数组的单点修改和区间查询,下面写一下,树状数组的区间修改单点查询。

首先介绍一下差分数组和前缀和。

前缀和就是记录前几个数的和,差分数组呢就是记录当前位置减去前一个位置的数的差。

然后要用到一个前缀和与差分数组的性质:前缀和数组的差分数组是原数组,差分数组的前缀和是原数组。

证明很显然,动手一推就知道了。

那么这与树状数组有什么关系呢,通过上面那个树状数组,我们知道,树状数组可以记录前几个数的和,现在我们要做的是区间修改和单点查询。

这时又要用到一个差分数组的一个性质。

差分数组进行区间加减时比较方便,比如如果将从i到j之间的数加k,那么只需将他们的差分数组i位置+k,并且将j+1位置的数-k即可。

证明同样很显然,动手推。

那么这就有联系了,我们树状数组维护的相当于是前缀和,然后我们如果用他维护原数组的差分数组,那么他就相当于维护的原数组,这样利用第一个性质单点查询就解决了。再就是区间修改,因为我们维护的是差分数组,所以利用性质二进行区间修改就好了。

看代码就很清楚明了了

 1 #include<cstdio>
 2 #include<iostream>
 3 using namespace std;
 4 const int N=500010;
 5 int n,m,a[N],tree[N];
 6 void read(int &x)
 7 {
 8   x=0;int f=1;char c=getchar();
 9   while(c<'0'||c>'9') {  if(c=='-') f=-1; c=getchar(); }
10   while(c>='0'&&c<='9'){ x=x*10+c-'0'; c=getchar();}
11   x=x*f;
12 }
13 void add(int pos,int w)
14 {
15   while(pos<=n)
16   {
17     tree[pos]+=w;
18     pos+=pos&-pos;
19   }
20   return;
21 }
22 int ff(int pos)
23 {
24   int ans=0;
25   while(pos>=1)
26   {
27     ans+=tree[pos];
28     pos-=pos&-pos;
29   }
30   return ans;
31 }
32 
33 
34 int main()
35 {
36   int last=0;
37   read(n),read(m);
38   for(int i=1,x;i<=n;++i)
39   {
40     read(x);
41     add(i,x-last);
42     last=x;
43   }
44   for(int i=1,x,y,z,k;i<=m;++i)
45   {
46     read(x);
47     if(x==1)
48     {
49       read(y),read(z),read(k);
50       add(y,k);
51       add(z+1,-k);
52     }
53     else if(x==2)
54     {
55       read(y);
56       printf("%d\n",ff(y));
57     }
58   }
59   return 0;
60 }

 

全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务