校园自行车分配~leetcode春季高校编程大赛~python

墨蓝 2022-02-18 18:07 224阅读 0赞

题目描述：

> 在由 2D 网格表示的校园里有 `n` 位工人（`worker`）和 `m` 辆自行车（`bike`），`n <= m`。所有工人和自行车的位置都用网格上的 2D 坐标表示。
> 
> 我们需要为每位工人分配一辆自行车。在所有可用的自行车和工人中，我们选取彼此之间曼哈顿距离最短的工人自行车对  (worker, bike) ，并将其中的自行车分配給工人。如果有多个 (worker, bike) 对之间的曼哈顿距离相同，那么我们选择工人索引最小的那对。类似地，如果有多种不同的分配方法，则选择自行车索引最小的一对。不断重复这一过程，直到所有工人都分配到自行车为止。
> 
> 给定两点 `p1` 和 `p2` 之间的曼哈顿距离为 `Manhattan(p1, p2) = |p1.x - p2.x| + |p1.y - p2.y|`。
> 
> 返回长度为 `n` 的向量 `ans`，其中 `a[i]` 是第 `i` 位工人分配到的自行车的索引（从 0 开始）。

![1261_example_1_v2.png][]示例1：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNjEyMjEx_size_16_color_FFFFFF_t_70][]![1261_example_1_v2.png][]

>     输入：workers = [[0,0],[2,1]], bikes = [[1,2],[3,3]]
>     输出：[1,0]
>     解释：
>     工人 1 分配到自行车 0，因为他们最接近且不存在冲突，工人 0 分配到自行车 1 。所以输出是 [1,0]。

示例2：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNjEyMjEx_size_16_color_FFFFFF_t_70 1][]

>     输入：workers = [[0,0],[1,1],[2,0]], bikes = [[1,0],[2,2],[2,1]]
>     输出：[0,2,1]
>     解释：
>     工人 0 首先分配到自行车 0 。工人 1 和工人 2 与自行车 2 距离相同，因此工人 1 分配到自行车 2，工人 2 将分配到自行车 1 。因此输出为 [0,2,1]。

提示：

> 1.  `0 <= workers[i][j], bikes[i][j] < 1000`
> 2.  所有工人和自行车的位置都不相同。
> 3.  `1 <= workers.length <= bikes.length <= 1000`

\----------------------------------------------------------答题区----------------------------------------------------------

思路：这道题有优先性要求，即优先级:距离>工人编号>自行车编号，还有就是返回的结果是包含工人和自行车信息的，所以构造数据结构时，需要用把三者都存储的结构。for循环计算出距离，并把(距离,工人编号，自行车编号),按照元组方式存储，按照上述的优先级进行排序，最后要判断该工人是否已经预约到自行车和自行车是否已经被分配。

class Solution:
        def assignBikes(self, workers: List[List[int]], bikes: List[List[int]]) -> List[int]:
            import math
            res=[]
            for i in range(len(workers)):
                for j in range(len(bikes)):
                    dis=abs(workers[i][0]-bikes[j][0])+abs(workers[i][1]-bikes[j][1])
                    res.append((dis,i,j))
            res.sort(key=lambda x:(x[0],x[1],x[2]))
            
            work=[-1]*len(workers)
            work_set=set()
            bike_set=set()
            for i in res:
                if (i[1] not in work_set) and (i[2] not in bike_set):
                    work[i[1]]=i[2]
                    work_set.add(i[1])
                    bike_set.add(i[2])
            return work

后话：做这道题时，起初并没有排序，而是暴力搜索比较，提交一直超时。简单总结一下，当遇到数据的顺序有一定含义时，应善于使用字典。Python提供了丰富的数据类型，后续会总结出各种数据类型的适用情形，让Python代码也飞起来。

[1261_example_1_v2.png]: /images/20220218/eac16420f0f040f4916022d6aa34155b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNjEyMjEx_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20190417212555472.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNjEyMjEx,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNjEyMjEx_size_16_color_FFFFFF_t_70 1]: /images/20220218/630843dbad0d46c2bb924eb23951382d.png