例题5-12 UVA221 城市正视图 （离散化）（综合应用）

#include<iostream>
#include<cstdio>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
int n;
struct building
{
    int id;
    int x;
    int y;
    int width;
    int depth;
    int height;
}B[105];

bool compare(building b1,building b2)
{
    return((b1.x<b2.x)||((b1.x==b2.x)&&(b1.y<b2.y)));
}

int cover(int midx,int bidd)
{
    if((midx>=B[bidd].x)&&(midx<=B[bidd].x+B[bidd].width))
    {
        return 1;
    }
    else
    {
        return 0;
    }
}
int judge(int bid,set<int>POS)
{
     //遍历bx中的各个坐标区间，看是否能够看到编号为bid的建筑物
     //看到的条件是：在坐标区间中任意取一个坐标，然后需要满足下列的条件
     //：建筑物的左右坐标区间内包含所选定的坐标
     //：该建筑物的南边不能有任何建筑物也包含这个坐标并且比带判断的建筑物高或者相等
     vector<int>NEWPOS;
     for(set<int>::iterator it=POS.begin();it!=POS.end();it++)
     {
         NEWPOS.push_back(*it);
     }
     int flag=0;
     for(int j=0;j<int(NEWPOS.size()-1);j++)
     {
         int mid=(NEWPOS[j]+NEWPOS[j+1])/2;
         if(cover(mid,bid))
         {
             int flag1=1;
             //如果南方有任何一个建筑物阻挡，在该区间内返回不可看见
             for(int k=0;k<n;k++)
             {
                 if((B[k].y>=B[bid].y)){continue;}
                 if((cover(mid,k))&&(B[k].height>=B[bid].height))
                 {
                     flag1=0;
                     break;
                 }
             }
             if(!flag1){flag=0;}
             else{flag=1;}
         }
         if(flag==1)
         {
             return 1;
         }
     }
     return 0;
}
int main()
{
    int num=1;
    while((cin>>n)&&n)
    {
        set<int>bx;
        for(int i=0;i<n;i++)
        {
            cin>>B[i].x>>B[i].y>>B[i].width>>B[i].depth>>B[i].height;
            B[i].id=i+1;
            //bx中存放所有建筑物的左右坐标x
            bx.insert(B[i].x);
            bx.insert(B[i].x+B[i].width);
        }
        sort(B,B+n,compare);
        if(num==1)
        {
            cout<<"For map #"<<num++<<", the visible buildings are numbered as follows:"<<endl;
        }
        else
        {
            cout<<endl<<"For map #"<<num++<<", the visible buildings are numbered as follows:"<<endl;
        }
        int cnt=0;
        for(int i=0;i<n;i++)
        {
            //对每一个建筑进行判断，如果正视图能够看到，则输出编号
            if(judge(i,bx))
            {
                cnt++;
                if(cnt==1)
                {
                    cout<<B[i].id;
                }
                else
                {
                    cout<<" "<<B[i].id;
                }
            }
        }
        cout<<endl;
    }
    return 0;
}