Leetcode 2200 - Find All K-Distant Indices in an Array

題目

Problem#

題目給你一個整數陣列 num 和兩個整數 key, k。數列中所有是 key 的數字的前後 k 個數字所包含的區間是 k-distant index
e.g. nums=[1, 2, 3, 4], key=2, k=1 => k-distant = [0, 1, 2],題目要你回傳 k-distant index

  • 1 <= nums.length <= 1000
  • 1 <= nums[i] <= 1000
  • 1 <= k <= nums.length

想法#

先找到所有等於 key 的 index i, 把 [i-k, i+k]加到 k-distant index 中

  • 時間複雜度: $\mathcal{O}(n^2)$
  • 空間複雜度: $\mathcal{O}(n)$

AC Code#

心得#

Weekly Contest 284 第一題