问题描述
输入一个数组和一个值,寻找数组中任意两个数的和等于给定的值的数组下标,如果有很多数字的和等于给定的值,则随意输出一组下标就可以了。
思路
先来一种简单的方法:遍历每个元素,对于任意a[i],只需要遍历寻找是否存在值为sum - a[i]的数组元素就好了。
不说废话了,直接上代码!
package interview;
/**
* @author xing xing
*/
public class TwoSum {
public static void main(String[] args) {
//定义一个数组
int [] array = {1,-2,-4,5,9,-3,-8,6};
twoSum(array,-2);
}
public static void twoSum(int [] array,int sum) {
for (int i = 0; i < array.length; i++) {
for (int j = i+1; j <array.length ; j++) {
//如果array[i]+array[j] == sum,说明找到了
if ((sum - array[i]) == array[j]) {
System.out.println("两个数分别为:"+array[i]+" "+array[j]);
return;
}
}
}
//如果遍历完毕以后没有找到
System.out.println("没找到!!!");
}
//打印数组
public static void print(int[] array) {
for (int i = 0; i < array.length; i++) {
System.out.println(array[i]);
}
}
}
这种方法虽然简单,但性能很差,在数值内容特别少的情况下也可以使用。
第二种方法:我们可以先遍历一遍数组,把值放入散列表中,再遍历一遍数组,分别·判断与其对应的值是否存在,直接上代码!
package interview;
import java.util.HashMap;
/**
* @author xing xing
*/
public class TwoSum {
public static void main(String[] args) {
//定义一个数组
int [] array = {1,-2,-4,5,9,-3,-8,6};
twoSum(array,-2);
}
public static void twoSum(int [] array,int sum) {
HashMap<Integer,Integer> hashMap = new HashMap<>();
for (int i = 0; i < array.length; i++) {
hashMap.put(array[i],i);
}
int index;
for (int i = 0; i < array.length; i++) {
index = hashMap.get(sum-array[i]);
if (index != -1 && index != i) {
System.out.println("两个数分别为:"+array[index]+" "+array[i]);
return;
}
}
//如果遍历完毕以后没有找到
System.out.println("没找到!!!");
}
//打印数组
public static void print(int[] array) {
for (int i = 0; i < array.length; i++) {
System.out.println(array[i]);
}
}
}
散列表的空间复杂度不太好,在最好的情况下是O(n),时间复杂度会比较好,我们认为是O(1),但实际要差一点(因为当导致哈希冲突的时候会影响性能),这种方法也算是一种比较好的,使用散列表时,只需要对数组进行两次遍历,所以时间复杂度为O(n)。
最后附上leetcode用一次遍历实现的代码:
class Solution {
public int[] twoSum(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[] { map.get(complement), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
}
总结
在学习算法题目的时候,多观察,多总结,先尝试用自己的方法做,然后总结每种方法有什么优缺点,之后再看看别人是怎么写的,吸收别人的经验与总结。