leetcode算法第一题Two sum

LEETCODE算法汇总

最近一直刷算法题,努力提高自身的算法功底.如果学习过程中不总结,很快就忘记了,所以要养成总结习惯,近期会不间断的写一写自己的心得以及对题目的认识.大佬可以跳过

leetcode第一题

题目

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

题目概述

我们看题目可以知道,这个题目是找到数组中和为target的下标.题目补充说明为,我们假设数组中只存在一组这样的数.因此题目就变的简单了.

第一种解题

大家应该都可以想起来的一种比较复杂的方法为,双重循环,这样的话就是复杂度为$o(n^2)$.

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

做完这个我们应该下一步提升算法的效率,参考一下discussion,得到一种较好的算法为第二种

进阶的算法

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] arr = new int[2];
        Map<Integer,Integer> map = new HashMap<Integer,Integer>();
        for(int i=0; i<nums.length;i++){
            if(map.containsKey(target-nums[i])){
                arr[1] = i;
                arr[0] = map.get(target-nums[i]);
            }
            map.put(nums[i],i);
        }
        return arr;
    }
}

我们可以看一下算法执行的时间