首页 > 其他 > 详细

A Simple Problem with Integers(线段树)

时间:2014-02-11 00:13:26      阅读:487      评论:0      收藏:0      [点我收藏+]

http://poj.org/problem?id=3468

题意:就是区间的查询与更新。。

bubuko.com,布布扣
#include <stdio.h>
#include <string.h>
const int maxn=100010;
const int maxm=6600000;
#define LL long long
struct node
{
    int a,b;
    int l,r;
    LL  sum;
    LL  h;
} A[maxm];
int Num[maxn],cnt;
void build(int root)
{
    int a = A[root].a;
    int b = A[root].b;
    int mid = (a+b)>>1;
    A[root].h = 0;
    if (a==b)
    {
        A[root].sum = Num[a];
        return ;
    }
    int l = ++cnt;
    int r = ++cnt;
    A[l].a = a;
    A[l].b = mid;
    A[r].a = mid+1;
    A[r].b = b;
    build(l);
    build(r);
    A[root].sum = A[l].sum+A[r].sum;
    A[root].l = l;
    A[root].r = r;
}
LL Query(int t,int a1,int b1,LL h)
{
    int a = A[t].a;
    int b = A[t].b;
    int mid = (a+b)>>1;
    if (a==a1&&b==b1)
        return A[t].sum+(b1-a1+1)*h;
    if (b1 <= mid)
        return Query(A[t].l,a1,b1,A[t].h+h);
    if (a1 > mid)
        return Query(A[t].r,a1,b1,A[t].h+h);
    LL suml = Query(A[t].l,a1,mid,A[t].h+h);
    LL sumr = Query(A[t].r,mid+1,b1,A[t].h+h);
    return suml+sumr;
}
void Update(int t,int a1,int b1,int c)
{
    int a = A[t].a;
    int b = A[t].b;
    int l = A[t].l;
    int r = A[t].r;
    int mid = (a+b)>>1;
    A[t].sum += (b1-a1+1)*c;
    if (a==a1&&b==b1)
    {
        A[t].h+=c;
        return ;
    }
    if (b1 <= mid)
        Update(l,a1,b1,c);
    else if (a1 > mid)
        Update(r,a1,b1,c);
    else
    {
        Update(l,a1,mid,c);
        Update(r,mid+1,b1,c);
    }
}
int main()
{
    int N,Q,a,b,c;
    char s[3];
    cnt = 0;
    scanf("%d%d",&N,&Q);
    for (int i = 1; i <= N; i++)
        scanf("%d",&Num[i]);
    getchar();
    int t = ++cnt;
    A[t].a = 1;
    A[t].b = N;
    build(t);
    while(Q--)
    {
        scanf("%s",s);
        if (s[0]==Q)
        {
            scanf("%d%d",&a,&b);
            printf("%lld\n",Query(1,a,b,0));
        }
        else
        {
            scanf("%d%d%d",&a,&b,&c);
            Update(1,a,b,c);
        }
    }
    return 0;
}
View Code

A Simple Problem with Integers(线段树)

原文:http://www.cnblogs.com/lahblogs/p/3543448.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!