算法 18.看楼房

#include <cstdio>
#include <stack>
#include <cstring>//需要使用 memset函数 
using namespace std;
struct building
{
	long long int color;
	long long int height;
}house[1000005];//构建一个结构体数组，每个结构体元素包含每栋楼的高度和颜色 
stack <struct building> s;//使用一个栈 
long long int colors[1000005];
long long int p;
int main()
{
	int m;
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
		{
			p=0;
			memset(colors,0,sizeof(colors));
			while(!s.empty())
				{
					s.pop();
				}//初始化参数为0 
			int count=0;
			long int n;
			scanf("%ld",&n);
			for(long int j=1;j<=n;j++)
				{
					scanf("%lld",&house[j].color);
				}
			for(long int j=1;j<=n;j++)
				{
					scanf("%lld",&house[j].height);
				}
			s.push(house[1]);//先把第一栋楼压进去 
			colors[house[1].color]++;
			p++;
			count++;
			printf("%lld ",p); 
			for(long int j=2;j<=n;j++)
			{
				if(count==0)
					{
						s.push(house[j]);
						colors[house[j].color]++;
						count++;
						p++;
					}//如果是空栈，直接把这栋楼的数据压进去 
				else//如果不是空栈，再进行分析 
				{
					while(house[j].height>=s.top().height)//当新的楼比栈顶的楼高的时候，不断让栈顶的楼出栈（因为已经看不到了），直到栈顶的楼没有新的楼高 
						{
							colors[s.top().color]--;//能看到的栈顶这种颜色的楼少一个 
							if(colors[s.top().color]==0)//如果能看到的这种颜色的楼被减没有了，能看到的颜色就要少一种 
								{
									p--;
								}
							s.pop();//让这栋楼出栈 
							count--;//栈的长度减1 
							if(s.empty())//如果栈被减光了，结束这个循环 
								{
									break;
								}	
						}
				if(count==0)
					{
						s.push(house[j]);
						colors[house[j].color]++;
						count++;
						p++;
					}//如果是空栈，直接把这栋楼的数据压进去 
				if(house[j].height<s.top().height)//如果新的楼没有栈顶的楼高或者这个时候是空栈，把新的楼放进去，可以看到的这种颜色的楼数量加1 
					{
				 		if(colors[house[j].color]==0)
				 			{
				 				p++;
				 			}
				 		colors[house[j].color]++;
				 		s.push(house[j]);
				 		count++;
					}
				}
				if(j<n)//如果不是最后一栋楼 
				{
					printf("%lld ",p);
				}
				else//如果是最后一栋楼 
				{
					printf("%lld\n",p);
				}	
			}
		}
	return 0;
}