HDU 1556 Color the ball 線段樹入門題

畫船聽雨發表於2014-02-20

典型的線段樹，區間的更新，統計區間內元素的和。

Color the ball

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 7031    Accepted Submission(s): 3668


Problem Description
N個氣球排成一排，從左到右依次編號為1,2,3....N.每次給定2個整數a b(a <= b),lele便為騎上他的“小飛鴿"牌電動車從氣球a開始到氣球b依次給每個氣球塗一次顏色。但是N次以後lele已經忘記了第I個氣球已經塗過幾次顏色了，你能幫他算出每個氣球被塗過幾次顏色嗎？
 

Input
每個測試例項第一行為一個整數N,(N <= 100000).接下來的N行，每行包括2個整數a b(1 <= a <= b <= N)。
當N = 0，輸入結束。
 

Output
每個測試例項輸出一行，包括N個整數，第I個數代表第I個氣球總共被塗色的次數。
 

Sample Input
3
1 1
2 2
3 3
3
1 1
1 2
1 3
0





 



Sample Output




1 1 1
3 2 1






#include <algorithm>
#include <iostream>
#include <stdlib.h>
#include <string.h>
#include <iomanip>
#include <stdio.h>
#include <string>
#include <queue>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#define eps 1e-7
#define M 10001000
#define LL __int64
//#define LL long long
#define INF 0x3f3f3f3f
#define PI 3.1415926535898

const int maxn = 1000100;
using namespace std;

struct node
{
    int a, b;
    int cnum;
}f[maxn*4];
int num[maxn];

void Bulid(int l, int r, int site)
{
    if(l == r)
    {
        f[site].a = l;
        f[site].b = l;
        f[site].cnum = 0;
        return;
    }
    int mid = (l+r)>>1;
    Bulid(l, mid, site<<1);
    Bulid(mid+1, r, site<<1|1);
    f[site].a = l;
    f[site].b = r;
    f[site].cnum = 0;
}

void Search(int l, int r, int site)
{
    if(f[site].a == l && f[site].b == r)
    {
        f[site].cnum++;
        return;
    }
   else
   {
       int mid = (f[site].a+f[site].b)>>1;
       if(r <= mid)
       {
           Search(l, r, site<<1);
       }
       else if(l > mid)
       {
           Search(l, r, site<<1|1);
       }
       else
       {
           Search(l, mid, site<<1);
           Search(mid+1, r, site<<1|1);
       }
   }
}
void Find(int x)
{
    int i;
    for(i = f[x].a; i<=f[x].b; i++)//該區間所有編號都被刷了一次
        num[i]+=f[x].cnum;
    if(f[x].a == f[x].b)
        return;
    Find(x<<1);
    Find(x<<1|1);
}

int main()
{
    int n;
    int a, b;
    while(scanf("%d",&n) && n != 0)
    {
        Bulid(1, n, 1);
        for(int i = 1; i <= n; i++)
        {
            scanf("%d %d",&a, &b);
            Search(a, b, 1);
        }
        memset(num, 0 , sizeof(num));
        Find(1);
        for(int i = 1; i <= n-1; i++)
            printf("%d ",num[i]);
        printf("%d\n",num[n]);
    }
    return 0;
}









相關文章