2766. 重新放置石块

重新放置石块

• 难度: medium
• 原始链接: https://leetcode.cn/problems/relocate-marbles
• 标签: 哈希

解法一: 哈希

go

func relocateMarbles(nums []int, moveFrom []int, moveTo []int) []int {
  positions := make(map[int]bool)
  for _, num := range nums {
    positions[num] = true
  }
  n := len(moveFrom)
  for i := 0; i < n; i++ {
    positions[moveFrom[i]] = false
    positions[moveTo[i]] = true
  }
  ans := []int{}
  for position, exists := range positions {
    if exists {
      ans = append(ans, position)
    }
  }
  sort.Ints(ans)
  return ans
}