LeetCode-两数之和

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]

示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]

方法一:暴力枚举
两层循环遍历数组,对每一个i,在第二层循环找找相应的target-i,记录下标即可。时间复杂度O(n^2) 空间复杂度O(1)
方法二:哈希表
使用哈希表进行查找,对数组进行一次遍历,对于每一个i,首先从哈希表中寻找有无对应的target-i,如果有,则返回二者的下标。如果哈希表中没有对应的target-i,则将i加入到哈希表中,当继续遍历数组遍历到target-i时,则会在哈希表中找到已经保存的i。
时间复杂度O(n),空间复杂度O(n)。

import java.util.HashMap;
import java.util.Map;

public class Main {

    //两层循环遍历数组,对每一个i,在第二层循环找找相应的target-i,记录下标即可。
    //时间复杂度O(n^2) 空间复杂度O(1)
    public static int[] twoSum(int[] nums, int target) {
        int[] res = new int[2];
        for(int i = 0; i < nums.length; i++){
            for(int j = i+1; j < nums.length; j++){
                if (nums[j] == target - nums[i]){
                    res[0] = i;
                    res[1] = j;
                    return res;
                }
            }
        }
        return new int[0];
    }

    //使用哈希表进行查找,对数组进行一次遍历,对于每一个i,首先从哈希表中寻找有无对应的target-i,如果有,则返回二者的下标。
    //如果哈希表中没有对应的target-i,则将i加入到哈希表中,当继续遍历数组遍历到target-i时,则会在哈希表中找到已经保存的i、
    public static int[] twoSumHash(int[] nums, int target) {
         int[] res = new int[2];
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        for(int i = 0; i < nums.length; i++){
            if(map.containsKey(target - nums[i])){
                res[0] = map.get(target - nums[i]);
                res[1] = i;
                return res;
            }
            map.put(nums[i], i);
        }
        return new int[0];
    }

    public static void main(String[] args) {
        int[] arr = new int[]{2, 7, 11, 15};
        int target = 18;

        int[] res = twoSum(arr, target);
        int[] res2 = twoSumHash(arr, target);

        System.out.println(res[0]+" "+res[1]);
        System.out.println(res2[0]+" "+res2[1]);


    }
}

你可能感兴趣的:(LeetCode,leetcode,算法,java)