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

题目:

给定一个整数数组 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] 
    

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

猜你喜欢

转载自blog.csdn.net/weixin_51749554/article/details/114326303
今日推荐