LeetCode 1---Tow Sum

从今天开始刷LeetCode,今天刷的第一题是:
这里写图片描述

题目的关键是两数相加等于target ,只有一个solution ,所以返回的数组一定只有2个下标,以下为我的代码:

public class Solution {
    public int[] twoSum(int[] nums, int target) {
    //创建一个数组保存符合题目要求的index值
        int[] res= new int[2]; 
        //已知数组从取出下标为1的和第二个、第三个、第四个...数字依次相加与target数字相比是否相等,相等即立刻记录下标值,退出循环
        for(int i = 0;i<nums.length;i++) {
            for(int j = 0;j<nums.length;j++) {
            //两个相加的值与target相比是否相等,并且两个数的下标不能相等,保证了不会出现同一个下标的两个数值相加
                if (nums[i] + nums[j] == target && i != j) {
                    res[0]=i;
                    res[1]=j;  
                    break;
                }   
                }       
        }   
       return res;
    }

虽然以上代码比较容易明白,但是从效率上很有问题,因此还有另外一种方式,利用map集合,只要遍历数组一次即可以得出答案

public int[] twoSum(int[] numbers, int target) {
    int[] result = new int[2];
    Map<Integer, Integer> map = new HashMap<Integer, Integer>();
    for (int i = 0; i < numbers.length; i++) {
        if (map.containsKey(target - numbers[i])) {
            result[1] = i + 1;
            result[0] = map.get(target - numbers[i]);
            return result;
        }
        map.put(numbers[i], i + 1);
    }
    return result;
}

猜你喜欢

转载自blog.csdn.net/wangdajiao/article/details/53824433