力扣(LeetCode) -- 算法第一题-- 两数之和

Posted 2021-06-14 一号程序员

题目：

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum

分析：

哈希查找，遍历数组 nums，i为当前下标，每个值都判断map中是否存在 target-nums[i] 的key值，如果存在则找到两个值，如果不存在则将当前（nums[i]，i）存入map中，继续遍历找到为止。

实现：

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;


public class Solution {
public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);

    int n = 0;//存放数组长度

    int[] nums = new int[10];//存放nums整数数组

    int target = 0;//存放目标target的值

    System.out.println("请输入nums数组的长度");

    n = sc.nextInt();

    System.out.println("请输入nums数组");

    for (int i = 0; i <= n - 1; i++) {
        nums[i] = sc.nextInt();
    }

    System.out.println("请输入目标target的值");

    target = sc.nextInt();

    int[] ints = twoSum(nums, target);

    System.out.println("结果为"+Arrays.toString(ints));
}

public static int[] twoSum(int[] nums, int target) {
    Map<Integer, Integer> codeMap = new HashMap<Integer, Integer>(nums.length - 1);//创建一个存储数据的Map集合
    for (int i = 0; i <= nums.length; i++) {
        if (codeMap.containsKey(target - nums[i])) {//存在 target-nums[i] 的key值
            return new int[]{codeMap.get(target - nums[i]), i};//返回值
        }
        codeMap.put(nums[i], i);//不存在，继续找
    }
    throw new IllegalArgumentException("No  two  sum  solution");
}
}

运行结果：

• 结果1：

  请输入nums数组的长度
  4
  请输入nums数组
  2
  7
  11
  15
  请输入目标target的值
  9
  结果为[0, 1]
  

• 结果2：

  请输入nums数组的长度
  3
  请输入nums数组
  3
  2
  4
  请输入目标target的值
  6
  结果为[1, 2] 
  

感谢你的阅读，不足之处欢迎批评指正！