A - 敌兵布阵 (HDU - 1166)
- 题目大意
中文题,一共有4种命令,然后来根据命令来进行操作。
- 解题思路
典型的线段树的问题,进行点的更新,然后进行区间的查询(查询的时候查的是这个区间上的点的和)。
- 代码
#include<cstdio> #include<iostream> #include<cstring> #include<algorithm> #include<cstring> using namespace std; const int maxn=50001; int ans; struct node { int l,r,sum; }t[4*maxn]; void build(int l,int r,int k) { int mid; if(l==r) { t[k].l=l; t[k].r=r; t[k].sum=0; return; } mid=(l+r)/2; t[k].l=l; t[k].r=r; t[k].sum=0; build(l,mid,2*k); build(mid+1,r,2*k+1); } void add(int n,int d,int k) { int mid; if(t[k].l==t[k].r&&t[k].l==d) { t[k].sum+=n; return; } mid=(t[k].r+t[k].l)/2; if(d<=mid) add(n,d,2*k); else add(n,d,2*k+1); t[k].sum=t[2*k].sum+t[2*k+1].sum; } void findd(int l,int r,int k) { int mid; if(t[k].l==l&&t[k].r==r) { ans+=t[k].sum; return; } mid=(t[k].r+t[k].l)/2; if(r<=mid) findd(l,r,2*k); else if(l>mid) findd(l,r,2*k+1); else { findd(l,mid,2*k); findd(mid+1,r,2*k+1); } } char str[20]; int main() { int n,x,y; scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&x); build(1,x,1); for(int j=1;j<=x;j++) { scanf("%d",&y); add(y,j,1); } printf("Case %d:\n",i); while(scanf("%s",str),strcmp(str,"End")) { int a,b; scanf("%d%d",&a,&b); if(strcmp(str,"Add")==0) add(b,a,1); else if(strcmp(str,"Sub")==0) add(-b,a,1); else { ans=0; findd(a,b,1); printf("%d\n",ans); } } } return 0; }