码迷,mamicode.com
首页 > 编程语言 > 详细

树状数组

时间:2018-09-30 21:53:14      阅读:232      评论:0      收藏:0      [点我收藏+]

标签:复制   for   def   线段树   char   include   std   class   函数   

 

主要要注意lowbit函数

然后add时向右跑,check时向左跑(此处的add只能单点修改)

#include<iostream>//从洛谷3374直接复制来的
#include<cstdio>
using namespace std;
#define lowbit(x) (x&(-(x)))
void read(int &x);
int n,m;
int a[500005];
void add(int s,int t)//给s+t 
{
    int ss=s;
    for(;ss<=n;ss+=lowbit(ss)) a[ss]+=t;
}
int check(int s,int t)//左闭右闭
{
    int res=0;
    int ret=0;
    int ss=s-1;
    int tt=t;
    for(;tt;tt-=lowbit(tt)) ret+=a[tt];
    for(;ss;ss-=lowbit(ss)) res+=a[ss];
    return ret-res;
}
int main()
{
    Read(n),Read(m);
    int w;
    for(int i=1;i<=n;i++) Read(w),add(i,w);
    for(int i=1;i<=m;i++)
    {
        int e,s,t;
        RRead(e),Read(s),Read(t);
        if(e==1) add(s,t);
        if(e==2) printf("%d\n",check(s,t));
    }
    return 0;
}
void Read(int &x)
{
    x=0;
    char c=getchar();
    int i=1;
    while(c<0 or c>9)
    {
        if(c==-) i=-1;
        c=getchar();
    }
    while(c>=0 and c<=9)
    {
        x=x*10+c-0;
        c=getchar();
    }
    x*=i;
}

若想实现区间修改,则要结合差分(不过这就只能单点查询了)

 1 #include<iostream>
 2 #include<cstdio>
 3 using namespace std;
 4 #define lowbit(x) ((x)&(-(x)))
 5 void read(int &x);
 6 int n,m;
 7 int a[500005];
 8 int c[500005];
 9 void add(int s,int t)//给s+t 
10 {
11     int ss=s;
12     for(;ss<=n;ss+=lowbit(ss)) a[ss]+=t;
13 }
14 int check(int s,int t)//左闭右闭
15 {
16     int res=0;
17     int ret=0;
18     int ss=s-1;
19     int tt=t;
20     for(;tt;tt-=lowbit(tt)) ret+=a[tt];
21     for(;ss;ss-=lowbit(ss)) res+=a[ss];
22     return ret-res;
23 }
24 int main()
25 {
26     read(n),read(m);
27     c[0]=0;
28     for(int i=1;i<=n;i++) read(c[i]);
29     for(int i=n;i>=1;i--) c[i]=c[i]-c[i-1];
30     //for(int i=1;i<=n;i++) printf("*%d",a[i]);
31     for(int i=1;i<=n;i++) add(i,c[i]);
32     //for(int i=1;i<=n;i++) printf("*%d",check(1,i));
33     for(int i=1;i<=m;i++)
34     {
35         int e,r;
36         read(e);
37         if(e==1) 
38         {
39             int x,y,z;
40             read(x),read(y),read(z);
41             add(x,z);
42             add(y+1,(z*(-1)));
43         }
44         if(e==2) read(r),printf("%d\n",check(1,r));
45     }
46     return 0;
47 }
48 void read(int &x)
49 {
50     x=0;
51     char c=getchar();
52     int i=1;
53     while(c<0 or c>9)
54     {
55         if(c==-) i=-1;
56         c=getchar();
57     }
58     while(c>=0 and c<=9)
59     {
60         x=x*10+c-0;
61         c=getchar();
62     }
63     x*=i;
64 }

所以最强大的还是线段树了

树状数组

标签:复制   for   def   线段树   char   include   std   class   函数   

原文地址:https://www.cnblogs.com/shzyk/p/9733383.html

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