每日一道算法面試題(1):leetcode1兩數之和

算法在編程領域的重要性不言而喻,而且也是好多大廠面試經常要考覈的重點。

兩數之和

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

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

示例:

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

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

算法思路:

  1. 暴力法:使用兩層循環,外層循環計算當前元素與 target 的差值,內層循環尋找該差值,若找到該差值,則返回兩個元素的下標;時間複雜度:O(n^2)
  2. 利用HashMap減少查詢時間:HashMap 可以根據下標快速查找數據,減少了查詢時間,結合這個特性使用一層循環即可以實現:每遍歷一個元素就計算該元素與 target 之間的差值,然後到 HashMap 中查找該差值,如果找到則返回兩個元素的下標,如果沒有找到,則將元素存入 HashMap 中(Key:nums[i],Value:i);時間複雜度:O(n)。

算法代碼:根據算法思路2,寫出的算法具體代碼如下:

    /**
     * 兩數之和
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int[] res = new int[2];
        for (int i = 0; i < nums.length; i ++) {
            int value = target - nums[i];
            if (map.get(value) != null) {
                res[0] = map.get(value);
                res[1] = i;
            }
            map.put(nums[i], i); // 構建 HashMap
        }
        return res;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章