2sum 算法

2024年6月11日 266次阅读

题目描述：

给出一个整数数组和一个目标值，判断数组中是否有两个数之和等于目标值

1、粗暴的方法

每次选中一个数，然后遍历整个数组，判断是否有另一个数使两者之和为target

时间复杂度：O(n^2)

 

2、利用集合可以优化时间复杂度

思路：在遍历数组的过程中，用集合每次保存当前值。假如集合中已经有一个数等于目标值减去当前当前值，则证明在之前的遍历中一定有一个数与当前值之和等于目标值

时间复杂度：O(n)

具体实现：

func twoSum(mums:[Int], _ target:Int)->Bool{
    var set = Set<Int>()
    for num in nuns{
        if set.contains(target-num){
            return true
        }
        
        set.insert(num)
    }
    return false
}

3、题目变形

描述：给定一个整型数组中有且仅有两个数之和等于目标值，求这两个数在数组中的序号

思路：算法思路与2类似，但是为了方便得到序列号，使用字典

时间复杂度：O(n)

具体实现：

func twoSum(nums : [Int], target : Int)->[Int]{
    var dict = [Int : Int]

    for(i,num) in nums.enumerated(){
        if let lastIndex = dict[target-num]{
            return [lastIndex,i]
        }else{
            dict[num]=i
        }
    }
    fatalError("no valid output!")
}

 

点赞