算法:来一到two sum 练练手

104 阅读1分钟

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。 你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例: 给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]

class Solution {
    public int[] twoSum(int[] nums, int target) {    
    if(nums.length<2){
         return new int[0];
    }
    Map<Integer , Integer > map = new HashMap<Integer,Integer>();
    for(int i=0;i<nums.length;i++){
        map.put(nums[i],i);
    }
    Set<Integer> set =map.keySet();

    //第一种解法,暴力求解  时间复杂度O(n^2)
    // for(int i=0;i<nums.length;i++){
            // for(int j=0;j<nums.length && i!=j;j++){
            //     if(nums[i] == target-nums[j]){
            //         return new int[]{i,j};
            //     }
            // }
    // }

    //将数据缓存起来,然后从缓存中取,时间复杂度O(n)
         for(int i=0;i<nums.length;i++){
             if(set.contains(target-nums[i]) && i!=map.get(target-nums[i])){
                return new int[]{i,map.get(target-nums[i])};
             }
        }
        return new int[0];
    }
}

对比下提交结果:

PS:以上解法也许不是最优的,有思路的小伙伴可以留言讨论哦~