码迷,mamicode.com
首页 > 其他好文 > 详细

树状树组区间修改,单点修改模板

时间:2019-08-21 11:29:27      阅读:44      评论:0      收藏:0      [点我收藏+]

标签:include   else   scan   cst   def   ==   while   scanf   sum   

单点修改

#include <iostream>
#include <cstring>
#include <stack>
#include <cstdio>
#include <cmath>
#include <queue>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
using namespace std;
#define ll long long
const int  N=500005;
int a,c[N],n,m=0;

int sum(int x)
{
    int ans=0;
    for(;x;x-=x&-x)
    {
        ans=ans+c[x];
    }
    return ans;
}
void add(int x,int y)
{
    for(;x<=N-1;x+=x&-x)
        c[x]+=y; 
}
int main()
{
    ll ans=0;
    int x,y,z;
    cin>>n>>m;
      for(int i=1;i<=n;i++)
      {
          scanf("%d",&a);
          add(i,a);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        if(x==1)
            add(y,z);
        else
            printf("%d\n",sum(z)-sum(y-1));
    }
    return 0;
}

区间修改

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) x&-x
using namespace std;
const int N=5e5+10;
  ll tree[N];
int n,m;
void add(int x,ll num)
{
    while(x<=n)
    {
        tree[x]+=num;
        x+=lowbit(x);
    }
}
ll query(int x)
{
    ll ans=0;
    while(x)
    {
        ans+=tree[x];
        x-=lowbit(x);
    }
    return ans;
}


int main()
{
    scanf("%d%d",&n,&m);
     long long last=0,now;
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&now);
        add(i,now-last);
        last=now;
     }
     int flag;
     while(m--)
     {
         int x,y;
         ll k;
         scanf("%d",&flag);
         if(flag==1)
         {
             scanf("%d%d%lld",&x,&y,&k);
             add(x,k);
             add(y+1,-k);
         }
         else
         {
             scanf("%d",&x);
             printf("%lld\n",query(x));
         }
         
      } 
    return 0;
}

 

树状树组区间修改,单点修改模板

标签:include   else   scan   cst   def   ==   while   scanf   sum   

原文地址:https://www.cnblogs.com/hh13579/p/11387398.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!