1. 程式人生 > >敵兵布陣 HDU1166

敵兵布陣 HDU1166

uil \n cin 兩個 return 敵兵布陣 端點 using sca

基礎線段樹

#include<bits/stdc++.h>
using namespace std;

int ans;
struct node
{
       int l,r,w;//l,r分別表示區間左右端點,w表示區間和
}tree[4*50000+1];//字典樹要開四倍空間

void build(int l,int r,int k)
{
    tree[k].l=l;tree[k].r=r;
    if(l==r)//葉子節點
    {
        scanf("%d",&tree[k].w);
        return ;
    }
    int m=(l+r)/2
; build(l,m,k*2);//左孩子 build(m+1,r,k*2+1);//右孩子 tree[k].w=tree[k*2].w+tree[k*2+1].w;//狀態合並,此結點的w=兩個孩子的w之和 } int ask(int k,int pos) { if(tree[k].l==tree[k].r) //當前結點的左右端點相等,是葉子節點,是最終答案 { return tree[k].w; } int m=(tree[k].l+tree[k].r)/2; if(pos<=m) ask(k*2,pos);//目標位置比中點靠左,就遞歸左孩子
else ask(k*2+1,pos);//反之,遞歸右孩子 } void add(int k,int pos,int v) { if(tree[k].l==tree[k].r)//找到目標位置 { tree[k].w+=v; return; } int m=(tree[k].l+tree[k].r)/2; if(pos<=m) add(k*2,pos,v); else add(k*2+1,pos,v); tree[k].w=tree[k*2].w+tree[k*2+1].w;//所有包含結點k的結點狀態更新
} void sum(int k,int left,int right) { if(tree[k].l>=left && tree[k].r<=right) { ans+=tree[k].w; return; } int m=(tree[k].l+tree[k].r)/2; if(left<=m) sum(k*2,left,right); if(right>m) sum(k*2+1,left,right); } int main() { int cas;cin>>cas; int flag=0; while(cas--) { printf("Case %d:\n",++flag); int n;scanf("%d",&n); build(1,n,1); char s[2000]; int x,y; while(scanf("%s",s)==1) { ans=0; if(s[0]==E)break; scanf("%d%d",&x,&y); if(s[0]==A)add(1,x,y); if(s[0]==S)add(1,x,-y); if(s[0]==Q){sum(1,x,y);printf("%d\n",ans);} } } return 0; }

敵兵布陣 HDU1166