HDU1166 （树状数组或线段树）
模板题。
树状数组法：
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=50005;
const double eps=1e-8;
int tree[maxn];
inline int lowbit(int x)
{
return x&-x;
}
void add(int x,int value) //输入数据时，需要调用此函数加到tree[]里面
{
for(int i=x;i<=maxn;i+=lowbit(i))
{
tree[i]+=value;
}
}
int get(int x) //获取的是tree[1]+……+tree[x]的值
{
int sum=0;
for(int i=x;i;i-=lowbit(i))
{
sum+=tree[i];
}
return sum;
}
int main()
{
int t,cas=0;
cin>>t;
while(t--)
{
int n,d;
cin>>n;
memset(tree,0,sizeof(tree));
for(int i=1;i<=n;i++)
{
cin>>d;
add(i,d);
}
char q[10];
cout<<"Case "<<++cas<<":"<<endl;
while(cin>>q&&q[0]!='E')
{
int a,b;
cin>>a>>b;
if(q[0]=='Q')
{
cout<<get(b)-get(a-1)<<endl;
}
else if(q[0]=='A')
{
add(a,b);
}
else if(q[0]=='S')
{
add(a,-b);
}
}
}
return 0;
}
线段树：
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=50005;
const double eps=1e-8;
const double PI = acos(-1.0);
#define lowbit(x) (x&(-x))
//#define ls l,m,rt<<1
//#define rs m+1,r,rt<<1|1
int Sum[maxn<<2],Add[maxn<<2];//Sum求和，Add为懒惰标记
int A[maxn],n;//存原数组数据下标[1,n]
//PushUp函数更新节点信息 ，这里是求和
void PushUp(int rt)
{
Sum[rt]=Sum[rt<<1]+Sum[rt<<1|1];
}
//Build函数建树
void Build(int l,int r,int rt) //l,r表示当前节点区间，rt表示当前节点编号
{
if(l==r) //若到达叶节点
{
Sum[rt]=A[l];//储存数组值
return;
}
int m=(l+r)>>1;
//左右递归
Build(l,m,rt<<1);
Build(m+1,r,rt<<1|1);
//更新信息
PushUp(rt);
}
//点修改 假设A[L]+=C:
void Update(int L,int C,int l,int r,int rt) //l,r表示当前节点区间，rt表示当前节点编号
{
if(l==r) //到叶节点，修改
{
Sum[rt]+=C;
return;
}
int m=(l+r)>>1;
//根据条件判断往左子树调用还是往右
if(L <= m) Update(L,C,l,m,rt<<1);
else Update(L,C,m+1,r,rt<<1|1);
PushUp(rt);//子节点更新了，所以本节点也需要更新信息
}
//下推标记
void PushDown(int rt,int ln,int rn)
{
//ln,rn为左子树，右子树的数字数量。
if(Add[rt])
{
//下推标记
Add[rt<<1]+=Add[rt];
Add[rt<<1|1]+=Add[rt];
//修改子节点的Sum使之与对应的Add相对应
Sum[rt<<1]+=Add[rt]*ln;
Sum[rt<<1|1]+=Add[rt]*rn;
//清除本节点标记
Add[rt]=0;
}
}
//区间修改 假设A[L,R]+=C
void Update(int L,int R,int C,int l,int r,int rt) //L,R表示操作区间，l,r表示当前节点区间，rt表示当前节点编号
{
if(L <= l && r <= R) //如果本区间完全在操作区间[L,R]以内
{
Sum[rt]+=C*(r-l+1);//更新数字和，向上保持正确
Add[rt]+=C;//增加Add标记，表示本区间的Sum正确，子区间的Sum仍需要根据Add的值来调整
return ;
}
int m=(l+r)>>1;
PushDown(rt,m-l+1,r-m);//下推标记
//这里判断左右子树跟[L,R]有无交集，有交集才递归
if(L <= m) Update(L,R,C,l,m,rt<<1);
if(R > m) Update(L,R,C,m+1,r,rt<<1|1);
PushUp(rt);//更新本节点信息
}
//区间查询
int Query(int L,int R,int l,int r,int rt) //L,R表示操作区间，l,r表示当前节点区间，rt表示当前节点编号
{
if(L <= l && r <= R)
{
//在区间内，直接返回
return Sum[rt];
}
int m=(l+r)>>1;
//下推标记，否则Sum可能不正确
PushDown(rt,m-l+1,r-m);
//累计答案
int ANS=0;
if(L <= m) ANS+=Query(L,R,l,m,rt<<1);
if(R > m) ANS+=Query(L,R,m+1,r,rt<<1|1);
return ANS;
}
/*
//建树
Build(1,n,1);
//点修改
Update(L,C,1,n,1);
//区间修改
Update(L,R,C,1,n,1);
//区间查询
int ANS=Query(L,R,1,n,1);
*/
int main()
{
std::ios::sync_with_stdio(false);
std::cin.tie(0);
std::cout.tie(0);
int t,k=0;
cin>>t;
while(t--)
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>A[i];
}
Build(1,n,1);
string s;
int a,b;
cout<<"Case "<<++k<<":"<<endl;
while(cin>>s&&s[0]!='E')
{
cin>>a>>b;
if(s[0]=='Q')
{
cout<<Query(a,b,1,n,1)<<endl;
}
else if(s[0]=='A')
{
Update(a,b,1,n,1);
}
else if(s[0]=='S')
{
Update(a,-b,1,n,1);
}
}
}
return 0;
}
