1. 程式人生 > >leetcode刷題之Two Sum(1)

leetcode刷題之Two Sum(1)

給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。

你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個陣列中同樣的元素。

示例:

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

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

思路
建立一個HashMap,key為陣列中的元素值,value為陣列下表,遍歷陣列,直到找到key為target - 陣列元素的鍵值對,時間複雜度O(n)

程式碼

Java

public class TwoSum {

public static int[] solution(int[] nums,int target){

    Map<Integer,Integer> map=new HashMap<>();

    int[] result=new int[2];

    for(int i=0;i<nums.length;i++){
        if(map.containsKey(target - nums[i])){
            result[0]=map.get(target - nums[i]);
            result[1]=i;
        }
        map.put(nums[i],i);
    }
    return result;
}

}