HDU-1166-敌兵布阵（线段树模板题）

博主个人博客

题目链接

题意：

给n个兵营及每个兵营的初始人数，然后选择4种操作中的一个，一个经典的线段树模板题

题解：

直接套线段树模板就可以了

代码：

#include
#include
using namespace std;
int n,p,a,b,m,x,y,ans;
struct node
{
    int l,r,w,f;
}tree[400001];
inline void build(int k,int ll,int rr) {//建树
    tree[k].l=ll,tree[k].r=rr;
    if(tree[k].l==tree[k].r){
        scanf("%d",&tree[k].w);
        return;
    }
    int m=(ll+rr)/2;
    build(k*2,ll,m);
    build(k*2+1,m+1,rr);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;
}
inline void down(int k) {//标记下传
    tree[k*2].f+=tree[k].f;
    tree[k*2+1].f+=tree[k].f;
    tree[k*2].w+=tree[k].f*(tree[k*2].r-tree[k*2].l+1);
    tree[k*2+1].w+=tree[k].f*(tree[k*2+1].r-tree[k*2+1].l+1);
    tree[k].f=0;
}
inline void ask_point(int k){//单点查询
    if(tree[k].l==tree[k].r){
        ans=tree[k].w;
        return ;
    }
    if(tree[k].f) down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(x<=m) ask_point(k*2);
    else ask_point(k*2+1);
}
inline void change_point(int k) {//单点修改
    if(tree[k].l==tree[k].r){
        tree[k].w+=y;
        return;
    }
    if(tree[k].f) down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(x<=m) change_point(k*2);
    else change_point(k*2+1);
    tree[k].w=tree[k*2].w+tree[k*2+1].w; 
}
inline void ask_interval(int k) {//区间查询
    if(tree[k].l>=a&&tree[k].r<=b) {
        ans+=tree[k].w;
        return;
    }
    if(tree[k].f) down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(a<=m) ask_interval(k*2);
    if(b>m) ask_interval(k*2+1);
}
inline void change_interval(int k) {//区间修改
    if(tree[k].l>=a&&tree[k].r<=b){
        tree[k].w+=(tree[k].r-tree[k].l+1)*y;
        tree[k].f+=y;
        return;
    }
    if(tree[k].f) down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(a<=m) change_interval(k*2);
    if(b>m) change_interval(k*2+1);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;
}
char s[15];
int main(){
	int t;
	scanf("%d",&t);
	int tt=1;
	while(tt<=t){
		scanf("%d",&n);//n个节点 
	    build(1,1,n);//建树 
	    printf("Case %d:\n",tt);
		while(scanf("%s",s)){
	        ans=0;
	        if(s[0]=='E')break;
	        else if(s[0]=='A'){
	        	scanf("%d%d",&x,&y);
	            change_point(1);//单点修改 
	        }
	        else if(s[0]=='S'){
	            scanf("%d%d",&x,&y);
	            y=-y;
	            change_point(1);//单点修改 
	        }
	        else if(s[0]=='Q'){
	            scanf("%d%d",&a,&b);//区间查询 
	            ask_interval(1);
	            printf("%d\n",ans);
	        }
	    }
	    tt++;
	}

}