试题编号: 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;
}