两数之和的hashmap问题（key和value的关系）

Description:

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的两个整数。

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

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

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

首先暴力求解

public static 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 };
                }
            }
        }
    }

当然如题，肯定有个hashmap的求解方法

public static 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) };
        }
    }
}
/*数组value 3 0 3
下标index 0 1 2
map里的key-value：(3,2)和（0,1);
当第二遍遍历数组的时候,访问到index=0,此时得到的map.get(complememt)为2,满足条件；*/

package p1;

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

public class _finally {
    public static void main(String[] args){
        Scanner scanner=new Scanner(System.in);
        int[] num=new int[10000];
        int target,n;
        n=scanner.nextInt();
        for(int i=0;i<n;i++)
            num[i]=scanner.nextInt();
        target=scanner.nextInt();
        int[] a=TwoSum(num,target);
        System.out.println(a[0]+" "+a[1]);
    }
    public static int[] TwoSum(int[] num,int target){
        Map<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<num.length;i++)
            map.put(num[i],i);
        for(int i=0;i<num.length;i++){
            int complement=target-num[i];
            if(map.containsKey(complement)&&map.get(complement)!=i){
                return new int[]{i,map.get(complement)};
            }
        }
        throw new IllegalArgumentException("No two sum solution");
    }
}

第一眼看到这个题，要考虑数组重复value的值。其实手拿笔画一画，这个HashMap的方法真是美妙。

还有HashMap的key和value的值一一对应的。key是唯一的，除非你重构了hashCode()方法和equals()方法。

如果原本已经存在对应的key，则直接改变对应的value，并返回旧的value；如果之前不存在则返回null；

两数之和的hashmap问题（key和value的关系）

猜你喜欢