1. 自我刷题之两数之和

回归次数:1
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处

题目:

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:

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

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

解答:

public class Tes {

    /**
     * 最简单思路,两次遍历
     * 空间复杂度:O(n * n)
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum1(int[] nums, int target) {

        for (int i = 0;i < nums.length;i ++){

            for(int j = i + 1;j < nums.length;j ++){

                if (nums[i] + nums[j] == target){

                    return new int[]{i,j};
                }
            }
        }

        return new int[0];
    }

    /**
     * 优化,放入hashmap一次,循环数组对比的时候,O(1)
     * 空间复杂度:O(n)
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum2(int[] nums, int target) {

        Map<Integer,Integer> map = new HashMap<>();
        for (int i = 0;i < nums.length;i ++){

            map.put(nums[i],i);
        }

        for (int i = 0;i < nums.length;i ++){

            int current = target - nums[i];
            if (map.containsKey(current) && i != map.get(current)){
                //说明存在
                return new int[]{i,map.get(current)};
            }
        }

        return new int[0];
    }

    /**
     * 一次hashmap,直接对比
     * 空间复杂度:O(n)
     * 但是比上一个优化n更优
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum(int[] nums, int target) {

        Map<Integer,Integer> map = new HashMap<>();
        for (int i = 0;i < nums.length;i ++){

            //首先判断需要的另外一个数
            int current = target - nums[i];
            if (map.containsKey(current)){

                return new int[]{i,map.get(current)};
            }else{
                map.put(nums[i],i);
            }
        }

        return new int[0];
    }

    public static void main(String[] args) {

        int[] test = new int[]{1,3,4,5,6};
        Tes tes = new Tes();
        int[] r = tes.twoSum(test,7);
        for (int i :
                r) {
            System.out.println(i);
        }
    }
}


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。