540. 有序数组中的单一元素 golang

540. 有序数组中的单一元素 golang

func singleNonDuplicate(nums []int) int {
    if len(nums) <= 1 {
        return nums[0]
    }
	i := 0
	for i < len(nums) - 1 {
		if nums[i] != nums[i+1] {
			return nums[i]
		} else {
			i+=2
		}
	}
	return nums[i]
}