解法一
两重循环,遍历所有情况看相加是否等于目标和,如果符合直接输出。
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] ans = new int[2];
for(int i=0;i<nums.length;i++){
for(int j=i+1;j<nums.length;j++){
if(nums[i]+nums[j]==target){
ans[0]=i;
ans[j]=j;
return ans;
}
}
}
return ans;
}
}
时间复杂度:两层 for 循环,O(n^2)
空间复杂度:O(1)
解法二
上一种解法中的第二层 for 循环只是为了再次遍历找到另一个值,时间复杂度为 O(n)。
有没有一种方法,不用遍历就可以找到另一个值的?
hash table !!!
我们可以把数组的每个元素保存为 hash 的 key,下标保存为 hash 的 value 。
这样只需判断另一个值在不在 hash 的 key 里就可以了,而此时的时间复杂度仅为 O(1)。
需要注意的地方是,还需判断找到的元素不是当前元素,因为题目里讲一个元素只能用一次。
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
for(int i=0;i<nums.length;i++){
map.put(nums[i],i);
}
for(int i=0;i<nums.length;i++){
int sub = target - nums[i];
if(map.containsKey(sub)&&map.get(sub)!=i){
return new int[]{i,map.get(sub)};
}
}
throw new IllegalArgumentException("No two sum solution");
}
}
map 是一个 key 和 value 的键值对的集合。有 key 和 value 键值对,就会有判断是否有 key。这方法就是 containsKey 方法。
如:
if(map.containsKey(“name”)){
value=map.get(“name”).toString();
System.out.println(“找到了name的值:”+value);
}
map 中的 containsKey(key) 方法是判断该 key 在 map 中是否有 key 存在。如果存在则返回 true。如果不存在则返回 false。
时间复杂度:仅用了一个 for 循环,O(n)
空间复杂度:开辟了一个 hash table,O(n)
解法三
解法二中两个 for 循环长的一样,我们可以把它们合起来,复杂度上没有什么变化,变化仅仅是不需要判断是不是当前元素了,因为当前元素还没有添加进 hash 里。
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
for(int i=0;i<nums.length;i++){
int sub = target - nums[i];
if(map.containsKey(sub)){
return new int[]{i,map.get(sub)};
}
map.put(nums[i],i);
}
throw new IllegalArgumentException("No two sum solution");
}
}
总结
对 hash 的应用,可以让时间复杂度从 O(n^2) 降为 O(n)。