树状数组 hdu 1166(敌兵布阵)

    xiaoxiao2026-10-01  9

    敌兵布阵

    Problem Description A国在海岸线沿直线布置了N个工兵营地 Input 第一行一个整数T,表示有T组数据。 每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。 接下来每行有一条命令,命令有4种形式: (1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30) (2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30); (3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数; (4)End 表示结束,这条命令在每组数据最后出现; 每组数据最多有40000条命令   Output 对第i组数据,首先输出“Case i:”和回车, 对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。   Sample Input 1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End   Sample Output Case 1: 6 33 59  题意:对一维数组操作。 题解:利用树状数组 #include<cstdio> #include<algorithm> #include<cstring> using namespace std; char c[6]; int C[50050]; int lowbit(int q){ return q&(-q); } void add(int x,int y,int n){ while (x<=n){ C[x]+=y; x+=lowbit(x); } } int qu(int n){ int s=0; int i; while (n>0){ s+=C[n]; n-=lowbit(n); } return s; } int main(){ int i,x,y; int t,n,a; int cas=1; scanf ("%d",&t); while (t--){ scanf ("%d",&n); memset (C,0,sizeof(C)); for (i=1;i<=n;i++){ scanf ("%d",&a); add(i,a,n); } printf ("Case %d:\n",cas++); while (scanf ("%s",c)&&c[0]!='E'){ scanf ("%d %d",&x,&y); if (c[0]=='A') add(x,y,n); else if (c[0]=='S') add(x,-y,n); else{ printf ("%d\n",qu(y)-qu(x-1)); } } } return 0; }

    转载请注明原文地址: https://ju.6miu.com/read-1312516.html
    最新回复(0)