Leetcode(1)兩數之和

題目:

給定一個整數數組 nums 和一個目標值 target,請你在該數組中找出和爲目標值的那 兩個 整數,並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個數組中同樣的元素。
示例:

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

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

思路:
這裏採用HashMap這種數據結構,將數組的元素值作爲key,index作爲value來建立map,這麼做的原因是HashMap這種數據類型有通過key獲取value的方法,但是沒有直接通過value獲取key的方法

    步驟:
            1、建立一個空的HashMap類型的數據表map
            2、對數組進行循環遍歷,在每次遍歷elem的時候,首先查看一下target-elem是否已經在map中
                    1)如果已經在了,則說明已經找到了,此時通過map的get方法查詢到target-elem的索引值,並把它和當前的elem的索引值一起返回即可
                    2)如果不在map中,則將elem作爲key,index所謂value存入到map中
            3、若2中的循環結束了,還沒有找到,則拋出一個異常即可

    注意:這裏需要對HashMap這種數據類型的常用方法比較熟悉,如果不熟悉可以通過查看JDK的API文檔即可
import java.util.HashMap;
import java.util.Map;
class Solution {
    public int[] twoSum(int[] nums, int target)  {
     Map<Integer,Integer> map=new  HashMap<Integer,Integer>();
     for(int i=0;i<nums.length;i++)
     {
         int temp=target-nums[i];
         if(map.containsKey(temp))
         {
             return new int[]{map.get(temp),i};
         }
         map.put(nums[i],i);
     }
     throw new IllegalArgumentException("在數組nums中沒有找到兩個數,使得他們的和胃target");
    }
}

public class SolutionDemo {
    public static void main(String[] args) {
        Solution s=new Solution();
        int[] nums=new int[]{2,7,11,15};
        int target=9;
        int[] index=s.twoSum(nums, target);
        System.out.println("index =  ["+index[0]+", "+index[1]+"]");
    }
}

執行結果爲:

index = [0, 1]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章