lowbit
x&(-x),当x为0时结果为0;x为奇数时,结果为1;x为偶数时,结果为x中2的最大次方的因子。
int lowbit(int x)
{
return x&(-x);
}
功能1:求前缀和
功能2:支持单点修改
模板
#include<bits/stdc++.h>
using namespace std;
const int N=500010;
int n,m;
int a[N],c[N];
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int k)
{
while(x<=n)
{
c[x]+=k;
x+=lowbit(x);
}
}
int sum(int x)
{
int A=0;
while(x>0)
{
A+=c[x];
x-=lowbit(x);
}
return A;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
add(i,a[i]); //建树
}
while(m--)
{
int t,b,c;
scanf("%d%d%d",&t,&b,&c);
if(t==1)
add(b,c);
else
printf("%d\n",sum(c)-sum(b-1);
}
}
区间修改 单点查询:
差分
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=500010;
ll n,m;
ll a[N],difa[N];
ll lowbit(ll x)
{
return x&(-x);
}
void adddif(ll x,ll k)
{
while(x<=n)
{
difa[x]+=k;
x+=lowbit(x);
}
}
ll sumdif(ll x)
{
ll cnt=0;
while(x)
{
cnt+=difa[x];
x-=lowbit(x);
}
return cnt;
}
int main()
{
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
adddif(i,a[i]-a[i-1]);
}
while(m--)
{
ll v;
scanf("%lld",&v);
if(v==1)
{
ll x,y,k;
scanf("%lld%lld%lld",&x,&y,&k);
adddif(x,k);
adddif(y+1,-k);
}
else
{
ll x;
scanf("%lld",&x);
printf("%lld\n",sumdif(x));
}
}
}