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

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
}

你可能感兴趣的:(leetcode)