Leetcode解题之路(golang版)：26. 删除排序数组中的重复项(Remove Duplicates from Sorted Array)

最新推荐文章于 2023-09-04 02:42:31 发布

原创最新推荐文章于 2023-09-04 02:42:31 发布 · 455 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

25 篇文章

订阅专栏

这篇博客主要探讨如何在原地修改已排序的数组，删除重复元素，确保每个元素仅出现一次。文章介绍了四种解法，虽然前三种解法使用了额外空间，但重点讨论了如何在O(1)额外空间限制下解决问题。对于每种解法，都分析了其空间和时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

26. 删除排序数组中的重复项

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:
给定数组 nums = [1,1,2], 

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。 

你不需要考虑数组中超出新长度后面的元素。

示例 2:
给定 nums = [0,0,1,1,1,2,2,3,3,4],

函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。

你不需要考虑数组中超出新长度后面的元素。

以下解法中一和三都使用了额外空间！

解法一：

空间复杂度： $O(n)$

时间复杂度： $O(n)$

func removeDuplicates(nums []int) int {
    //使用map，天然去重
    myMap := make(map[int]struct{})
    for _ , v := range nums {
        myMap[v] = struct{}{}
    }
    ind := 0
    for i, _ := range myMap {
        nums[ind] = i
        ind++
    }
    nums = nums[:ind]
    //由于range map不是有序输出，需要再次排序
    sort.Ints(nums)
    return len(myMap)
}

解法二：

空间复杂度： $O(1)$

时间复杂度： $O(n)$

func removeDuplicates(nums []int) int {
    if n := len(nums); n <= 1 {
        return n
    }
    locIndex := len(nums) - 1
    nowIndex := len(nums) - 2
    //从后往前，把所有没重复的数字放在后面
    for i := nowIndex; i >= 0; i-- {
        for i >= 0 && nums[i] == nums[locIndex] {
            i--
        }
        if i >= 0 {
            locIndex--
            nums[locIndex] = nums[i]
        }
    }
    //由于未重复数字在后面，需要移动到前面
    ans := len(nums) - locIndex
    for i := 0; locIndex < len(nums); i++ {
        nums[i] = nums[locIndex]
        locIndex++
    }
    return ans
}

解法三：

空间复杂度： $O(n)$

时间复杂度： $O(n)$

func removeDuplicates(nums []int) int {
    //这里也是使用map，但是key记录的是数字，value记录的是当前下标
	p := make(map[int]int)
	for _, v := range nums {
		if _, ok := p[v]; !ok {
			p[v] = len(p)
			nums[p[v]] = v
		}
	}
	return len(p)
}

解法四：
空间复杂度： $O(1)$

时间复杂度： $O(n)$

func removeDuplicates(nums []int) int {
    if len(nums) <= 1 {
        return len(nums)
    }
    //这里的思想是把为重复数字放在前面，和解法一类似，但是省去了移动的过程
    rlen := 1
    for i := 1; i < len(nums); i++ {
        if nums[i] != nums[rlen - 1] {
            nums[rlen] = nums[i]
            rlen++
        }
    }
    fmt.Println(rlen)
    return rlen
}