HDU 1556 Color the ball 樹狀陣列 區間更新
阿新 • • 發佈:2019-01-09
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<stdio.h> #include<iostream> #include<string.h> using namespace std; const int MAXN=1e5+5; int c[MAXN]; int n; int lowbit(int x) //計算末尾的0 { return x&(-x); } void add(int i,int val) //數進行更新 { while(i<=n) { c[i]+=val; i+=lowbit(i); } } int sum(int i) //計算單點值 { int s=0; while(i>0) { s+=c[i]; i-=lowbit(i); } return s; } int main() { int a,b; while(scanf("%d",&n),n) { memset(c,0,sizeof(c)); for(int i=0;i<n;i++) { scanf("%d%d",&a,&b); add(a,1); //區間更新 add(b+1,-1); } for(int i=1;i<n;i++) printf("%d ",sum(i)); printf("%d\n",sum(n)); } return 0; }