文章目錄

• • 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;} };

888 ms 131.7 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 1182. 与目标颜色间的最短距离（二分查找/DP）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。