Java哈希表求解两数之和

88 阅读1分钟

我报名参加金石计划1期挑战——瓜分10万奖池,这是我的第2篇文章,点击查看活动详情

题目描述:

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

可以按任意顺序返回答案。

这里我们使用哈希表进行解决 速度会比较快 不用对数组进行遍历

代码如下

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

class Solution {
    public int[] twoSum(int[] nums, int target) {

        int v;
        int index[] = new int[2];
        Map<Integer, Integer> mp = new HashMap<Integer, Integer>();

        for(int k=0; k<nums.length; k++){
            v = target - nums[k];
            if(mp.containsKey(v)){
                index[0] = mp.get(v);
                index[1] = k;
                break;
            }else{
                mp.put(nums[k], k);
            }
        }
        return index;
    }
}

主要思路就是判断哈希表中是否存在target - nums[i]的值,

若存在,直接返回,不存在,则把nums[i]加入到哈希表中

然后就能得出最后的结果