poj 3468 A Simple Problem with Integers

线段树的变形很的，考线段树我觉得就步仅仅是简单的线段树。如果那样很好写，写出来也肯定超时。现在我就写了个超时的程序，现在先贴一下。由于线段树有很多变形，还有很多优化的技巧。我还真的学一下，好好学一学。现在只是停留在线段树的初级阶段。很多技巧还等着我去学，我要努力了！

超时的代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100002
struct SegTree
{
    int l,r,mid;
    __int64 num;
}tree[N*3];
int s[N];
__int64 sum;
void BuildTree(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].mid=(l+r)/2;
    if(l==r)
    {
        tree[root].num=s[l];
        return;
    }
    BuildTree(2*root,l,tree[root].mid);
    BuildTree(2*root+1,tree[root].mid+1,r);
    tree[root].num=tree[2*root].num+tree[2*root+1].num;
}
void Query(int root, int l,int r)
{
    if(tree[root].l==l&&tree[root].r==r)
    {
        sum+=tree[root].num;
        return ;
    }
    if(l>tree[root].mid)
    {
        Query(2*root+1,l,r);
    }
    else if(r<=tree[root].mid)
    {
        Query(2*root,l,r);
    }
    else
    {
        Query(2*root,l,tree[root].mid);
        Query(2*root+1,tree[root].mid+1,r);
    }
}
void Updata(int root,int a,int b,int c)
{
    tree[root].num+=c;
    if(tree[root].l==tree[root].r)return;
    if(a>tree[root].mid)
    {
        Updata(2*root+1,a,b,c);
    }
    else if(b<=tree[root].mid)
    {
        Updata(2*root,a,b,c);
    }
    else
    {
        Updata(2*root,a,tree[root].mid,c);
        Updata(2*root+1,tree[root].mid+1,b,c);
    }
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&s[i]);
        }
        BuildTree(1,1,n);
        while(m--)
        {
            int a,b;
            char ch[3];
            scanf("%s%d%d",ch,&a,&b);
            if(ch[0]=='Q')
            {
                sum=0;
                Query(1,a,b);
                printf("%I64d\n",sum);
            }
            else
            {
                int c;
                scanf("%d",&c);
                Updata(1,a,b,c);
            }
        }
    }
    return 0;
}

各种优化过后，还是超时，真不知道为什么，但是我觉得这个程序已经比上个程序省时多了，代码如下

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
#define N 100005
struct SegTree
{
    __int64 sum,a;
    int l,r,mid;

}tree[N*4];
int s[N];
void BuildTree(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].mid=(l+r)/2;
    if(l==r)
    {
        tree[root].sum=s[l];
        return ;
    }
    BuildTree(root<<1,l,tree[root].mid);
    BuildTree(root<<1|1,tree[root].mid+1,r);
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
void Updata(int root,int l,int r,int c)
{
    if(tree[root].l==l&&tree[root].r==r)
    {
        tree[root].a+=c;
        tree[root].sum+=c*(r-l+1);
        return ;
    }
    if(tree[root].a)
    {
        tree[root<<1].a+=tree[root].a;
        tree[root<<1|1].a+=tree[root].a;
        tree[root<<1].sum+=tree[root].a*(tree[root<<1].r-tree[root<<1].l+1);
        tree[root<<1|1].sum+=tree[root].a*(tree[root<<1|1].r-tree[root<<1|1].l+1);
        tree[root].a=0;
    }
    if(l>tree[root].mid)
    {
        Updata(root<<1|1,l,r,c);
    }
    else if(r<=tree[root].mid)
    {
        Updata(root<<1,l,r,c);
    }
    else
    {
        Updata(root<<1,l,tree[root].mid,c);
        Updata(root<<1|1,tree[root].mid+1,r,c);
    }
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
__int64 Query(int root,int l,int r)
{
    if(tree[root].l==r&&tree[root].r==r)
    {
        return tree[root].sum;
    }
    if(tree[root].a)
    {
        tree[root<<1].a+=tree[root].a;
        tree[root<<1|1].a+=tree[root].a;
        tree[root<<1].sum+=tree[root].a*(tree[root<<1].r-tree[root<<1].l+1);
        tree[root<<1|1].sum+=tree[root].a*(tree[root<<1|1].r-tree[root<<1|1].l+1);
        tree[root].a=0;
    }
    if(l>tree[root].mid)return Query(root<<1|1,l,r);
    else if(r<=tree[root].mid)return Query(root<<1,l,r);
    else
    {
       return Query(root<<1,l,tree[root].mid)+Query(root<<1|1,tree[root].mid+1,r);
    }

}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&s[i]);
        }
        memset(tree,0,sizeof(tree));
        BuildTree(1,1,n);
        while(m--)
        {
            char ch[3];
            int a,b;
            scanf("%s%d%d",ch,&a,&b);
            if(ch[0]=='Q')
            {
               printf("%I64d\n",Query(1,a,b));
            }
            else
            {
                int c;
                scanf("%d",&c);
                if(c==0)continue;
                Updata(1,a,b,c);
            }
        }
    }
    return 0;
}

