您的位置 首页 java

1_两数之和(java)

两数之和

问题描述:

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例:

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

因为 nums[0] + nums[1] = 2 + 7 = 9

所以返回 [0, 1]

代码:

package algorithm;

import java.util.Arrays;

import java.util.HashMap;

import java.util.Map;

import java.util.Scanner;

public class Solution {

// 解法1:暴力法

public int[] two sum (int[] nums, int target) {

int i, j;

int sum;

// 外层循环遍历每一个元素

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

// 内层循环遍历除i以外的值

// 并判断是否存在一个值与nums[i]之和等于target

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

sum = nums[i] + nums[j];

if (sum == target) {

return new int[] { i, j };

}

}

}

throw new IllegalArgumentException(“No two sum solution”);

}

// 解法2 两遍哈希表

public int[] twoSum_1(int[] nums, int target) {

Map< Integer , Integer> map = new HashMap<>();

// 第一个循环,将数据元素的值和索引存到哈希表中

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

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

}

// 第二个循环,遍历各元素与目标值target的差值,是否存在于哈希表中

// 存在则输出该值与差值的索引值

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

int complement = target – nums[i];

if (map.containsKey(complement) && map.get(complement) != i) {

return new int[] { i, map.get(complement) };

}

}

throw new IllegalArgumentException(“No two sum solution”);

}

// 解法3 一遍哈希表

// 在迭代并将元素插入到哈希表中的同时,先校验哈希表中是否存在当前元素所对应的目标元素,如果有则直接返回

public int[] twoSum_2(int[] nums, int target) {

Map<Integer, Integer> map = new HashMap<>();

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

int complement = target – nums[i];

if (map.containsKey(complement)) {

return new int[] {i,map.get(complement)};

}

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

}

throw new IllegalArgumentException(“No two sum solution”);

}

// main

public static void main(String[] args) {

Solution solution = new Solution();

Scanner input = new Scanner(System.in);

System.out.println(“请输入数组长度:”);

int n = input.nextInt();

int[] nums = new int[n];

for (int i = 0; i < n; i++) {

System.out.println(“输入第” + i + “个数:”);

nums[i] = input.nextInt();

}

System.out.println(“请输入目标值:”);

int m = input.nextInt();

int target = m;

System.out.println(“目标值:” + target);

System.out.println(“整数数组 nums=” + Arrays. toString (nums));

// 计算一个方法执行时长,可用模板来写

long startTime = System.currentTimeMillis();

//int[] twoSum = solution.twoSum(nums, target);

//int[] twoSum_1 = solution.twoSum_1(nums, target);

int[] twoSum_2 = solution.twoSum_2(nums, target);

long endTime = System.currentTimeMillis();

float excTime = (float) (endTime – startTime) / 1000;

System.out.println(“结果=” + Arrays.toString(twoSum_2) + “,执行时间=” + excTime + ‘s’);

}

}

文章来源:智云一二三科技

文章标题:1_两数之和(java)

文章地址:https://www.zhihuclub.com/182599.shtml

关于作者: 智云科技

热门文章

网站地图