题意:根据就近原则为taxi分配乘客(且乘客与两辆taxi距离相同时,选择坐标位置小的taxi。最后计算每辆taxi需要搭载的乘客总数。
分析:按坐标位置从小到大遍历即可求出:每位乘客左边 (pos(taxi)< pos(people)) 距离最近的taxi的编号。
同样地按坐标位置从大到小遍历即可求出:每位乘客右边距离最近的taxi的编号。
代码
#include<bits/stdc++.h>
using namespace std;
int ans[200005];
struct node
{
int pos,dir;
int left,right;
};
node d[200005];
int main()
{
memset(ans,0,sizeof(ans));
int n,m;
scanf("%d%d",&n,&m);
for(int i=0;i<n+m;i++)
scanf("%d",&d[i].pos);
for(int i=0;i<n+m;i++)
scanf("%d",&d[i].dir);
int near_dir=-1;
for(int i=0;i<n+m;i++)
{
if(d[i].dir) near_dir=i;
else d[i].left=near_dir;
}
near_dir=-1;
for(int i=n+m-1;i>=0;i--)
{
if(d[i].dir) near_dir=i;
else d[i].right=near_dir;
}
for(int i=0;i<n+m;i++)
{
if(!d[i].dir)
{
if(d[i].left==-1) ans[d[i].right]++;
else if(d[i].right==-1) ans[d[i].left]++;
else if(d[i].left!=-1&&d[i].right!=-1)
{
int k1=d[i].pos-d[d[i].left].pos;
int k2=d[d[i].right].pos-d[i].pos;
if(k1<=k2)
ans[d[i].left]++;
else
ans[d[i].right]++;
}
}
}
for(int i=0;i<n+m;i++)
{
if(d[i].dir) printf("%d ",ans[i]);
}
return 0;
}