201912-2 回收站选址

试题编号： 201912-2
试题名称： 回收站选址
时间限制： 1.0s
内存限制： 512.0MB

范围太大，所以选择遍历+剪枝

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef pair<int,int> P;
int score[6];
vector<P>v;
int back(int x,int y,int pos){
	bool flag1=false,flag2=false;
	int res=0;
	for(int i=pos;i>=0;i--){
		int nx=v[i].first,ny=v[i].second;
		if(nx<x-1) break;
		if(nx==x&&ny==y-1) flag1=true;
		if(nx==x-1&&ny==y+1) res++;
		if(nx==x-1&&ny==y) flag2=true;
		if(nx==x-1&&ny==y-1) res++;
	}
	if(flag1&&flag2) return res;
	else return -1;
}
int front(int x,int y,int pos){
	bool flag1=false,flag2=false;
	int res=0;
	for(int i=pos;i<v.size();i++){
		int nx=v[i].first,ny=v[i].second;
		if(nx>x+1) break;
		if(nx==x&&ny==y+1) flag1=true;
		if(nx==x+1&&ny==y+1) res++;
		if(nx==x+1&&ny==y) flag2=true;
		if(nx==x+1&&ny==y-1) res++;
	}
	if(flag1&&flag2) return res;
	else return -1;	
}
int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		P t;
		cin>>t.first>>t.second;
		v.push_back(t);
	}
	sort(v.begin(),v.end());
	for(int i=0;i<v.size();i++){
		int x=v[i].first,y=v[i].second;
		int b=back(x,y,i),f=front(x,y,i);
		if(b>=0&&f>=0) score[b+f]++;
	}
	for(int i=0;i<5;i++){
		cout<<score[i]<<endl;
	}
	return 0;
}