454.四数相加II
Map
这道题和四数之和很像,但是由于这道题看的是数组的下标是否相同,所以不用去重,那么直接用哈希表可以做到。时间复杂度O(n2)
sum, map.getOrDefault(sum, 0)
map.getOrDefault(sum, 0)
是 Java 中用于获取 Map 中指定键的值,如果键不存在,则返回指定的默认值。这里键sum存在返回value,否则返回0。
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i : nums1) {
for (int j : nums2) {
int sum = i + j;
// 键为num1和num2的和,值为出现的次数
map.put(sum, map.getOrDefault(sum, 0) + 1);
}
}
int count = 0;
for (int i : nums3) {
for (int j : nums4) {
if (map.containsKey(0 - i - j)) {
count += map.get(0 - i - j);
}
}
}
return count;
}
}
383. 赎金信
两种方法,一种是用数组记录,用时比Map的时间短,时间复杂度都为O(n)。就是将magazine出现的字母记录次数,在拿到ransoNote中减掉,map中的value小于0或者不存在这个字符,就return false。
数组
class Solution {
public boolean canConstruct(String ransomNote, String magazine) {
// 数组解决,同等情况下,使用数组的时间更快
int[] arr = new int[26];
char[] c1 = magazine.toCharArray();
char[] c2 = ransomNote.toCharArray();
for (char c : c1) {
arr[c - 'a']++;
}
for (char c : c2) {
if (arr[c - 'a'] == 0) {
return false;
}
arr[c - 'a']--;
if (arr[c - 'a'] < 0) {
return false;
}
}
return true;
}
}
Map
class Solution {
public boolean canConstruct(String ransomNote, String magazine) {
// 用哈希表解决
HashMap<Character, Integer> map = new HashMap<Character, Integer>();
char[] str1 = ransomNote.toCharArray();
char[] str2 = magazine.toCharArray();
for (char c : str2) {
map.put(c, map.getOrDefault(c, 0) + 1);
}
for (char c : str1) {
if (map.containsKey(c)) {
map.put(c, map.get(c) - 1);
if (map.get(c) < 0) {// 这里不是不等于0,可以有剩余
return false;
}
} else {
return false;
}
}
return true;
}
}
15. 三数之和
双指针
考虑使用双指针的方法,方法在于去重,对于第一个数字a,直接往下取就行,这里是nums[i-1]==nums[i],而不是nums[i]==nums[i+1],因为对于[0,0,0],i=0,left=1,如果是后者,那么就相当于判断a和b的值是否相等,这显然是不合理的,我们要判断的是a是否是重复值,例如[-1,-1,2]这个数组,当i=0时,nums[i]==nums[i+1],那么此时i=1,那么就会错过[-1,-1,2]这个解,相反,i=1的时候,nums[i-1]==nums[i]成立,那么就会跳过这个解,因为当i=0的时候,这个解已经被放进最终的数组了。
剪枝:这里由于数组已经排序了,所以第一个数字a>0,呢么直接就可以返回了。
对于b和c的去重,是在找到一个解的情况下进行的,因为[-1,-1,-1,0,1,1,1]这个时候,找到一个解[-1,1]就可以让left和right一直移动了,直到找到不重复的数。这个条件不是写在while循环里面,因为[-2,0,1,1,2],对于这个数组,[[-2,0,2],[-2,1,1]],后者的写法只返回第一个而缺失了[-2,1,1]这个数组。时间复杂度O(n2)
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
// 双指针方法
Arrays.sort(nums);
List<List<Integer>> result = new ArrayList<>();
int len = nums.length;
for (int i = 0; i < len; i++) {
if (nums[i] > 0) {
return result;
}
// 对a进行去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = len - 1;
while (left < right) {
int sum = nums[i] + nums[left] + nums[right];
if (sum > 0) {
right--;
} else if (sum < 0) {
left++;
} else {
result.add(Arrays.asList(nums[i], nums[left], nums[right]));
// 对left和right去重,注意只有存入一个数组,才去重,所以在else循环内部
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
left++;
right--;
}
}
}
return result;
}
}
18. 四数之和
四数之和和三数之和思想一样,只是加上一层for循环,记得在对j进行去重的时候,j>i+1,因为开始是在i开始的,由于要比较nums[j]和nums[j-1],所以j要大于i+1。注意尽管for循环规定了j从i+1开始,但是最终的比较即使j从第二个位置开始,它也不需要和第一个位置比较,而是从i+1第三个位置开始比较。时间复杂度O(n3)
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
// 双指针方法
Arrays.sort(nums);
List<List<Integer>> result = new ArrayList<>();
int len = nums.length;
for (int i = 0; i < len; i++) {
if (nums[i] > target) {
return result;
}
// 对a进行去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
for (int j = i + 1; j < len; j++) {
if(nums[i]+nums[j]>target){
return result;
}
// 对b去重
if (j > i + 1 && nums[j] == nums[j - 1]) {// 注意是j>i+1,而不是j>0,这里的j是下标
continue;
}
int left = j + 1;
int right = len - 1;
while (left < right) {
int sum = nums[i] + nums[j] + nums[left] + nums[right];
if (sum > target) {
right--;
} else if (sum < target) {
left++;
} else {
result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
// 对c和d去重,注意只有存入一个数组,才去重,所以在else循环内部
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
left++;
right--;
}
}
}
}
return result;
}
}