CCF 201912-2 回收站选址（100分）

题目描述

代码

#include <iostream>
#include <set>
#include <map>
#include <vector>
#include <unordered_map>
using namespace std;
int n;
map<int,set<int>> xy;//存放垃圾的位置
int pos[1010][2];//存放输入，一维存放x，二维存放y
int score[5];
int main()
{
    cin>>n;
    for(int i=0;i<n;i++){
        int x,y;
        cin>>x>>y;
        xy[x].insert(y);
        pos[i][0]=x;
        pos[i][1]=y;
    }

    //遍历所有点
    for(int i=0;i<n;i++){
        int x=pos[i][0],y=pos[i][1];//取出当前遍历的点的坐标
        int cnt=0;//记录回收站的评分
        int flag=xy[x].count(y+1)&&xy[x].count(y-1)&&xy[x+1].count(y)&&xy[x-1].count(y);//如果符合称为垃圾站的基本条件
        if(flag){
        //计算回收站的评分
            if(xy[x-1].count(y-1))
                cnt++;
            if(xy[x-1].count(y+1))
                cnt++;
            if(xy[x+1].count(y-1))
                cnt++;
            if(xy[x+1].count(y+1))
                cnt++;
            score[cnt]++;
        }
    }
    for(int i=0;i<5;i++){
        cout<<score[i]<<endl;
    }
    return 0;
}