网站开发和推广的不同,北京seo运营,天河区门户网站,网站关键词优化互点文章目录1. 题目2. 解题2.1 二分查找2.2 DP1. 题目
给你一个数组 colors#xff0c;里面有 1、2、 3 三种颜色。
我们需要在 colors 上进行一些查询操作 queries#xff0c;其中每个待查项都由两个整数 i 和 c 组成。
现在请你帮忙设计一个算法#xff0c;查找从索引 i 到…
文章目录1. 题目2. 解题2.1 二分查找2.2 DP1. 题目
给你一个数组 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:vectorint shortestDistanceColor(vectorint colors, vectorvectorint queries) {unordered_mapint,setint m;for(int i 0; i colors.size(); i){m[colors[i]].insert(i);}vectorint 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 nextm[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] disINT_MAX ? -1 : dis;}return ans;}
};972 ms 114.4 MB
2.2 DP
记住左右两侧的最近花的位置正反扫描2次
class Solution {
public:vectorint shortestDistanceColor(vectorint colors, vectorvectorint queries) {int i, j, n colors.size(), dis, idx, c;vectorint pos(4,-1);vectorvectorint left(n, vectorint(4, INT_MAX));vectorvectorint right(n, vectorint(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;}vectorint 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] disINT_MAX ? -1 : dis;}return ans;}
};888 ms 131.7 MB 我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号Michael阿明一起加油、一起学习进步
