Codeup100000636问题 B: 区间查询

题目描述：

食堂有N个打饭窗口，现在正到了午饭时间，每个窗口都排了很多的学生，而且每个窗口排队的人数在不断的变化。
现在问你第i个窗口到第j个窗口一共有多少人在排队？

输入：

输入的第一行是一个整数T，表示有T组测试数据。
每组输入的第一行是一个正整数N（$N\le 30000$），表示食堂有N个窗口。
接下来一行输入N个正整数，第i个正整数ai表示第i个窗口最开始有ai个人排队。（$1\le a_i\le 50$）
接下来每行有一条命令，命令有四种形式：
(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\le j$，表示询问第i到第j个窗口的总人数；
(4)End 表示结束，这条命令在每组数据最后出现；
每组数据最多有40000条命令。

输出：

对于每组输入，首先输出样例号，占一行。
然后对于每个Query询问，输出一个整数，占一行，表示询问的段中的总人数，这个数保持在int以内。

样例输入：

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

样例输出：

Case 1:
6
33
59

实现代码：

#include<stdio.h>
#include<string.h>

const int maxn = 3e4 + 10;
const int sqrN = 173;
int table[maxn];
int block[174];

void Add(int i, int j) {
    table[i] += j;
    block[(i - 1) / sqrN] += j;
}

void Sub(int i, int j) {
    table[i] -= j;
    block[(i - 1) / sqrN] -= j;
}

void Query(int i, int j) {
    int l = (i - 1) / sqrN;
    int r = (j - 1) / sqrN;
    int sum = 0;
    if(l == r) {
        for(int k = i; k <= j; k++) {
            sum += table[k];
        }
    } else {
        for(int k = i; k <= sqrN * (l + 1); k++) {
            sum += table[k];
        }
        for(int k = l + 1; k < r; k++) {
            sum += block[k];
        }
        for(int k = sqrN * r + 1; k <= j; k++) {
            sum += table[k];
        }
    }
    printf("%d\n", sum);
}

int main() {
    int T, N;
    char cmd[20];
    int i, j;
    while(scanf("%d", &T) != EOF) {
        for(int k = 1; k <= T; k++) {
            memset(table, 0, sizeof(table));
            memset(block, 0, sizeof(block));
            scanf("%d", &N);
            for(int j = 1; j <= N; j++) {
                scanf("%d", &table[j]);
                block[(j - 1) / sqrN] += table[j];
            }
            printf("Case %d:\n", k);
            scanf("%s", cmd);
            while(strcmp(cmd, "End") != 0) {
                if(strcmp(cmd, "Add") == 0) {
                    scanf("%d%d", &i, &j);
                    Add(i, j);
                } else if(strcmp(cmd, "Sub") == 0) {
                    scanf("%d%d", &i, &j);
                    Sub(i, j);
                } else if(strcmp(cmd, "Query") == 0) {
                    scanf("%d%d", &i, &j);
                    Query(i, j);
                }
                scanf("%s", cmd);
            }
        }
    }
    return 0;
}