7.25 线段树浅浅预习
-------------------蒟蒻成长日记
1.不含区间修改的裸线段树模板acwing.1275
题目描述:
给定一个正整数数列 a1,a2,…,ana1,a2,…,an,每一个数都在 0∼p−10∼p−1 之间。
可以对这列数进行两种操作:
- 添加操作:向序列后添加一个数,序列长度变成 n+1n+1;
- 询问操作:询问这个序列中最后 LL 个数中最大的数是多少。
程序运行的最开始,整数序列为空。
一共要对整数序列进行 mm 次操作。
写一个程序,读入操作的序列,并输出询问操作的答案。
输入格式
第一行有两个正整数 m,pm,p,意义如题目描述;
接下来 mm 行,每一行表示一个操作。
如果该行的内容是 Q L
,则表示这个操作是询问序列中最后 LL 个数的最大数是多少;
如果是 A t
,则表示向序列后面加一个数,加入的数是 (t+a) mod p(t+a) mod p。其中,tt 是输入的参数,aa 是在这个添加操作之前最后一个询问操作的答案(如果之前没有询问操作,则 a=0a=0)。
第一个操作一定是添加操作。对于询问操作,L>0L>0 且不超过当前序列的长度。
输出格式
对于每一个询问操作,输出一行。该行只有一个数,即序列中最后 LL 个数的最大数。
输入样例:
10 100
A 97
Q 1
Q 1
A 17
Q 2
A 63
Q 1
Q 1
Q 3
A 99
输出样例:
97
97
97
60
60
97
这是一个很裸的线段树(不含区间查询操作)的简单模板,主要是为以后线段树的题目打基础
以下是对应的代码:
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
struct Node{
int l,r,v;
}tr[4*N];
int n,m;
typedef long long LL;
void pushup(int u)
{
tr[u].v=max(tr[u<<1].v,tr[u<<1|1].v);
}
void build(int u,int l,int r)
{
tr[u]={l,r};
if(l==r)return;
int mid=l+r>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
//pushup(u);
}
void modify(int u,int x,int v)
{
if(tr[u].l==x&&tr[u].r==x)tr[u].v=v;
else
{
int mid=tr[u].l+tr[u].r>>1;
if(x<=mid)modify(u<<1,x,v);
else modify(u<<1|1,x,v);
pushup(u);
}
}
int query(int u,int l,int r)
{
if(tr[u].l>=l&&tr[u].r<=r)return tr[u].v;
int mid=tr[u].l+tr[u].r>>1;
int res=-1;
if(l<=mid)res=query(u<<1,l,r);
if(r>mid)res=max(res,query(u<<1|1,l,r));
return res;
}
int main()
{
char op[2];
int num,last=0,p;
int n=0;
cin>>m>>p;
build(1,1,m);
while(m--)
{
scanf("%s%d",op,&num);
if(*op=='Q')
{
last=query(1,n-num+1,n);
printf("%d\n",last);
}
else
{
modify(1,n+1,LL(num+last)%p);
n++;
}
}
return 0;
}
2.不含区间修改的维护较多信息的线段树模板题
题目描述
给定长度为 NN 的数列 AA,以及 MM 条指令,每条指令可能是以下两种之一:
1 x y
,查询区间 [x,y][x,y] 中的最大连续子段和,即 maxx≤l≤r≤ymaxx≤l≤r≤y{∑i=lrA[i]∑i=lrA[i]}。2 x y
,把 A[x]A[x] 改成 yy。
对于每个查询指令,输出一个整数表示答案。
输入格式
第一行两个整数 N,MN,M。
第二行 NN 个整数 A[i]A[i]。
接下来 MM 行每行 33 个整数 k,x,yk,x,y,k=1k=1 表示查询(此时如果 x>yx>y,请交换 x,yx,y),k=2k=2 表示修改。
输出格式
对于每个查询指令输出一个整数表示答案。
每个答案占一行。
数据范围
N≤500000,M≤100000N≤500000,M≤100000,
−1000≤A[i]≤1000
输入样例:
5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 3 2
输出样例:
2
-1
思路:本题只涉及单点更新,所以可以只使用不带区间修改的线段树就可以解决,关键是维护信息的选择,如图所示我们需要维护的最大连续区间和不仅需要该区间的左端点和右端点,还要包括本区间的最大前缀和和最大后缀和,因为在子区间更新父区间的时候,父区间的最大前缀和可能是横跨两个儿子区间的。
本题的关键在于如何选取维护的信息来达到可以在子区间更新父区间的时候算出需要的最大连续区间和。
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
typedef long long LL;
int n,m;
int q[N];
struct Node
{
int l,r,tmax,lmax,rmax,sum;
}tr[N*4];
void pushup(Node&root,Node&left,Node&right)
{
root.sum=left.sum+right.sum;
root.lmax=max(left.lmax,left.sum+right.lmax);
root.rmax=max(right.rmax,right.sum+left.rmax);
root.tmax=max(max(left.tmax,right.tmax),left.rmax+right.lmax);
}
void pushup(int u)
{
pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
void build(int u,int l,int r)
{
if(l==r)tr[u]={l,r,q[l],q[l],q[l],q[l]};
else
{
tr[u]={l,r};
int mid=l+r>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
pushup(u);
}
}
void modify(int u,int x,int v)
{
if(tr[u].l==x&&tr[u].r==x)tr[u]={x,x,v,v,v,v};
else
{
int mid=tr[u].r+tr[u].l>>1;
if(x<=mid)modify(u<<1,x,v);
else modify(u<<1|1,x,v);
pushup(u);
}
}
Node query(int u,int l,int r)
{
if(tr[u].l>=l&&tr[u].r<=r)return tr[u];
int mid=tr[u].l+tr[u].r>>1;
if(r<=mid)return query(u<<1,l,r);
else if(l>mid)return query(u<<1|1,l,r);
else
{
Node res;
auto left=query(u<<1,l,r);
auto right=query(u<<1|1,l,r);
pushup(res,left,right);
return res;
}
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)scanf("%d",&q[i]);
int k,a,b;
build(1,1,n);
while(m--)
{
scanf("%d%d%d",&k,&a,&b);
// printf("hehe\n");
if(k==1)
{
if(a>b)swap(a,b);
printf("%d\n",query(1,a,b).tmax);
}
else modify(1,a,b);
}
}
线段树的代码都比较长一些,细节很多,很容易写错,也不好调试,只能反复写题来练习熟练度。
3.含区间修改的裸线段树模板 洛谷3372
题目描述
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上 kk。
- 求出某区间每一个数的和。
输入格式
第一行包含两个整数 n, mn,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 nn 个用空格分隔的整数,其中第 ii 个数字表示数列第 ii 项的初始值。
接下来 mm 行每行包含 33 或 44 个整数,表示一个操作,具体如下:
1 x y k
:将区间 [x, y][x,y] 内每个数加上 kk。2 x y
:输出区间 [x, y][x,y] 内每个数的和。
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
输入输出样例
输入 #1复制
5 5 1 5 4 2 3 2 2 4 1 2 3 2 2 3 4 1 1 5 1 2 1 4
输出 #1复制
11 8 20 思路:就是一个裸的带pushdown操作的线段树,注意开longlong就可以了。
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long LL;
struct Node
{
int l,r;
LL sum,lazy;
}tr[4*N];
int n,m;
int w[N];
void pushup(int u)
{
tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void pushdown(int u)
{
Node&root=tr[u];
Node&left=tr[u<<1];
Node&right=tr[u<<1|1];
if(root.lazy)
{
left.sum+=LL(left.r-left.l+1)*root.lazy;
right.sum+=LL(right.r-right.l+1)*root.lazy;
left.lazy+=root.lazy;
right.lazy+=root.lazy;
root.lazy=0;
}
}
void build(int u,int l,int r)
{
if(l==r)tr[u]={l,r,w[l],0};
else
{
tr[u]={l,r};
int mid=l+r>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
pushup(u);
}
}
void modify(int u,int l,int r,int d)
{
if(tr[u].l>=l&&tr[u].r<=r)
{
tr[u].sum+=(LL)(tr[u].r-tr[u].l+1)*d;
tr[u].lazy+=d;
}//注意这里懒标记同时也要处理
else
{
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid)modify(u<<1,l,r,d);
if(r>mid)modify(u<<1|1,l,r,d);
pushup(u);
}
}
LL query(int u,int l,int r)
{
if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
LL sum=0;
if(l<=mid)sum+=query(u<<1,l,r);
if(r>mid)sum+=query(u<<1|1,l,r);
return sum;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)scanf("%d",&w[i]);
build(1,1,n);
while(m--)
{
int k,a,b,c;
scanf("%d",&k);
if(k==1)
{
scanf("%d%d%d",&a,&b,&c);
modify(1,a,b,c);
}
else
{
scanf("%d%d",&a,&b);
printf("%lld\n",query(1,a,b));
}
}
return 0;
}
参考资料:acwing算法提高课,《算法进阶指南》