A Simple Problem with Integers POJ - 3468 线段树区间修改+区间查询
2024-09-06 20:21:17
//add,懒标记,给以当前节点为根的子树中的每一个点加上add(不包含根节点)
//
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = ;
int n, m;
int w[N];
struct Node
{
int l, r;
//总和
//如果只考虑当前节点及子节点上的标记,当前区间和是多少 ,没考虑所有祖先节点上的标记
LL sum;
//懒标记
//给当前区间的所有儿子加上add
LL add;
}tr[N * ];
//用子节点的信息来计算父节点的信息
void pushup(int u)
{
tr[u].sum = tr[u << ].sum + tr[u << | ].sum;
}
void pushdown(int u)
{
Node &root = tr[u], &left = tr[u << ], &right = tr[u << | ];
//如果当前根节点有标记,往下传,还要清空
if (root.add)
{
left.add += root.add;
left.sum += (LL)(left.r - left.l + ) * root.add;
right.add += root.add;
right.sum += (LL)(right.r - right.l + ) * root.add;
root.add = ;
}
}
void build(int u, int l, int r)
{
if (l == r)
tr[u] = {l, r, w[r], };
else
{
tr[u] = {l, r};
int mid = l + r >> ;
build(u << , l, mid);
build(u << | , mid + , 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 + ) * d;
//懒标记
tr[u].add += d;
}
// 区间太大,一定要分裂
else
{
pushdown(u);
int mid = tr[u].l + tr[u].r >> ;
if (l <= mid)
modify(u << , l, r, d);
if (r > mid)
modify(u << | , 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 >> ;
LL sum = ;
if (l <= mid)
sum = query(u << , l, r);
if (r > mid)
sum += query(u << | , l, r);
return sum;
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = ; i <= n; i ++ )
scanf("%d", &w[i]);
build(, , n);
char op[];
int l, r, d;
while (m -- )
{
scanf("%s%d%d", op, &l, &r);
if (*op == 'C')
{
scanf("%d", &d);
modify(, l, r, d);
}
else printf("%lld\n", query(, l, r));
}
return ;
}
最新文章
- [C#] Linq To Objects - 如何操作文件目录
- for循环的执行顺序
- 原生JS实现轮播+学前端的感受(防止走火入魔)
- css3 @font-face设置嵌入字体
- JavaScript Math 对象
- Jquery_AjaxFileUpload插件的使用记录
- java实现求数组中元素第二大的元素
- MFC双缓冲绘图实例
- Win7局域网文件共享方法
- How to send Email using C#
- WebApi及Fiddler工具
- [国嵌攻略][060][LCD工作原理解析]
- Docker学习笔记 - Docker容器与外部网络的连接
- idea 模板注释设置
- IT面试技巧终身受益
- flask使用基础
- nginx实践(二)之静态资源web服务(浏览器缓存场景)
- Behavior Question - Most challenging project.
- MySql查询时间段的方法(转)
- Lintcode: Unique Paths
热门文章
- eclipse里新建work set,将项目分组放在不同文件夹
- Thingsboard源码安装部署
- java sql语句 like%?%报错的问题
- 终于解决 k8s 集群中部署 nodelocaldns 的问题
- numpy 介绍与使用
- light oj 1014 - Ifter Party分解因子
- jps jmap 的使用
- C# bubble sort,selection sort,insertion sort
- ES6 - 基础学习(5): 数值扩展
- 区间操作---树状数组&;&;线段树