线段树重温——刷墙
描述
在一面很长的墙壁上,工人们用不同的油漆去刷墙,然而可能有些地方刷过以后觉得不好看,他们会重新刷一下。有些部分因为重复刷了很多次覆盖了很多层油漆,toshio很好奇那些地方被刷过多少种颜色的油漆。输入若干行输入,每行两个数字B[i],E[i](0<=B[i]<=E[i]<=200000)表示这次刷的墙壁是哪一段(假设每次刷的时候油漆颜色都和之前的不同),以0 0结束
又若干行输入,每行两个数字begin[i],end[i](0<=begin[i]<=end[i]<=200000)表示toshio询问的段,以0 0结束输出对于每个toshio的询问输出(end[i]-begin[i]+1)行,表示对应询问段的每个点被多少种颜色的油漆覆盖过。
样例输入
1 20
5 10
10 20
0 0
4 6
10 11
0 0
样例输出
1
2
2
3
关键点1:
在这里要提醒大家一下,我们要开开的结构体数目得
是线段的最大长度的至少3倍,一般开到4倍。注意一
下这点就行了,不然的话,你连输入就过不了,更别
说得到正确的结果了。
关键点2:
另外,线段树是以段为单位的,这个怎么理解呢?
以
if(Tree[level].left==left && Tree[level].right==right)
Tree[level].icount += 1;
这句话来说,假如left=1, right=5时,
那么,从1到5的这个段上的1,2,3,4,5中的icount值
都是作了Tree[level].icount += 1;这个动作。
#include<iostream> #include<cstdio> using namespace std; struct Node { int left; int right; int mid; int icount; }; Node Tree[2000000]; void BuildTree(int level, int left, int right) { Tree[level].icount=0; Tree[level].left = left; Tree[level].right = right; Tree[level].mid = (left+right)/2; if(left==right) return; else { BuildTree(2*level, left, Tree[level].mid); ///建立左子结点 BuildTree(2*level+1, Tree[level].mid+1, right); ///建立右子结点 } } void Insert(int level, int left, int right) //统计经过每个点的次数 { if(Tree[level].left==left && Tree[level].right==right) //if语句是在区间吻合时起作用的,这时恰好要是结束的时候。 Tree[level].icount += 1; else if(left > Tree[level].mid) //往右边进行搜索(这是被搜索的区间完全落在右边的情况) Insert(2*level+1, left, right); else if(right <= Tree[level].mid) //往左边进行搜索(这是被搜索的区间完全落在左边的情况) Insert(2*level, left, right); else //往两边进行搜索(这是被搜索的区间落在两边的情况,所以要向两边同时进行搜索) { Insert(2*level, left, Tree[level].mid); Insert(2*level+1, Tree[level].mid+1, right); } } int Find(int level, int target) //对不同的区间进行遍历查询 { if(target==Tree[level].left && target==Tree[level].right) return Tree[level].icount; else if(target > Tree[level].mid) return Tree[level].icount+Find(2*level+1, target); else if(target <= Tree[level].mid) return Tree[level].icount+Find(2*level, target); } int main() { int left, right; BuildTree(1, 0, 200000); while(cin>>left>>right && (left+right)) Insert(1, left, right); while(cin>>left>>right && (left+right)) for(int i=left; i<=right; i++) cout<<Find(1, i)<<endl; return 0; }
posted on 2011-10-01 14:45 More study needed. 阅读(397) 评论(0) 编辑 收藏 举报