首页 > 代码库 > POJ 3468 A Simple Problem with Integers
POJ 3468 A Simple Problem with Integers
题目大意:给一组数据,Q 为查询求a,b区间和,C为为区间a,b间的各个元素都加上c。
题目思路:线段树模板
#include<iostream>#include<algorithm>#include<cstring>#include<vector>#include<stdio.h>#include<queue>#include<math.h>#define INF 0x3f3f3f3f#define MAX 1000005#define Temp 1000000000using namespace std;long long a[MAX];struct node{ int l,r,root; long long s,e;}tree[MAX];void MakeTree(int L,int R,int root){ tree[root].l=L; tree[root].r=R; tree[root].e=0; int Mid=(L+R)/2; if(L==R) { tree[root].s=a[L]; return; } MakeTree(L,Mid,root*2); MakeTree(Mid+1,R,root*2+1); tree[root].s=tree[root*2].s+tree[root*2+1].s;}void Down(int root){ tree[root*2].s+=(tree[root*2].r - tree[root*2].l + 1)*tree[root].e; tree[root*2+1].s+=(tree[root*2+1].r - tree[root*2+1].l + 1)*tree[root].e; tree[root*2].e+=tree[root].e; tree[root*2+1].e+=tree[root].e; tree[root].e=0;}void UpDate(int L,int R,int date,int root){ tree[root].s+=(R-L+1)*date; int Mid=(tree[root].l + tree[root].r)/2; if(tree[root].l==L && tree[root].r==R) { return; } Down(root); if(R <= Mid) UpDate(L,R,date,root*2); else if(L > Mid) UpDate(L,R,date,root*2+1); else { UpDate(Mid+1,R,date,root*2+1); UpDate(L,Mid,date,root*2); }}long long Query(int L,int R,int root){ if(tree[root].l==L && tree[root].r==R) { return tree[root].s; } Down(root); int Mid=(tree[root].l + tree[root].r)/2; if(R <= Mid) return Query(L,R,root*2); else if(L > Mid) return Query(L,R,root*2+1); else { long long lsum=Query(L,Mid,root*2); long long rsum=Query(Mid+1,R,root*2+1); return lsum+rsum; }}int main(){ int n,m,l,r,date; char op[2]; while(scanf("%d%d",&n,&m)!=EOF) { for(int i=1;i<=n;i++) { scanf("%lld",&a[i]); } MakeTree(1,n,1); while(m--) { scanf("%s",op); if(op[0]==‘Q‘) { scanf("%d%d",&l,&r); long long ans = Query(l,r,1); printf("%lld\n",ans); } else { scanf("%d%d%d",&l,&r,&date); UpDate(l,r,date,1); } } } return 0;}
POJ 3468 A Simple Problem with Integers
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。