HDU 1166 敌兵布阵

最新推荐文章于 2020-07-27 15:54:26 发布

原创最新推荐文章于 2020-07-27 15:54:26 发布 · 597 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构与算法专栏收录该内容

338 篇文章

订阅专栏

本文介绍了一个使用树状数组解决HDU1166敌兵布阵问题的C语言实现案例。通过具体代码示例展示了树状数组的基本操作，包括求区间和、增加指定位置的值及减少指定位置的值。需要注意的是，在此问题中，使用C++的输入输出会导致程序运行超时。

题目链接：HDU 1166 敌兵布阵

树状数组。

下午刚学会树状数组，感觉不简单，慢慢学吧。

这个题必须得用c语言的输入输出，用c++的会超时。

#include <iostream>
#include <cstring>
#include <stdio.h>

using namespace std;

const int MAX_N = 50000 + 1000;

int T,n,temp;
int num[MAX_N];
char c[15];

int lowbit(int i)
{
    return i & (-i);
}
int sum(int x)
{
    int sum = 0;
    while (x > 0)
    {
        sum += num[x];
        x -= lowbit(x);
    }
    return sum;
}
void add(int i,int val)
{
    while (i <= n)
    {
        num[i] += val;
        i += lowbit(i);
    }
}
void sub(int i,int val)
{
    while (i <= n)
    {
        num[i] -= val;
        i += lowbit(i);
    }
}
int main()
{
    scanf("%d",&T);
    int cnt = 0;
    while(T--)
    {
        printf("Case %d:\n",++cnt);
        memset(num,0,sizeof(num));
        scanf("%d",&n);
        for(int i = 1;i <= n;i++)
        {
            scanf("%d",&temp);
            add(i,temp);
        }
        while(scanf("%s",c))
        {
            if(c[0] == 'E')
                break;
            int a,b;
            scanf("%d %d",&a,&b);
            if(c[0] == 'Q')
                printf("%d\n", sum(b) - sum(a - 1));
            else if(c[0] == 'A')
                add(a,b);
            else if(c[0] == 'S')
                sub(a,b);
        }
    }
    return 0;
}