首页 > 代码库 > hdu 4031(树状数组+辅助数组)

hdu 4031(树状数组+辅助数组)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4031

Attack

Time Limit: 5000/3000 MS (Java/Others)    Memory Limit: 65768/65768 K (Java/Others)
Total Submission(s): 1890    Accepted Submission(s): 554


Problem Description
Today is the 10th Annual of “September 11 attacks”, the Al Qaeda is about to attack American again. However, American is protected by a high wall this time, which can be treating as a segment with length N. Al Qaeda has a super weapon, every second it can attack a continuous range of the wall. American deployed N energy shield. Each one defends one unit length of the wall. However, after the shield defends one attack, it needs t seconds to cool down. If the shield defends an attack at kth second, it can’t defend any attack between (k+1)th second and (k+t-1)th second, inclusive. The shield will defend automatically when it is under attack if it is ready.

During the war, it is very important to understand the situation of both self and the enemy. So the commanders of American want to know how much time some part of the wall is successfully attacked. Successfully attacked means that the attack is not defended by the shield.
 

Input
The beginning of the data is an integer T (T ≤ 20), the number of test case.
The first line of each test case is three integers, N, Q, t, the length of the wall, the number of attacks and queries, and the time each shield needs to cool down.
The next Q lines each describe one attack or one query. It may be one of the following formats
1. Attack si ti
  Al Qaeda attack the wall from si to ti, inclusive. 1 ≤ si ≤ ti ≤ N
2. Query p
  How many times the pth unit have been successfully attacked. 1 ≤ p ≤ N
The kth attack happened at the kth second. Queries don’t take time.
1 ≤ N, Q ≤ 20000
1 ≤ t ≤ 50
 

Output
For the ith case, output one line “Case i: ” at first. Then for each query, output one line containing one integer, the number of time the pth unit was successfully attacked when asked.
 

Sample Input
2 3 7 2 Attack 1 2 Query 2 Attack 2 3 Query 2 Attack 1 3 Query 1 Query 3 9 7 3 Attack 5 5 Attack 4 6 Attack 3 7 Attack 2 8 Attack 1 9 Query 5 Query 3
 

Sample Output
Case 1: 0 1 0 1 Case 2: 3 2
 

Source
The 36th ACM/ICPC Asia Regional Chengdu Site —— Online Contest
 
思路:这题出的好!。树状数组的成段更新+单点查询
           successfully attacked=attack-defend;
          总的attack 由树状数组求和就可以。至于defend数得想办法求出~
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <string>
#include <cstdio>
#include <cmath>
const int N=20200;
using namespace std;

int c[N],n,m,t,T;
int shield[N],pos[N];

struct node
{
  int l,r;
}attack[N];

int lowbit(int x)
{
    return x&(-x);
}

void update(int x,int d)
{
  while(x<=n)
  {
    c[x]+=d;
    x+=lowbit(x);
  }
}

int getsum(int x)
{
  int ans=0;
  while(x>0)
  {
    ans+=c[x];
    x-=lowbit(x);
  }
  return ans;
}

void Init()
{
   scanf("%d%d%d",&n,&m,&t);
   memset(c,0,sizeof(c));
   memset(shield,0,sizeof(shield));
   memset(pos,0,sizeof(pos));
}

int main()
{
        char s[10];
        int test=1;
        scanf("%d",&T);
        while(T--)
        {

         Init();
         int cnt=0;
         printf("Case %d:\n",test++);
         while(m--)
         {
            scanf("%s",s);
            if(s[0]==‘A‘)
            {
             cnt++;
             int si,ti;
             scanf("%d%d",&si,&ti);
             attack[cnt].l=si;
             attack[cnt].r=ti;
             update(si,1);
             update(ti+1,-1);
            }
            else
            {
             int a;
             scanf("%d",&a);
             for(int i=pos[a];i<=cnt;)
             {
               if(a>=attack[i].l&&a<=attack[i].r)
               {
                 pos[a]=i+t;
                 shield[a]++;
                 i=i+t;
               }
               else i++;
             }
             printf("%d\n",getsum(a)-shield[a]);
            }
         }

        }
        return 0;
}


hdu 4031(树状数组+辅助数组)