学做甜品的网站,建立自己的影视网站,微信公众号的h5网站开发6,宁波网站推广营销公司文章目录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阿明一起加油、一起学习进步