标签:return 树状 int cin 查询 add 数组
树状数组
单点修改和区间查询
单点修改:add(x,k)
需要一层一层向上找到父节点并修改
for(int i=x;i<=n;i+=lowbit(i))t[i]+=k;
区间查询:query(x) (快速求出前缀和)
需要一层一层向左上寻找
int res=0;
for(int i=x;i;i-=lowbit(i))res+=t[i];
return res;
code
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define pre(i,a,b) for(int i=(a);i>=(b);i--)
#define close cin.tie(nullptr)->ios::sync_with_stdio(false)
using namespace std;
using ll =long long;
const int N=5e5+10;
int n,m;
int t[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,int c)
{
for(int i=x;i<=n;i+= lowbit(i))t[i]+=c;
}
int query(int x)
{
int res=0;
for(int i=x;i;i-= lowbit(i))res+=t[i];
return res;
}
void solve()
{
cin>>n>>m;
rep(i,1,n){
int x;cin>>x;
add(i,x);
}
while(m--)
{
int a,b,c;
cin>>a>>b>>c;
if(a==1)add(b,c);
else cout<<query(c)-query(b-1)<<endl;
}
}
int main()
{
close;
// int _;cin>>_;
// while(_--)
solve();
return 0;
}
区间修改和单点查询
题目链接
区间的修改 很容易想到用差分数组来实现 因此我们用树状数组来维护差分数组b[]的前缀和
区间修改 add(l,x) add(r+1,-x)
单点查询 query(x)
差分数组的前缀和是原数组 因此我们要快速查询某一个点的值只需要求差分数组的前缀和即可
code
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define pre(i,a,b) for(int i=(a);i>=(b);i--)
#define close cin.tie(nullptr)->ios::sync_with_stdio(false)
using namespace std;
using ll =long long;
const int N=5e5+10;
int n,m;
int t[N];
int a[N],b[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,int c)
{
for(int i=x;i<=n;i+= lowbit(i))t[i]+=c;
}
int query(int x)
{
int res=0;
for(int i=x;i;i-= lowbit(i))res+=t[i];
return res;
}
void solve()
{
cin>>n>>m;
rep(i,1,n)cin>>a[i],b[i]=a[i]-a[i-1],add(i,b[i]);
while(m--)
{
int op,x,y,k;
cin>>op;
if(op==1)cin>>x>>y>>k,add(x,k),add(y+1,-k);
else if(op==2)cin>>x,cout<<query(x)<<endl;
}
}
int main()
{
close;
// int _;cin>>_;
// while(_--)
solve();
return 0;
}
区间修改和区间查询
标签:return,树状,int,cin,查询,add,数组 来源: https://www.cnblogs.com/xinxinlan/p/15863541.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。