LeetCode 1057 校园自行车分配

题目

在由 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 开始）。

数据范围:

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

思路

一个思路是枚举处所有曼哈顿距离和对应索引(distance, i, j) 然后按照从小到大顺序排列, 遍历排好序的数组, dis,i, j记录索引i, j是否被使用, 如果没有被使用, 可以有ret[i] = j, 按理说不会超时的, 我是用数据结果是vector<vecotr<int> > dis , 其中每个vector<int> 都是类似的三元组, 但是调用的时候一直超时, 后来自己定义了 node 类和比较函数通过了.

代码如下, 算法复杂度$mn\log (mn)$

class node{
   
   
    public:
    int dis;
    int i;
    int j;
    node(int a, int b, int c):dis(a), i(b), j(c){
   
   }
    bool operator<(node &b){
   
   
        if (dis > b.dis) return false;
        if (dis < b.dis ) return true;
        if (i > b.i) return false;
        if (i < b.i) return true;
        return j < b.j;