Lyft Level 5 Challenge 2018 - Final Round (Open Div. 2) B. Taxi drivers and Lyft

题意：根据就近原则为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;
}