昨天写了两个超时的程序。我都无语。今天再写，就过了。1594ms过去了。好暴力啊！

这是一个比较经典的线段树的题目，就是要成段更新区间，但是，不会更新到节点。没次记录更新到的地方，如果下次用到了，可以再次更新下去，这样比较节省时间。但是还是感觉好暴力。不知道更好的算法如何来写了。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int N=100010;
struct SegTree
{
    int l,r,mid;
    __int64 num,sum;
}tree[N*4];
int s[N];
void BuildTree(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].num=0;
    tree[root].mid=(l+r)>>1;
    if(l==r)
    {
        tree[root].sum=s[l];
        return;
    }
    BuildTree(root<<1,l,tree[root].mid);
    BuildTree(root<<1|1,tree[root].mid+1,r);
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
void Updata(int root,int a,int b,int c)         //更新到所要求的区间，不在往下更新，如果下次要用到这个区间，再往下更新
{
    if(a<=tree[root].l&&tree[root].r<=b)        //表明这段区间的和，已经更新了
    {
        tree[root].num+=c;
        tree[root].sum+=(tree[root].r-tree[root].l+1)*c;
        return ;
    }
    if(tree[root].num)           //这是一个处理上的技巧，要学会了
    {                            //更新到的地方，进行标记
        tree[root<<1].sum+=(tree[root<<1].r-tree[root<<1].l+1)*tree[root].num;
        tree[root<<1|1].sum+=(tree[root<<1|1].r-tree[root<<1|1].l+1)*tree[root].num;
        tree[root<<1].num+=tree[root].num;
        tree[root<<1|1].num+=tree[root].num;
        tree[root].num=0;        //更新过的地方就标记为0,标记叶子节点需要在下次访问的时候需要更新
    }
    if(a<=tree[root].mid)        //两个if,这个技巧也要学会了
    {
        Updata(root<<1,a,b,c);
    }
    if(b>tree[root].mid)        
    {
        Updata(root<<1|1,a,b,c);
    }
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
__int64 Query(int root,int a,int b)
{
    __int64 ans;
    if(tree[root].l==a&&tree[root].r==b)
    {
        return tree[root].sum;
    }
    if(tree[root].num)                  //想想，为什么更新的时候执行这步，访问的时候也执行
    {
        tree[root<<1].sum+=(tree[root<<1].r-tree[root<<1].l+1)*tree[root].num;
        tree[root<<1|1].sum+=(tree[root<<1|1].r-tree[root<<1|1].l+1)*tree[root].num;
        tree[root<<1].num+=tree[root].num;
        tree[root<<1|1].num+=tree[root].num;
        tree[root].num=0;
    }
    if(a>tree[root].mid)
    {
       return Query(root<<1|1,a,b);
    }
    else if(b<=tree[root].mid)
    {
       return Query(root<<1,a,b);
    }
    else
    {
        ans=Query(root<<1,a,tree[root].mid);
        return ans+Query(root<<1|1,tree[root].mid+1,b);
    }
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&s[i]);
        }
        BuildTree(1,1,n);
        while(m--)
        {
            char ch[3];
            int a,b,c;
            scanf("%s%d%d",ch,&a,&b);
            if(a>b)
            {
                int temp=a;
                a=b;
                b=temp;
            }
            if(ch[0]=='Q')
            {
                printf("%I64d\n",Query(1,a,b));
            }
            else
            {
                scanf("%d",&c);
                Updata(1,a,b,c);
            }
        }
    }
    return 0;
}