当前位置: 代码迷 >> 综合 >> poj - 3468 - A Simple Problem with Integers(线段树(区间更新))
  详细解决方案

poj - 3468 - A Simple Problem with Integers(线段树(区间更新))

热度:36   发布时间:2024-01-10 13:56:14.0

题意:给出一个数组A,数组来度为N,接着来Q个操作,操作有两种,C a b c:将数组A在[a, b]上的值全部加c,Q a b:询问数组A在[a, b]上的区间和。

题目链接:http://poj.org/problem?id=3468

——>>数据量之大,直接for统计与修改的结果该是TLE,这也正是线段树区间修改的模版题,照着白书加强版的思路与提示WA了十余次也没过,最后换了下传增量的写法,终于给了我一个AC!

        原想,8个叶子的线段树,就2*8-1个结点,对于maxn个叶子,开2*maxn的空间够了,事实证明,不够,对于本题来说,开3*maxn的空间足够,不过发现比较多的写法开了4*maxn的空间,空间不够,我尝到了两种结果,一种是RE,另一种是WA。

        把查询区间传到函数参数里,写出来的代码长长的(1594ms),于是改用全局变量来做,却发现,这样做运行时间变长了(1657ms)。

全局变量写法:

#include <iostream>
#include <cstdio>
#include <cstring>using namespace std;const int maxn = 100000 + 10;
long long sumv[3*maxn], addv[3*maxn], a[maxn], _sum, y1, y2, v;
//sumv[o]表示如果只执行结点o及其子孙结点的add的区间和,_sum为要求的结果,a为输入数组,addv[o]表示结点o被加了多少,y1、y2为改动区间,v为改动量void build(int o, int L, int R)     //建树,给sumv赋初值
{if(L == R) sumv[o] = a[L];else{int M = L + (R-L) / 2;if(L <= M) build(2*o, L, M);if(R > M) build(2*o+1, M+1, R);sumv[o] = sumv[2*o] + sumv[2*o+1];}
}
void update(int o, int L, int R)        //更新
{if(y1 <= L && R <= y2){addv[o] += v;       //记录增量return;}else if(L <= y1 && y2 <= R) sumv[o] += v * (y2-y1+1);       //加上增量else if(y1 <= L && y2 <= R) sumv[o] += v * (y2-L+1);        //加上增量else if(L <= y1 && R <= y2) sumv[o] += v * (R-y1+1);        //加上增量int M = L + (R-L) / 2;if(y1 <= M) update(2*o, L, M);if(y2 > M) update(2*o+1, M+1, R);
}
void query(int o, int L, int R)     //查询
{if(y1 <= L && R <= y2){_sum += sumv[o] + addv[o] * (R-L+1);        //加上自己的增量return;}addv[2*o] += addv[o];       //增量下传addv[2*o+1] += addv[o];     //增量下传sumv[o] += addv[o] * (R-L+1);       //更新sumv值addv[o] = 0;        //修改增量int M = L + (R-L) / 2;if(y1 <= M) query(2*o, L, M);if(y2 > M) query(2*o+1, M+1, R);
}
int main()
{int N, Q, i;char s[7];scanf("%d%d", &N, &Q);for(i = 1; i <= N; i++) scanf("%I64d", &a[i]);build(1, 1, N);     //建树memset(addv, 0, sizeof(addv));while(Q--){scanf("%s", s);if(s[0] == 'Q'){scanf("%I64d%I64d", &y1, &y2);_sum = 0;query(1, 1, N);printf("%I64d\n", _sum);}else{scanf("%I64d%I64d%I64d", &y1, &y2, &v);update(1, 1, N);}}return 0;
}

传参写法:

#include <iostream>
#include <cstdio>
#include <cstring>using namespace std;const int maxn = 100000 + 10;
long long sumv[3*maxn], addv[3*maxn], a[maxn], _sum;
//sumv[o]表示如果只执行结点o及其子孙结点的add的区间和,_sum为要求的结果,a为输入数组,addv[o]表示结点o被加了多少void build(int o, int L, int R)     //建树,给sumv赋初值
{if(L == R) sumv[o] = a[L];else{int M = L + (R-L) / 2;if(L <= M) build(2*o, L, M);if(R > M) build(2*o+1, M+1, R);sumv[o] = sumv[2*o] + sumv[2*o+1];}
}
void update(int o, int L, int R, int y1, int y2, long long v)       //更新
{if(y1 <= L && R <= y2){addv[o] += v;       //记录增量return;}else if(L <= y1 && y2 <= R) sumv[o] += v * (y2-y1+1);       //加上增量else if(y1 <= L && y2 <= R) sumv[o] += v * (y2-L+1);        //加上增量else if(L <= y1 && R <= y2) sumv[o] += v * (R-y1+1);        //加上增量int M = L + (R-L) / 2;if(y1 <= M) update(2*o, L, M, y1, y2, v);if(y2 > M) update(2*o+1, M+1, R, y1, y2, v);
}
void query(int o, int L, int R, int y1, int y2)     //查询
{if(y1 <= L && R <= y2){_sum += sumv[o] + addv[o] * (R-L+1);        //加上自己的增量return;}addv[2*o] += addv[o];       //增量下传addv[2*o+1] += addv[o];     //增量下传sumv[o] += addv[o] * (R-L+1);       //更新sumv值addv[o] = 0;        //修改增量int M = L + (R-L) / 2;if(y1 <= M) query(2*o, L, M, y1, y2);if(y2 > M) query(2*o+1, M+1, R, y1, y2);
}
int main()
{int N, Q, i;long long y1, y2, v;char s[7];scanf("%d%d", &N, &Q);for(i = 1; i <= N; i++) scanf("%I64d", &a[i]);build(1, 1, N);     //建树memset(addv, 0, sizeof(addv));while(Q--){scanf("%s", s);if(s[0] == 'Q'){scanf("%I64d%I64d", &y1, &y2);_sum = 0;query(1, 1, N, y1, y2);printf("%I64d\n", _sum);}else{scanf("%I64d%I64d%I64d", &y1, &y2, &v);update(1, 1, N, y1, y2, v);}}return 0;
}

  相关解决方案