1. 程式人生 > 實用技巧 >1、兩數之和

1、兩數之和

題目描述:

給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那兩個整數,並返回他們的陣列下標。
你可以假設每種輸入只會對應一個答案。但是,陣列中同一個元素不能使用兩遍。

示例:

給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/two-sum

語法複習:

//HashMap的使用語法,map裡的鍵不重複恰好符合題意
boolean containsKey(Object key);//是否存在指定 鍵 的對映關係
void put(Object key,Object value);//儲存鍵和值
Object get(Object key);  //返回指定鍵的對映關係,不存在則返回null

樸素版:

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] result=new int[2];
        int len=nums.length;
        for(int i=0;i<len-1;i++){
            for(int j=i+1;j<len;j++){
                if(nums[i]+nums[j]==target){
                    result[0]=i;
                    result[1]=j;
                }
            }
        }
        return result;
    }
}

優化版:

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

        int len = nums.length;

        HashMap<Integer,Integer> hashMap = new HashMap();//建立 HashMap

        for(int i = 0 ; i < len ; i++ ){
            if(hashMap.containsKey(target - nums[i])){
                //如果hashMap裡存在一個 鍵 能和 nums[i] 組成target
                //就獲取該 鍵 的 值
                return new int[]{hashMap.get(target - nums[i]), i};//注意i 是比 hashMap.get(target-nums[i]) 大的
            }
            else{
                hashMap.put(nums[i], i); //從前向後放
            }
        }
        return new int[]{};
    }
}