CCF201912-2垃圾站选址

/*将垃圾坐标存储在一个map中，即map的键是垃圾点的坐标；map的值设为1，便于后面的评分统计。
这样可以快速的判断该垃圾点的上下左右是否有垃圾和该点是否有垃圾，即是否合适做回收站。
在适合做回收站的情况下判断在其左上角、右上角、左下角、右下角位置的垃圾数目给予该收回站评分。
最后统计每个评分回收站的个数即可。*/
#include <bits/stdc++.h>
using namespace std;
const int N = 5;
int ans[N] = {0};
const int M = 1000;
pair<int,int> p[M];

int main(){
    int n;
    cin>>n;
    map<pair<int,int>,int> m;
    for(int i = 0; i < n; i++){
        int x,y;
        cin>>x>>y;
        p[i] = make_pair(x,y);
       // m.insert(make_pair(p[i],1));
        m[p[i]]=1;
    }
    for(int i = 0; i < n; i++){
        int x = p[i].first;
        int y = p[i].second;
        if(m.count(make_pair(x,y-1)) && m.count(make_pair(x,y+1)) && m.count(make_pair(x+1,y)) && m.count(make_pair(x-1,y)))
		   {
            ans[m.count(make_pair(x-1,y-1)) + m.count(make_pair(x-1,y+1))+ m.count(make_pair(x+1,y-1)) + m.count(make_pair(x+1,y+1))]++;
           }
    }
    for(int i = 0; i < 5; i++){
        cout<<ans[i]<<endl;
    }
    return 0;
}