zoukankan      html  css  js  c++  java
  • 线段树模板1

    功能

    连续区间加
    连续区间和

    //#pragma warning(disable:4996)
    #include <iostream>
    #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;
    LL add[maxn << 2];
    LL sum[maxn << 2];
    
    // PushUP(int rt)是把当前结点的信息更新到父结点
    // PushDown(int rt)是把当前结点的信息更新给儿子结点
    void PushUp(int rt) {			
    	sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
    }
    
    void PushDown(int rt, int m) {			
    	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 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) {
    	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);
    	if (m < R) update(L, R, c, rson);
    	PushUp(rt);
    }
    //区间和
    LL query(int L, int R, int l, int r, int rt) {
    	if (L <= l && r <= R) {
    		return sum[rt];
    	}
    	PushDown(rt, r - l + 1);
    	int m = (l + r) >> 1;
    	LL ret = 0;
    	if (L <= m) ret += query(L, R, lson);
    	if (m < R) ret += query(L, R, rson);
    	return ret;
    }
    
    int main() {
    	int N, Q;
    	scanf("%d%d", &N, &Q);
    	build(1, N, 1);
    	while (Q--) {
    		int op;
    		int a, b, c;
    		scanf("%d", &op);
    		if (op == 1) {
    			scanf("%d%d%d", &a, &b, &c);
    			update(a, b, c, 1, N, 1);
    		}
    		else {
    			scanf("%d%d", &a, &b);
    			printf("%lld
    ", query(a, b, 1, N, 1));
    		}
    
    	}
    
    	return 0;
    
    }
    
  • 相关阅读:
    hash
    C#执行Sql事务处理
    数据库的锁表
    页面的刷新 和图片的替换
    单点登录 Webservice
    js 动态调用js文件
    .net生成EXCEL
    JS : 连续滚动
    引用指定类型的对象
    对象序列化为字符串
  • 原文地址:https://www.cnblogs.com/CSE-kun/p/13943705.html
Copyright © 2011-2022 走看看