HDU 1556 Color the ball（線段樹|樹狀陣列）

Mr_Treeeee發表於2020-04-06
陣列

Color the ball

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


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





 



Author



8600



 



Source



HDU 2006-12 Programming Contest 



 


中文題意不做解釋了。


法一：


用的是線段樹，就是區間更新和單點查詢，就是單點查詢每個都要查詢到，感覺效率有點慢。


#include <stdio.h>
#include <string.h>
#include <iostream>
#include <map>
#include <algorithm>
#include <math.h>
using namespace std;
#define rt x<<1|1
#define lt x<<1
#define  LL long long
const int N = 100000*5;
int num[N];
int Add[N];
void pushdown(int x)
{
    Add[lt]+=Add[x];
    Add[rt]+=Add[x];
    num[lt]+=Add[x];
    num[rt]+=Add[x];
    Add[x]=0;
}
void add(int x,int l,int r,int ll,int rr)
{
    ///if(Add[x]) pushdown(x);
    if(ll<=l&&rr>=r)
    {
        num[x]++;
        Add[x]++;
    }
    else
    {
        int mid=(l+r)>>1;
        if(ll<=mid) add(lt,l,mid,ll,rr);
        if(mid<rr) add(rt,mid+1,r,ll,rr);
        num[x]=num[lt]+num[rt];
    }
}
int query(int x,int l,int r,int i)
{
    if(Add[x]) pushdown(x);
    int ans=0;
    if(l==r&&l==i)
    {
        ans+=num[x];
    }
    else
    {
        int mid=(l+r)>>1;
        if(i<=mid) ans+=query(lt,l,mid,i);
        if(mid<i) ans+=query(rt,mid+1,r,i);
    }
    return ans;
}
int main()
{
    int n;
    while(~scanf("%d",&n)&&n)
    {
        int l,r;
        memset(num,0,sizeof num);
        memset(Add,0,sizeof Add);
        for(int i=1;i<=n;i++)
        {
            scanf("%d %d",&l,&r);
            add(1,1,n,l,r);
        }
        for(int i=1;i<=n;i++)
        {
            if(i!=1) printf(" ");
            printf("%d",query(1,1,n,i));
        }
        printf("\n");
    }
    
}




法二：


樹狀陣列，把問題當作單點更新，區間查詢。


遇到左端點，左端點所在的+1，右端點+1所在的-1。查詢點就直接變成了查詢1-x區間上的值。這個思維就和法一完全不一樣了。


#include <stdio.h>
#include <string.h>
#include <iostream>
#include <map>
#include <algorithm>
#include <math.h>
using namespace std;
#define rt x<<1|1
#define lt x<<1
#define  LL long long
const int N = 100000*5;
int n;
int num[N];
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int c)
{
    while(x<=n)
    {
        num[x]+=c;
        x+=lowbit(x);
    }
}
int query(int x)
{
    int ans=0;
    while(x>=1)
    {
        ans+=num[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
    while(~scanf("%d",&n)&&n)
    {
        memset(num,0,sizeof num);
        int l,r;
        for(int i=1;i<=n;i++)
        {
            scanf("%d %d",&l,&r);
            add(l,1);
            add(r+1,-1);
        }
        for(int i=1;i<=n;i++)
        {
            printf("%d",query(i));
            if(i==n) printf("\n");
            else printf(" ");
        }
    }
}








相關文章