1057校园自行车分配

it2024-02-01  63

题目描述: 在由 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: 输入:workers = [[0,0],[2,1]], bikes = [[1,2],[3,3]] 输出:[1,0] 解释: 工人 1 分配到自行车 0,因为他们最接近且不存在冲突,工人 0 分配到自行车 1 。所以输出是 [1,0]。

示例 2: 输入: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]。

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

方法1: 主要思路: (1)先使用map来统计同一种长度下的各种可能的工人和自行车的组合,组合的过程中,按照先工人,后自行车的顺序压入统计中; (2)从统计的数据中,按序找出各个员工和自行车的组成,进行统计;

class Solution { public: vector<int> assignBikes(vector<vector<int>>& workers, vector<vector<int>>& bikes) { map<int,vector<pair<int,int>>> lables; vector<bool> w_visited(workers.size(),false); vector<bool> b_visited(bikes.size(),false); //按照长度进行统计各种工人和自行车的组合,工人和自行车是按顺序压入统计中的 for(int i=0;i<workers.size();++i){ for(int j=0;j<bikes.size();++j){ int dis=abs(workers[i][0]-bikes[j][0])+abs(workers[i][1]-bikes[j][1]); lables[dis].push_back({i,j}); } } //存储结果 vector<int> res(workers.size()); int count=0; //统计各个可能的组成 for(auto& it:lables){ for(auto&v:it.second){ //保证之前没有使用过 if(!w_visited[v.first]&&!b_visited[v.second]){ res[v.first]=v.second; ++count; w_visited[v.first]=true; b_visited[v.second]=true; } //跳出循环的条件 if(count==res.size()){ break; } } } return res; } };
最新回复(0)