HDU ACM 4031 Attack (树状数组--单点查询+区间更新)

 

http://acm.hdu.edu.cn/showproblem.php?pid=4031

 用了树状数组的区间更新 单点查找（一般为单点更新 区间查找）

例如 区间（2,4）加1

则Updata(2,1)   Updata(4+1,-1)

实现了更新（2,4）的值而不改变其他值

求Sum时即可得到某一点的值

 

 

 

 

#include <iostream>
using namespace std;
const int MAX = 20000 + 100;
int C[MAX];
int N;
struct Node
{
    int l;
    int r;
};
int LowBit(int x)
{
    return x & (-x);
}
void Update(int num,int key)
{
    while(num <= N)
    {
        C[num] += key;
        num += LowBit(num);
    }
}

int CalSum(int num)
{
    int sum = 0;
    while(num > 0)
    {
        sum += C[num];
        num -= LowBit(num);
    }
    return sum;
}
int main()
{
    int T;
    cin>>T;
    int Case = 0;
    while(T--)
    {
        int Q;//操作数量
        int CD;//护盾冷却时间
        int time = 0;//当前经过时间
        int used_time[MAX] = {0};//记录护盾什么时候进入冷却
        int porected[MAX] = {0};//记录保护次数
        Node attack[MAX] = {0};//定义结构体,记录攻击的位置
        memset(C,0,sizeof(C));
        Case++;
        cout<<"Case "<<Case<<":"<<endl;
        scanf("%d%d%d",&N,&Q,&CD);
        int i;
        for(i=0;i<Q;i++)
        {
            char str[20];
            scanf("%s",str);
            if(str[0] == 'A')
            {
                scanf("%d%d",&attack[time].l,&attack[time].r);
                Update(attack[time].l,1);
                Update(attack[time].r+1,-1);
                time++;
            }
            else
            {
                int num;
                scanf("%d",&num);
                int i;
                for(i = used_time[num] ; i< time ;i++)//每一次询问计算被保护的次数
                {
                    if(attack[i].l <=num    && num <= attack[i].r )
                    {
                        porected[num]++;
                        used_time[num] = i + CD;
                        i = i + CD - 1;
                    }
                }
                printf("%d\n",CalSum(num)-porected[num]);//成功攻击次数= 总攻击数 - 被保护次数
            }
        }
    }
    return 0;
}