首页 > 代码库 > poj3468A Simple Problem with Integers区间和线段树
poj3468A Simple Problem with Integers区间和线段树
同上。。。
#include <cstdio>#include <cstring>#include <algorithm>#include <climits>#include <string>#include <iostream>#include <map>#include <cstdlib>#include <list>#include <set>#include <queue>#include <stack>#include<math.h>using namespace std;typedef long long LL;#define lson l,mid,rt<<1#define rson mid+1,r,rt<<1|1LL sum[444444];LL color[444444];void up(LL rt){ sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];}void down(LL rt, LL len){ if (color[rt]){ color[rt << 1] += color[rt]; color[rt << 1 | 1] += color[rt]; LL mid = len >> 1; sum[rt << 1] += color[rt] * (len - mid); sum[rt << 1 | 1] += color[rt] * mid; color[rt] = 0; }}void build(LL l, LL r, LL rt){ color[rt] = 0; if (l == r){ scanf("%I64d", &sum[rt]); return; } LL mid = (l + r) >> 1; build(lson); build(rson); up(rt);}void update(LL L, LL R, LL ad, LL l, LL r, LL rt){ if (L <= l&&r <= R){ sum[rt] += ad*(r - l + 1); color[rt] += ad; return; } down(rt, r - l + 1); LL mid = (l + r) >> 1; if (L <= mid) update(L, R, ad, lson); if (R>mid) update(L, R, ad, rson); up(rt);}LL ask(LL L, LL R, LL l, LL r, LL rt){ if (L <= l&&r <= R) return sum[rt]; LL mid = (l + r) >> 1; LL ans = 0; down(rt, r - l + 1); if (L <= mid) ans += ask(L, R, lson); if (R>mid) ans += ask(L, R, rson); return ans;}int main(){ LL n, m; LL a, b, c; char str[100]; while (~scanf("%I64d%I64d", &n, &m)){ build(1, n, 1); for (LL i = 0; i<m; i++){ scanf("%s", str); scanf("%I64d%I64d", &a, &b); if (str[0] == ‘C‘){ scanf("%I64d", &c); update(a, b, c, 1, n, 1); } else{ printf("%I64d\n", ask(a, b, 1, n, 1)); } } } return 0;}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。