leetcode2295_go_替换数组中的元素

题目

给你一个下标从 0 开始的数组 nums ,它包含 n 个 互不相同 的正整数。

请你对这个数组执行 m 个操作,在第 i 个操作中,你需要将数字 operations[i][0] 替换成 operations[i][1] 。

题目保证在第 i 个操作中:

operations[i][0] 在 nums 中存在。

operations[i][1] 在 nums 中不存在。

请你返回执行完所有操作后的数组。

示例 1:输入:nums = [1,2,4,6], operations = [[1,3],[4,7],[6,1]] 输出:[3,2,7,1]

解释:我们对 nums 执行以下操作:

- 将数字 1 替换为 3 。nums 变为 [3,2,4,6] 。

- 将数字 4 替换为 7 。nums 变为 [3,2,7,6] 。

- 将数字 6 替换为 1 。nums 变为 [3,2,7,1] 。

返回最终数组 [3,2,7,1] 。

示例 2:输入:nums = [1,2], operations = [[1,3],[2,1],[3,2]] 输出:[2,1]

解释:我们对 nums 执行以下操作:

- 将数字 1 替换为 3 。nums 变为 [3,2] 。

- 将数字 2 替换为 1 。nums 变为 [3,1] 。

- 将数字 3 替换为 2 。nums 变为 [2,1] 。

返回最终数组 [2,1] 。

提示:n == nums.length

m == operations.length

1 <= n, m <= 105

nums 中所有数字 互不相同 。

operations[i].length == 2

1 <= nums[i], operations[i][0], operations[i][1] <= 106

在执行第 i 个操作时,operations[i][0] 在 nums 中存在。

在执行第 i 个操作时,operations[i][1] 在 nums 中不存在。

解题思路分析

1、哈希;时间复杂度O(n),空间复杂度O(n)

leetcode2295_go_替换数组中的元素

func arrayChange(nums []int, operations [][]int) []int {
   m := make(map[int]int)
   for i := 0; i < len(nums); i++ {
      m[nums[i]] = i
   }
   for i := 0; i < len(operations); i++ {
      a, b := operations[i][0], operations[i][1]
      index := m[a]
      delete(m, a)
      m[b] = index
      nums[index] = b
   }
   return nums
}

2、哈希;时间复杂度O(n),空间复杂度O(n)

func arrayChange(nums []int, operations [][]int) []int {
   m := make(map[int]int)
   for i := len(operations) - 1; i >= 0; i-- {
      a, b := operations[i][0], operations[i][1]
      if v, ok := m[b]; ok {
         b = v // 在后面出现过,使用最后的结果
      }
      m[a] = b
   }
   for i := 0; i < len(nums); i++ {
      if v, ok := m[nums[i]]; ok {
         nums[i] = v
      }
   }
   return nums
}

总结

Medium题目,借助map模拟操作即可

发表评论
留言与评论(共有 0 条评论) “”
   
验证码:

相关文章

推荐文章