初入leetcode-两数和

来看一下题目,给定一个整数数组,另外给定一个数,如果在数组中两个数相加等于这个数,获得这两个数的索引,两个数不能重复,而且假设每个数最多只有一组数存在于数组中。

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

假设已经给出 nums 和target, 写出其对应的算法。

第一种方法,暴力破解

public int[] twoSum(int[] nums, int target) {
    for (int i = 0; i < nums.length; i++) {
        for (int j = i + 1; j < nums.length; j++) {
            if (nums[j] == target - nums[i]) {
                return new int[] { i, j };
            }
        }
    }
    throw new IllegalArgumentException("No two sum solution");
}

这个算法的过程,遍历整个数组,内循环为当前值向后遍历,故其时间复杂度为o(n^2), 效率比较低。

第二种方法,利用哈希表,key-value对 ,这里需要注意的是,nums[i]作为key,i 作为value,因为我们需要得到索引,但map 中没有根据value查询对应key的方法。

public int[] twoSum(int[] nums, int target) {
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        map.put(nums[i], i);
    }
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];
        if (map.containsKey(complement) && map.get(complement) != i) {
            return new int[] { i, map.get(complement) };
        }
    }
    throw new IllegalArgumentException("No two sum solution");
}

第一个for,初始化map, 第二个for,遍历nums,相减,从map需要找的另一半,如果有,并且不是它自身,那么说明找到了。这个算法的时间复杂度为o(n),相较于上一个算法效率提升不少。但是还是美中不足,还能改进。

第三种方法, 第二种方法的基础上进改进。

 public static int[] twoSum(int[] nums, int target) {
    	Map<Integer, Integer> map = new HashMap<>();
    	for(int i=0; i<nums.length; i++){
    		int temp = target - nums[i];
    		if(map.containsKey(temp)){
    			return new int[]{map.get(temp), i};
    		}
    		map.put(nums[i], i);
    	}
    	throw new IllegalArgumentException("No two sum solution");
    }

在把nums[i]放到map之前,判断map是否包含它的另一半, 如果包含就找到了,如果不包含,就继续添加,添加继续判断,直到把所有的nums添加到map中,这个虽然时间复杂度也是o(n),但是运行时间却比方法二少了很多。

总结
    利用map快速判断是否存在差值,在初始化map的同时判断效率也不错。



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值