Codevs1082 线段树练习 3

题目描述 Description

给你N个数，有两种操作：

1：给区间[a,b]的所有数增加X

2：询问区间[a,b]的数的和。

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，

再接下来一个正整数Q，每行表示操作的个数，

如果第一个数是1，后接3个正整数，

表示在区间[a,b]内每个数增加X,如果是2，

表示操作2询问区间[a,b]的和是多少。

pascal选手请不要使用readln读入

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 2 3

样例输出 Sample Output

9

数据范围及提示 Data Size & Hint

数据范围

1<=n<=200000

1<=q<=200000

 

树状数组40分（剩下超时）：

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 using namespace std;

 int n,m;
 int f[];

 int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-f;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 int lowbit(int x)
 {
     return x&(-x);
 }

 void update(int x,int num)
 {
     while(x<=n)
     {
         f[x]+=num;
         x+=lowbit(x);
     }
 }

 int sum(int x)
 {
     int sum=;
     while(x>)
     {
         sum+=f[x];
         x-=lowbit(x);
     }
     return sum;
 }

 int main()
 {
     n=read();
     for(int i=;i<=n;i++)
         update(i,read());
     m=read();
     for(int i=;i<=m;i++)
     {
         int t=read(),x,y,z;
         if(t==)
         {
             x=read();y=read();z=read();
             for(int j=x;j<=y;j++)
                 update(j,z);
         }
         if(t==)
         {
             x=read();y=read();
             printf("%d\n",sum(y)-sum(x-));
         }
     }
     return ;
 }

线段树：

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 using namespace std;

 struct node
 {
     int left,right,flag;
     long long sum;
 }tree[];

 int n,q;
 int a[];

 int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-f;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 void build(int node,int left,int right)
 {
     tree[node].left=left;tree[node].right=right;
     if(left==right)
     {
         tree[node].sum=a[left];
         return;
     }
     int mid=(left+right)>>;
     build(node<<,left,mid);
     build(node<<|,mid+,right);
     tree[node].sum=tree[node<<].sum+tree[node<<|].sum;
 }

 void pushdown(int node)
 {
     int x=tree[node].right-tree[node].left+;
     tree[node<<].flag+=tree[node].flag;
     tree[node<<|].flag+=tree[node].flag;
     tree[node<<].sum+=(x-(x>>))*tree[node].flag;
     tree[node<<|].sum+=(x>>)*tree[node].flag;
     tree[node].flag=;
 }

 void update(int node,int left,int right,int x)
 {
     int mid=(tree[node].left+tree[node].right)>>;
     tree[node].sum+=(right-left+)*x;
     if(tree[node].left==left&&tree[node].right==right)
     {
         tree[node].flag+=x;
         return;
     }
     if(tree[node].left==tree[node].right) return;
     if(tree[node].flag>) pushdown(node);
     if(right<=mid) update(node<<,left,right,x);
     else if(left>mid) update(node<<|,left,right,x);
     else
     {
         update(node<<,left,mid,x);
         update(node<<|,mid+,right,x);
     }
     tree[node].sum=tree[node<<].sum+tree[node<<|].sum;
 }

 long long query(int node,int left,int right)
 {
     int mid=(tree[node].left+tree[node].right)>>;
     if(tree[node].left==left&&tree[node].right==right)
         return tree[node].sum;
     if(tree[node].flag>) pushdown(node);
     if(right<=mid)
         return     query(node<<,left,right);
     else if(left>mid)
         return query(node<<|,left,right);
     else
         return query(node<<,left,mid)+query(node<<|,mid+,right);
 }

 int main()
 {
     n=read();
     for(int i=;i<=n;i++)
         a[i]=read();
     build(,,n);
     q=read();
     for(int i=;i<=q;i++)
     {
         int t=read(),a=read(),b=read(),x;
         if(t==)
         {
             x=read();
             update(,a,b,x);
         }
         if(t==)
         {
             cout<<query(,a,b)<<endl;
         }
     }
     return ;
 }