两数之和golang

package main

import (
    "fmt"
    "sort"
)

func twoSum(nums []int, target int) []int {
    var locs []int
    var start = 0
    var end = len(nums)
    var sum = 0
    // 排序
    oldNums := make([]int, end)
    copy(oldNums, nums)
    sort.Ints(nums)
    // fmt.Println(nums)

    for {
        if start >= end-1 {
            break
        }
        sum = nums[start] + nums[end-1]
        if sum == target {
            // locs = append(locs, start)
            // locs = append(locs, end-1)
            break
        } else if sum > target {
            end -= 1
            continue
        } else if sum < target {
            start += 1
            continue
        }

    }
    // 找出排序后两个元素对应原来数组的位置

    for idx, num := range oldNums {
        if num == nums[start] {
            start = idx
            locs = append(locs, idx)
            break
        }
    }
    for idx, num := range oldNums {
        if num == nums[end-1] && idx != start {
            locs = append(locs, idx)
            break
        }
    }

    return locs
}

func main() {
    //nums := []int{3, 2, 4}
    //target := 6

    //nums := []int{2, 7, 11, 15}
    //target := 9

    nums := []int{3, 2, 3}
    target := 6
    fmt.Println(twoSum(nums, target))
}