【力扣经典面试题150道(java)】1. 两数之和

题目

给定一个整数数组 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]

提示：

2 <= nums.length <= 10⁴
-10⁹ <= nums[i] <= 10⁹
-10⁹ <= target <= 10⁹
只会存在一个有效答案

解题思路

遍历nums数组，找到一个num[i]和一个nums[j]，满足i!=j且num[i] + num[j] == target，则输出i和j即可。

代码

时间复杂度：O(n²) （来自于2个for循环）
空间复杂度：O(1) （来自于定义的一些int变量）

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int t = nums.length;
        int i,j=0;
        int find = 0; //表示还没找到答案
        for (i=0;i<t;i++){
            for (j=i+1;j<t;j++) {
                if (nums[i]+nums[j]==target) {
                    find = 1; //已找到答案
                    break;
                }
            }

            if (find==1)
                break;
        }

        int []a = new int[2];
        a[0] = i;
        a[1] = j;
        return a;
    }
}

进阶拓展

进阶： 你可以想出一个时间复杂度小于 O(n²) 的算法吗？

思路： 建立一个Map记录每一个数值所在的位置，遍历nums数组，如果对于nums[i]在Map中能找到key == (target-nums[i])，则输出i和j。

时间复杂度： O(n) （主要来自于for循环）
空间复杂度： O(n)（主要来自于hash表）

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map <Integer, Integer> hash = new HashMap();
        int t = nums.length;
        int i = 0;
        for (i=0;i<t;i++) {
            if (hash.containsKey(target-nums[i])==true) //表示已经存在(target-nums[i])这个key
                return new int[]{i,hash.get(target-nums[i])};
            hash.put(nums[i],i);
        }
        return new int[]{i,hash.get(target-nums[i])};
    }
}