两数之和
问题描述:
给定一个整数数组 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[] twoSum(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');
}
}