LeetCode 1182. 与目标颜色间的最短距离（二分查找/DP）

文章目录

- 1. 题目
- 2. 解题
- - 2.1 二分查找
  - 2.2 DP

1. 题目

给你一个数组 colors，里面有 1、2、 3 三种颜色。

我们需要在 colors 上进行一些查询操作 queries，其中每个待查项都由两个整数 i 和 c 组成。

现在请你帮忙设计一个算法，查找从索引 i 到具有目标颜色 c 的元素之间的最短距离。

如果不存在解决方案，请返回 -1。

示例 1：
输入：colors = [1,1,2,1,3,2,2,3,3], queries = [[1,3],[2,2],[6,1]]
输出：[3,0,3]
解释： 
距离索引 1 最近的颜色 3 位于索引 4（距离为 3）。
距离索引 2 最近的颜色 2 就是它自己（距离为 0）。
距离索引 6 最近的颜色 1 位于索引 3（距离为 3）。示例 2：
输入：colors = [1,2], queries = [[0,3]]
输出：[-1]
解释：colors 中没有颜色 3。提示：
1 <= colors.length <= 5*10^4
1 <= colors[i] <= 3
1 <= queries.length <= 5*10^4
queries[i].length == 2
0 <= queries[i][0] < colors.length
1 <= queries[i][1] <= 3

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shortest-distance-to-target-color
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

找到下标 i 左右最近的 c 颜色花

2.1 二分查找

class Solution {
public:vector<int> shortestDistanceColor(vector<int>& colors, vector<vector<int>>& queries) {unordered_map<int,set<int>> m;for(int i = 0; i < colors.size(); ++i){m[colors[i]].insert(i);}vector<int> ans(queries.size(), -1);int l, r, dis, i = 0;for(auto& q : queries){if(colors[q[0]] == q[1]){ans[i++] = 0;continue;}auto next = m[q[1]].lower_bound(q[0]);//二分查找r = next==m[q[1]].end() ? INT_MAX : *next-q[0];l = INT_MAX;if(next != m[q[1]].begin()){l = q[0] - *(--next);}dis = min(l, r);ans[i++] = dis==INT_MAX ? -1 : dis;}return ans;}
};

972 ms 114.4 MB

2.2 DP

记住左右两侧的最近花的位置，正反扫描2次

class Solution {
public:vector<int> shortestDistanceColor(vector<int>& colors, vector<vector<int>>& queries) {int i, j, n = colors.size(), dis, idx, c;vector<int> pos(4,-1);vector<vector<int>> left(n, vector<int>(4, INT_MAX));vector<vector<int>> right(n, vector<int>(4, INT_MAX));for(i = 0; i < n; ++i){pos[colors[i]] = i;left[i][1] = pos[1]==-1 ? INT_MAX : i-pos[1];left[i][2] = pos[2]==-1 ? INT_MAX : i-pos[2];left[i][3] = pos[3]==-1 ? INT_MAX : i-pos[3];}pos[1] = pos[2] = pos[3] = -1;for(i = n-1; i >= 0; --i){pos[colors[i]] = i;right[i][1] = pos[1]==-1 ? INT_MAX : pos[1]-i;right[i][2] = pos[2]==-1 ? INT_MAX : pos[2]-i;right[i][3] = pos[3]==-1 ? INT_MAX : pos[3]-i;}vector<int> ans(queries.size(), -1);for(i = 0; i < queries.size(); ++i){idx = queries[i][0];c = queries[i][1];dis = min(left[idx][c], right[idx][c]);ans[i] = dis==INT_MAX ? -1 : dis;}return ans;}
};