【 NO.1 找出数组中的所有 K 近邻下标】
解题思路
遍历 nums 找到所有 key 的下标,然后将其左右 k 范围内的下标加到答案数组中即可。
代码展示
【 NO.2 统计可以提取的工件】
解题思路
二维前缀和的典型应用场景。
代码展示
【 NO.3 K 次操作后最大化顶端元素】
解题思路
枚举每个位置,判断这个位置经过 k 次操作后能否变成栈顶。详见注释。
代码展示
【 NO.4 得到要求路径的最小带权子图】
解题思路
求出 src1 到其他所有节点的最短路,然后枚举 src2 到 dest 的所有路径,每找到一条路径 X,枚举 src1 到这个路径上每一个点的最短路 Y, 此时 X 和 Y 的长度和就可以作为备选答案。
代码展示