代码随想录 -- 哈希表--两数之和

力扣hot1:两数之和

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9

所以返回 [0, 1]

完成代码可运行:

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

public class Two {
    public int[] twoSum(int[] nums, int target) {
        int[] res = new int[2];
        if(nums == null || nums.length == 0){
            return res;
        }
        Map map = new HashMap<>();
        for(int i = 0; i < nums.length; i++){
            int temp = target - nums[i];   // 遍历当前元素,并在map中寻找是否有匹配的key
            if(map.containsKey(temp)){
                res[1] = i;
                res[0] = map.get(temp);
                break;
            }
            map.put(nums[i], i);    // 如果没找到匹配对,就把访问过的元素和下标加入到map中
        }
        return res;
    }

    public static void main(String[] args) {
        int[] n = new int[]{4, 6, 9, 5};
        Two two = new Two();
        int t = 11;
        int[] result = two.twoSum(n, t);
        System.out.println("[" + result[0] + ", " + result[1] + "]");

    }
}

你可能感兴趣的:(代码随想录,数据结构,算法)