POJ - 3468 - A Simple Problem with Integers

#include<iostream>
using namespace std;
struct CNode
{
    int L,R;
    CNode *pLeft,*pRight;
    long long nSum;
    long long Inc;
};
CNode Tree[1000000];
int nCount=0;
int Mid(CNode *pRoot)
{
    return (pRoot->L +pRoot->R)/2;
}
void BuildTree(CNode *pRoot,int L,int R)
{
    pRoot->L=L;
    pRoot->R=R;
    pRoot->nSum=0;
    pRoot->Inc=0;
    if(L==R)return ;
    nCount++;
    pRoot->pLeft=Tree+nCount;
    nCount++;
    pRoot->pRight=Tree+nCount;
    BuildTree(pRoot->pLeft,L,(L+R)/2);
    BuildTree(pRoot->pRight,(L+R)/2+1,R);
}
void Insert(CNode *pRoot,int i,int v)
{
    if(pRoot->L==i&&pRoot->R==i)
    {
        pRoot->nSum=v;
        return ;
    }
    pRoot->nSum+=v;
    if(i<=Mid(pRoot))
    Insert(pRoot->pLeft,i,v);
    else 
    Insert(pRoot->pRight,i,v);
}
void Add(CNode *pRoot,int a,int b,long long c)
{
    if(pRoot->L==a&&pRoot->R==b)
    {
        pRoot->Inc+=c;
        return ;
    }
    pRoot->nSum+=c*(b-a+1);
    if(b<=(pRoot->L+pRoot->R)/2)
    Add(pRoot->pLeft,a,b,c);
    else if(a>=(pRoot->L+pRoot->R)/2+1)
    Add(pRoot->pRight,a,b,c);
    else
    {
        Add(pRoot->pLeft,a,(pRoot->L+pRoot->R)/2,c);
        Add(pRoot->pRight,(pRoot->L+pRoot->R)/2+1,b,c);
    }
}
long long QuerynSum(CNode *pRoot,int a,int b)
{
    if(pRoot->L==a&&pRoot->R==b)
    return pRoot->nSum+(pRoot->R-pRoot->L+1)*pRoot->Inc;
     pRoot->nSum+=(pRoot->R-pRoot->L+1)*pRoot->Inc;
     Add(pRoot->pLeft,pRoot->L,Mid(pRoot),pRoot->Inc);
     Add(pRoot->pRight,Mid(pRoot)+1,pRoot->R,pRoot->Inc);
     pRoot->Inc=0;
     if(b<=Mid(pRoot))
     return QuerynSum(pRoot->pLeft,a,b);
     else if(a>=Mid(pRoot)+1)
     return QuerynSum(pRoot->pRight,a,b);
     else
     {
         return QuerynSum(pRoot->pLeft,a,Mid(pRoot))+QuerynSum(pRoot->pRight,Mid(pRoot)+1,b);
     }
 }
 int main()
 {
     int n,q,a,b;
     long long c;
     char cmd[10];
     scanf("%d%d",&n,&q);
     int i,j,k;
     nCount=0;
     BuildTree(Tree,1,n);
     for(i=1;i<=n;i++)
     {
         scanf("%d",&a);
         Insert(Tree,i,a);
     }
     for(i=0;i<q;i++)
     {
         scanf("%s",cmd);
         if(cmd[0]=='C')
         {
             scanf("%d%d%I64d",&a,&b,&c);
             Add(Tree,a,b,c);
         }
         else
         {
            scanf("%d%d",&a,&b);
            printf("%I64d\n",QuerynSum(Tree,a,b));
        }
    }
    return 0;
}                                             


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值