nyoj ACM:士兵杀敌（五）(数组的巧妙运用)

士兵杀敌（五）
时间限制：2000 ms | 内存限制：65535 KB
难度：5
描述
南将军麾下有百万精兵，现已知共有M个士兵，编号为0~M，每次有任务的时候，总会有一批编号连在一起人请战（编号相近的人经常在一块，相互之间比较熟悉），最终他们获得的军功，也将会平分到每个人身上，这样，有时候，计算他们中的哪一个人到底有多少军功就是一个比较困难的事情。

在这样的情况下，南将军却经常会在许多次战役之后询问军师小工第i号士兵到第j号士兵所有人的总军功数。

请你帮助军师小工回答南将军的提问。

输入
只有一组测试数据
第一行是三个整数N,C,Q(1<=N,C,Q<=1000000)，其中N表示士兵的总数。
随后的C行，每行有三个整数Mi,Ni,Ai(0<=Mi<=Ni<=N,0<=Ai<=100)，表示从第Mi号到第Ni号士兵所有人平均增加了Ai的军功。
再之后的Q行，每行有两个正正数m,n，表示南将军询问的是第m号士兵到第n号士兵。
输出
请对每次询问输出m号士兵到第n号士兵的总军功数，由于该数值可能太大，请把结果对10003取余后输出
样例输入
5 3 2
1 3 2
2 4 1
5 5 10
1 5
2 3
样例输出
19
6

做这个题前，把士兵杀敌(一到四）做了，所以开始认为可以用树状数组的 区间查询+区间修改来做，于是下来下面代码：

#include<iostream>
#include<cstdio>

int N,C,Q;
int b[1000000+1];
int bi[1000000+1];
void questy(int* a,int i,int data)
{

    while(i<=N)
    {
        a[i]+=data;
        i+=(i&(-i));
    }
}


int SUM(int* a,int i)
{
    int sum=0;
    while(i>0)
    {
        sum+=a[i];
        i-=(i&(-i));
    }

    return sum;
}


int main()
{
    scanf("%d%d%d",&N,&C,&Q);
    int mi,ni,ai;
    for(int i=1;i<=C;i++)
    {

        scanf("%d%d%d",&mi,&ni,&ai);

        questy(b,mi,ai);
        questy(b,ni+1,-ai);

        questy(bi,mi,mi*ai);
        questy(bi,ni+1,-1*(ni+1)*ai);
    }

    int sum_a , sum_b; 
    for(int i=1;i<=Q;i++)
    {
        scanf("%d%d",&mi,&ni);
        //segma(org[i]) + (x+1)*segma(delta[i]) -segma(delta[i]*i)，1 <= i <= x  
        //因为A为空，所以orj[i] ==0 
        sum_a=0+(ni+1)*SUM(b,ni)-SUM(bi,ni);
        sum_b=0+mi*SUM(b,mi-1)-SUM(bi,mi-1);
        printf("%d\n",sum_a-sum_b);

    }
    return 0;
}

但是超时了，后来看了网上的讨论，才发现这个题是线下查询，也就是想把数据都修改好后，一直查询，不再修改。所以用树状数组这种动态修改的数据结构，就有些过了，导致超时。
下面是网上的思路，非常巧妙的运用数组。

,因为本题的区间插入和查询是分开的,所以可以在插入时只在区间头和尾部加上一个增量记号,最后添加操作结束后一次性修改所有的区间,然后再遍历一遍数组修改成前N个数的和, 最后可以O(1)时间求解,总体时间复杂度为O(N)

#include<stdio.h>  
int array[1000010];  
int main()  
{  
    int n, c, q;  
    scanf("%d%d%d", &n, &c, &q);  
    int i;  
    int s, e, num;  
    //一开始每个士兵的值为0即array[i]都为0
    for(i = 0; i < c; i++)  
    {  
        scanf("%d%d%d", &s, &e, &num);  
        //要在s~e范围内增加增量num，那么先 把 增量num放在s上面，-num放在e+1上面。这样，在下面的操作中，由于array[e+1]会加上array[e]，所以这里就提前把加到的num 减去了。
        array[s] += num;  
        array[e + 1] -= num;  
    }  
    //第一遍把标记下的增量修改到每一个人身上  
    for(i = 0; i <= n; i++)  
    {  

        array[i] += array[i - 1];  
    }  

    //第二遍求出前N个人的和  
    for(i = 0; i <= n; i++)  
    {  
        array[i] = (array[i] + array[i - 1]) % 10003;  
    }  

    for(i = 0; i < q; i++)  
    {  
        scanf("%d %d", &s, &e);  
        printf("%d\n", (array[e] - array[s - 1] + 10003) % 10003);//这里注意两个数都是10003的余数,因此为了防止array[e] 小于 array[s - 1],相减得到负数，所以要加上10003  
    }  
    return 0;  
}