线段树总结 与 模板

线段树总结 与 模板

单节点更新的 hdu1166 敌兵布阵

#include<cstdio>
#include<queue>
#include<cmath>
#include<iostream>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=100005;
int sum[maxn<<2];
void PushUp(int rt)//单节点更新 求和 求最大 求最小
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
if(l==r)
{
scanf("%d",&sum[rt]);return ;
}
int m=(l+r)>>1;
build(lson);
build(rson);
PushUp(rt);
}
void update(int p,int add,int l,int r,int rt)//对应的区间[l,r]在线段树中的节点(rt)
{
if(l==r){
sum[rt]+=add;return ;
}
int m=(l+r)>>1;
if(p<=m) update(p,add,lson);
else update(p,add,rson);
PushUp(rt);//向上更新父节点
}
int query(int L,int R,int l,int r,int rt)
{
if(L<=l && r<=R)
{
return sum[rt];
}
int m=(l+r)>>1;
int ret=0;//求最大或最小时值相应也要修改
if(L<=m) ret+=query(L,R,lson);//这里为什么是加等于 可以直接赋值吧?  这里是两边都要算  下一行是if
if(R>m)ret += query(L , R , rson);//这里修改成ret = max or min(ret , query(L , R , lson));
return ret;
}
int main()
{
int T , n;
scanf("%d",&T);
for (int cas = 1 ; cas <= T ; cas ++)
{
printf("Case %d:\n",cas);
scanf("%d",&n);
build(1 , n , 1);
char op[10];
while (scanf("%s",op))
{
if (op[0] == 'E') break;
int a,b;
scanf("%d%d",&a,&b);
if (op[0] == 'Q')
printf("%d\n",query(a , b , 1 , n ,1));//a是更新的节点 b是增加的值 最左区间 最右区间 根结点
else if (op[0] == 'S')
update(a , -b , 1 , n , 1);
else
update(a , b , 1 , n , 1);
}
}
return 0;
}

区间增减 区间求和 最大 或最小 poj3468

#include <cstdio>
#include<algorithm>
#include<iostream>
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
#define LL long long
#define inf 0x3f3f3f
using namespace std;
const int maxn = 100005;
LL sum[maxn<<2];
LL add[maxn<<2];
void PushDown(int rt,int m)///m是以rt为节点的区间的长度  
{
    if (add[rt]) {
        add[rt<<1] += add[rt];
        add[rt<<1|1] += add[rt];//因为之前是这个增加值没有向下传递
        sum[rt<<1] += add[rt] * (m - (m >> 1));
        sum[rt<<1|1] += add[rt] * (m >> 1);
        add[rt] = 0;//表示这个节点没有增加的值 或者说是已经向下传递
    }
}
void PushUp(int rt) {
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l,int r,int rt) {
    add[rt] = 0;
    if (l == r) {
        scanf("%lld",&sum[rt]);
        return ;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    PushUp(rt);
}
void update(int L,int R,int c,int l,int r,int rt) {//对应的区间[l,r]在线段树中的节点(rt) p是叶子节点[p,p]
    if (L <= l && r <= R) {
        add[rt] += c;
        sum[rt] += (LL)c * (r - l + 1);
        return ;
    }
    PushDown(rt , r - l + 1);//如果没有找到子节点 那么将值向下传递

    int m = (l + r) >> 1;
    if (L <= m) update(L , R , c , lson);//用p 和 区间[l,r]比较  如果l==r 那么就是p原数组的位置
    if (m < R) update(L , R , c , rson);
    PushUp(rt);
}
LL query(int L,int R,int l,int r,int rt) {//这里是long long 型 要不然会出错  
    if (L <= l && r <= R) {
        return sum[rt];
    }
    PushDown(rt,r-l+1);//查询时 也一样 因为update时并不是每个节点都增加的
    int m = (l + r) >> 1;
    LL ret = 0;
    if (L <= m) ret += query(L , R , lson);//这里为什么是加等于 可以直接赋值吧?
    if (m<R) ret += query(L , R , rson);//这里修改成ret = max or min(ret , query(L , R , lson));
    return ret;
}
int main() {
    int n,q,a,b,c;char ch;
    scanf("%d%d",&n,&q);
    build(1,n,1);
    while(q--)
    {
        scanf(" %c %d %d",&ch,&a,&b);
        if(ch=='Q')
            printf("%d\n",query(a,b,1,n,1));
        else
        {
            scanf("%d",&c);
            update(a,b,c,1,n,1);
        }
    }
    return 0;
}

区间值设置 区间操作

#include <cstdio>
#include <algorithm>
using namespace std;
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
#define LL long long
const int maxn = 111111;
int setv[maxn<<2];
int sum[maxn<<2];
void PushUp(int rt) {
sum[rt] = sum[rt<<1] + sum[rt<<1|1];//跟上面的一样的思路 模拟一遍有助理解
}
void PushDown(int rt,int m) {
if (setv[rt]) {
setv[rt<<1] = setv[rt];
setv[rt<<1|1] = setv[rt];
sum[rt<<1] = setv[rt] * (m - (m >> 1));
sum[rt<<1|1] = setv[rt] * (m >> 1);
setv[rt] = 0;
}
}
void build(int l,int r,int rt) {
setv[rt] = 0;
sum[rt]=0;
if (l == r) {
sum[rt]=1;
return ;
}
int m = (l + r) >> 1;
build(lson);
build(rson);
PushUp(rt);
}
void update(int L,int R,int s,int l,int r,int rt) {
if (L <= l && r <= R) {
setv[rt] = s;
sum[rt] = s * (r - l + 1);
return ;
}
PushDown(rt , r - l + 1);
int m = (l + r) >> 1;
if (L <= m) update(L , R , s , lson);
if (m < R) update(L , R , s, rson);
PushUp(rt);
}
int query(int l,int r,int rt) {
if (setv[rt] || l==r){
return sum[rt];
}
PushDown(rt , r - l + 1);
int m = (l + r) >> 1;
int ret = 0;
ret += query(lson);
ret += query(rson);
return ret;
}
int main() {
int T;scanf("%d",&T);
for(int i=1;i<=T;i++)
{
int N,Q;
scanf("%d%d",&N,&Q);
build(1 , N , 1);
while (Q--)
{
int a,b,s;
scanf("%d%d%d",&a,&b,&s);
update(a,b,s,1,N,1);
}
printf("Case %d: The total value of the hook is %d.\n",i,query(1 , N , 1));
}
return 0;
}