您的位置:首页 > 其它

poj 3468 线段树区间更新

2016-07-24 08:59 232 查看
区间更新和区间查询有点类似

区间更新是由上往下的

每到一个点 判断这个点所代表的区间是不是在要更改的区间内 

是的话在这个点更改这个点的数值标记 并且把这个点的值在原来值的基础增减(这个点区间范围*更改值 )并且返回 (不一定到最底层)

如果这个点本身的区间和他的子区间都不满足目标区间的范围 直接返回

否则 pushdown  再进入子区间进行判断  

要注意返回后的路径上的值的更新

因为是自上而下的 所以一直到更新完目标区间 根节点到目标区间路径上点都被更新了

查询的时候 也是把查询路径上的点一路pushdown 所以最后的答案也是更新过的

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cctype>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
#define pi acos(-1.0)
#define inf 1<<29
#define INF 0x3f3f3f3f
#define zero 1e-8

const int li[] = { -1, 0, 1, 0};
const int lj[] = {0, -1, 0, 1};

const int N = 1e5 + 10;

using namespace std;

struct node {

long long data;
long long tag;

} tree[N * 4];

int arr
;

void build(int node, int Begin, int End)
{
tree[node].tag = 0;
if (Begin == End) {
tree[node].data = arr[Begin];
return;
}
build(node * 2, Begin, (Begin + End) / 2);
build(node * 2 + 1, (Begin + End) / 2 + 1, End);
tree[node].data = tree[node * 2].data + tree[node * 2 + 1].data;
}
long long fin;

void pushdown(int node, int l, int r)
{

tree[node * 2].tag += tree[node].tag;
tree[node * 2].data += tree[node].tag * ((l + r) / 2 - l + 1);
tree[node * 2 + 1].tag += tree[node].tag;
tree[node * 2 + 1].data += tree[node].tag * (r - (l + r) / 2);
tree[node].tag = 0;
}

void update(int node, int b, int e, int l, int r, long long data)
{

if (l > e || r < b) return;

if (l >= b && r <= e) {
tree[node].tag += data;
tree[node].data += data * (r - l + 1);
return;
}

pushdown(node, l, r);
update(node * 2, b, e, l, (l + r) / 2, data);
update(node * 2 + 1, b, e, (l + r) / 2 + 1, r, data);

tree[node].data = tree[node * 2].data + tree[node * 2 + 1].data;
}

void query(int node, int b, int e, int l, int r)
{

if (l > e || r < b) return;

if (l >= b && r <= e) {
fin += tree[node].data;
return;
}
pushdown(node, l, r);
query(node * 2, b, e, l, (l + r) / 2);
query(node * 2 + 1, b, e, (l + r) / 2 + 1, r);
}
int main()
{

int n, q;
scanf("%d %d", &n, &q);

for (int i = 1; i <= n; ++i)
scanf("%d", &arr[i]);

build(1, 1, n);

for (int i = 0; i < q; ++i) {

char ch[10];
scanf("%s", ch);
if (ch[0] == 'C') {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
update(1, a, b, 1, n, c);
} else if (ch[0] == 'Q') {
int a, b;
scanf("%d%d", &a, &b);
fin = 0;
query(1, a, b, 1, n);
printf("%lld\n", fin);
}
}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: