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

最新推荐文章于 2021-10-22 21:45:13 发布

原创最新推荐文章于 2021-10-22 21:45:13 发布 · 546 阅读

0 ·

CC 4.0 BY-SA版权

codeforces 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种基于坐标位置的出租车乘客分配算法。该算法通过遍历乘客与出租车的位置信息，根据距离最近的原则，将乘客分配给合适的出租车。同时考虑了在距离相等的情况下，选择坐标位置较小的出租车作为分配目标。最终，算法计算出了每辆出租车需要搭载的乘客总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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