NYOJ 123 士兵杀敌（四）

http://acm.nyist.net/JudgeOnline/problem.php?pid=123

这题本来交上去就超时了, 然后自己就囧了.... 不知道该用什么更高级的结构了...

然后去百度的题解....

看到的是和我一样的思路, 但怎么就过了呢.. 仔细看了下, 别人的更新都是从当前点更新到0位置.

然而我是从当前点, 更新到数组的结尾, 结果就不行了.

然后自己改了下代码就过了......感觉是数据出的问题，偏向了某一边把....

用线段树是不是会更好呢?

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
const int maxn = 1000005;
int ans[maxn], n;
int lowbit(int x)
{
    return x & (-x);
}

int getSum(int x)
{
    int i, sum=0;
    for(i = x; i<= n; i+=lowbit(i))
    sum += ans[i];
    return sum;
}

int mod(int x, int num)
{
    int i;
    for(i = x; i > 0; i-=lowbit(i))
    ans[i] += num;
}

int main()
{
    int m, a, b, c;
    memset(ans,0,sizeof(ans));
    scanf("%d%d",&n,&m);
    char s[20];
    while(m--)
    {
        scanf("%s",s);
        if(strcmp(s,"ADD")==0)
        {
            scanf("%d%d%d",&a,&b,&c);
            mod(a-1,-c);
            mod(b,c);
        }
        else
        {
            scanf("%d",&a);
            printf("%d\n",getSum(a));
        }
    }
    return 0;
}