应用:搜索区间最值,总和,更新区间,更新点。时间复杂度:O(log n)
一直用一分为二的方法将区间数组层层变小,可以看成是一颗二叉树,故名线段树。
记录树状数组求和及点修改模板代码:
#include<stdio.h> #include<algorithm> #include<iostream> using namespace std; #define maxn 100007 int sum[maxn*2*2];///sum求和 int a[maxn],n;///存原数组,下标为n-1 ///l,r表示当前节点区间,rt表示当前节点编号 void build(int l,int r,int rt)///建树 {/// if(l==r)///到达叶节点 { sum[rt]=a[l];///存储数组值 return ; } int m=(l+r)/2;///左右递归 build(l,m,rt*2); build(m+1,r,rt*2+1); ///更新信息,子节点更新了,所以本节点也需要更新信息 sum[rt]=sum[rt*2]+sum[rt*2+1];///只有拆分的点才走一步,叶节点直接标记sum数组了 } ///点修改,假设a[p]=a[p]+3 ///l,r表示当前节点区间,rt表示当前节点编号 void update(int p,int c,int l,int r,int rt) { if(l==r)///到达叶节点,修改后返回 { sum[rt]=sum[rt]+c; return ; } int m=(l+r)/2; ///根据条件判断往左调用还是往右 if(p<=m) update(p,c,l, m, rt*2); else update(p,c,m+1, r, rt*2+1); sum[rt] = sum[rt*2] + sum[rt*2+1]; } ///区间查询, L,R表示操作区间 int query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R)///在区间内,直接返回 return sum[rt]; int m=(l+r)/2; ///累计答案 int ans=0; if(L<=m) ans=ans+query(L,R,l,m,rt*2); if(R>m) ans=ans+query(L,R,m+1,r,rt*2+1); return ans; } int main() { cin>>n;///n=13 for(int i=1;i<=n;i++) cin>>a[i]; build(1,n,1); int p,c; cin>>p>>c;///p=3; c=7 update(p,c,1,n,1); int L,R; cin>>L>>R;///L=5; R=8 cout<<query(L,R,1,n,1)<<endl; return 0; }
NYOJ119
士兵杀敌(三)
时间限制:2000 ms | 内存限制:65535 KB
难度:5
- 描述
-
南将军统率着N个士兵,士兵分别编号为1~N,南将军经常爱拿某一段编号内杀敌数最高的人与杀敌数最低的人进行比较,计算出两个人的杀敌数差值,用这种方法一方面能鼓舞杀敌数高的人,另一方面也算是批评杀敌数低的人,起到了很好的效果。
所以,南将军经常问军师小工第i号士兵到第j号士兵中,杀敌数最高的人与杀敌数最低的人之间军功差值是多少。
现在,请你写一个程序,帮小工回答南将军每次的询问吧。
注意,南将军可能询问很多次。
- 输入
- 只有一组测试数据
第一行是两个整数N,Q,其中N表示士兵的总数。Q表示南将军询问的次数。(1<N<=100000,1<Q<=1000000)
随后的一行有N个整数Vi(0<=Vi<100000000),分别表示每个人的杀敌数。
再之后的Q行,每行有两个正正数m,n,表示南将军询问的是第m号士兵到第n号士兵。 - 输出
- 对于每次询问,输出第m号士兵到第n号士兵之间所有士兵杀敌数的最大值与最小值的差。
- 样例输入
-
5 2 1 2 6 9 3 1 2 2 4
- 样例输出
-
1 7
AC代码:
#include<stdio.h> #include<iostream> #include<algorithm> #include<map> #define ll long long using namespace std; const int maxx=1e6+5; int n,t,L,R; int big,small;///全局变量存储各区间的最值 struct node { int maxa,minn; };///结构体保存最大最小值 int a[maxx]; node sum[4*maxx]; void build(int l,int r,int rt)///建立线段树 { if(l==r)///遇到叶子节点,递归出口 { sum[rt].maxa=a[l]; sum[rt].minn=a[l]; return ; } int mid=(l+r)/2; build(l,mid,rt*2); build(mid+1,r,rt*2+1); sum[rt].maxa=max( sum[rt*2].maxa,sum[rt*2+1].maxa );///同时更新父亲结点的最值 sum[rt].minn=min( sum[rt*2].minn,sum[rt*2+1].minn ); } void query1(int l,int r,int rt)///更变查询区间 { if(L<=l && r<=R) { big=max(big,sum[rt].maxa);///更新最值 small=min(small,sum[rt].minn); return ; } int mid=(l+r)/2; if(L<=mid) query1(l,mid,rt*2); if(R>mid) query1(mid+1,r,rt*2+1); } int main() { scanf("%d%d",&n,&t); for(int i=1;i<=n;i++) scanf("%d",&a[i]); build(1,n,1); while(t--) { scanf("%d%d",&L,&R); big=-1,small=100000000;///每次初始化最值 query1(1,n,1); printf("%d\n",big-small);///最值相减 } return 0; }