HDU 1556 Color the Ball 线段树 题解

本题使用线段树自然能够,由于区间的问题。

这里比較难想的就是:

1 最后更新须要查询全部叶子节点的值,故此须要使用O(nlgn)时间效率更新全部点。

2 截取区间不能有半点差错。否则答案错误。

这两点卡了我下。看来我的线段树还是不够熟,须要多多练习。

线段树是二分法的高级应用,可是却不是简单应用,要锻炼好并应用好线段树思维还是比二分法难非常多的。


static const int SIZE = 100005;
static const int TREESIZE = SIZE<<2;
int arr[SIZE];
int segTree[TREESIZE];

inline int lChild(int r) { return r<<1; }
inline int rChild(int r) { return r<<1|1; }

void pushDown(int rt)
{
	segTree[lChild(rt)] += segTree[rt];
	segTree[rChild(rt)] += segTree[rt];
}

void build(int l, int r, int rt)
{
	int mid = l + ((r-l)>>1);  

	segTree[rt] = 0;  
	if(l == r) return; 

	build(l, mid, lChild(rt));  
	build(mid+1, r, rChild(rt)); 
}

void update(const int L, const int R, int l, int r, int rt)
{
	if (L <= l && r <= R)
	{
		segTree[rt]++;
		return ;
	}

	int m = l + ((r-l)>>1);

	//注意怎样准确截取区间
	if (R <= m) update(L, R, l, m, lChild(rt));
	else if (L > m) update(L, R, m+1, r, rChild(rt));
	else
	{
		update(L, m, l, m, lChild(rt));
		update(m+1, R, m+1, r, rChild(rt));//须要截取准确区间
	}
}

void query(int l, int r, int rt)
{
	if (l == r)
	{
		arr[l] = segTree[rt];
		return;
	}
	pushDown(rt);

	int m = l + ((r-l)>>1);
	query(l, m, lChild(rt));
	query(m+1, r, rChild(rt));
}

int main()
{
	int n, a, b;
	while (scanf("%d", &n) && n != 0)
	{
		fill(arr, arr+n+1, 0);
		build(1, n, 1);

		for(int i = 0; i < n; i++)  
		{  
			scanf("%d %d",&a,&b);  
			update(a, b, 1, n, 1);
		} 
		query(1, n, 1);

		for(int i = 1; i < n; i++)  
		{  
			printf("%d ", arr[i]);  
		}  
		printf("%d\n",arr[n]);  
	}
	return 0;
}




你可能感兴趣的:(HDU 1556 Color the Ball 线段树 题解)