[PKU] 3468 A Simple Problem with Integers -- 线段树

一开始超时，然后优化了好久。。

开始给的n个值就算是初始值吧。我在建树的时候就把每个结点的和都算好了，存在info里，其实先建树再一个一个插入也是可以的。后面的操作中要增加的值就是附加值，我把它存在add里 。

关键的优化就是，如果一个附加值正好加在线段树里的一个结点所代表的那个区间上，就不用再往子区间加下去了，加在那个结点就行。然后在查询的时候如果碰到了有附加值的结点又要往子区间去查询，就顺便把那个点的附加值传给子区间 。

这样就少了很多操作，不然在极端一些的情况下，加一个数要操作到线段树中的每一个结点，操作数就是O（n）了。

Source Code

Problem: 3468		User: yueashuxia
Memory: 7188K		Time: 1563MS
Language: C++		Result: Accepted

• Source Code

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define size 100001
struct xds
{
    int a, b ;
    int lson, rson ;
    long long info, add ;//info存储区间和，add存储该区间每个增加的数 
}tree[size*3] ;
int tot ;
long long num[size+1] ;
void make_xds(int s, int t)
{
    int now = ++tot ;
    tree[now].a = s ;
    tree[now].b = t ;
    tree[now].add = 0 ;
    if(t - s > 0)
    {
    	int mid = (s + t) >> 1 ;
     	tree[now].lson = tot + 1 ;
      	make_xds(s, mid) ;
       	tree[now].rson = tot + 1 ;
       	make_xds(mid + 1, t) ;
        tree[now].info = tree[tree[now].lson].info + tree[tree[now].rson].info ; //建树时把和存好 
    } 
    else { //子结点 
    	tree[now].lson = tree[now].rson = 0 ;
     	tree[now].info = num[s] ;
    }
}
void update(int p, int s, int t, int info)
{
	if(tree[p].a == s && tree[p].b == t)//正好加在这个节点就直接返回 
	{
 		tree[p].add += info ;
 		return  ;
 	}
 	tree[p].info += info * (t - s + 1) ; //s和t一定在p结点的区间内 
    if(tree[p].a == tree[p].b) return ;
    int mid = (tree[p].a + tree[p].b) >> 1 ; //往下传递 
    if(s > mid) update(tree[p].rson, s, t, info) ;
    else if(t <= mid) update(tree[p].lson, s, t, info) ;
    else update(tree[p].lson, s, mid, info), update(tree[p].rson, mid + 1, t, info) ;
}
long long query(int p, int s, int t)
{
    if(tree[p].a == s && tree[p].b == t) {
         return tree[p].info + tree[p].add * (tree[p].b - tree[p].a + 1);
    }
    if(tree[p].add) // 把add值往左右子树传递 
    {
    	tree[p].info += (tree[p].b - tree[p].a + 1) * tree[p].add ;
    	tree[tree[p].lson].add += tree[p].add ;
    	tree[tree[p].rson].add += tree[p].add ;
    	tree[p].add = 0 ;
    }
    int mid = (tree[p].a + tree[p].b) >> 1 ;
    if(t <= mid) return query(tree[p].lson, s, t) ;
    else if(s > mid) return query(tree[p].rson, s, t) ;
    else return query(tree[p].lson, s, mid) + query(tree[p].rson, mid + 1, t) ;
}
int main (){
    int Q, i, j, N ;
    long long k ;
    char ord[2] ;
    scanf("%d%d", &N, &Q) ;
    tot = 0 ;    
    for (i = 1; i <= N; i ++)
    {
    	scanf("%lld", &num[i]);//先存好 
    }
    make_xds(1, N) ; //再建树 
    while(Q --)
   	{
    	scanf("%s%d%d", ord, &i, &j);
     	switch (ord[0])
      	{ 
       		case 'Q': printf("%lld/n", query(1, i, j)); break ;
         	case 'C': scanf("%lld", &k);update(1, i, j, k) ; 
      	}
    }
    //system ("pause") ;
    return 0 ;
}