四数相加 II
这到题没想出太好的办法,直接看的视频。
思路:将四个数组两两分组遍历,可以先遍历A、B数组,将A[i] + B[j]每一个求出来放入map中,map的值记录相加后的值出现几次。时间复杂度O(n^2). 然后对于剩下的两个数组也是相同的操作,不过加入map中的是 0 - (C + D),因为 A + B + C + D = 0,所以 A + B = 0 - (C + D),然后用一个变量记录总出现次数即可。时间复杂度O(n^2).
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
HashMap<Integer, Integer> map = new HashMap<>();
int res = 0;
for (int i : nums1) {
for (int j : nums2) {
int tmp = i + j;
if (map.containsKey(tmp)) {
map.put(tmp, map.get(tmp) + 1);
} else {
map.put(tmp, 1);
}
}
}
for (int i : nums3) {
for (int j : nums4) {
int tmp = i + j;
if (map.containsKey(0 - tmp)) {
res += map.get(0 - tmp);
}
}
}
return res;
}
}
赎金信
第一次过的代码。
class Solution {
public boolean canConstruct(String ransomNote, String magazine) {
int[] ans = new int[26];
for (int i = 0; i < ransomNote.length(); i++) {
ans[ransomNote.charAt(i) - 'a']++; //记录ransomNote字符串中每个字符出现次数
}
for(int i = 0; i < magazine.length(); i++) {
ans[magazine.charAt(i) - 'a']--; //记录magazine字符串中每个字符出现次数
}
for (int i = 0; i < 26; i++) {
if (ans[i] < 0) { //如果小于零,说明ransomNote中不包含这个字符,继续循环
continue;
}
if (ans[i] > 0) {
return false;
}
}
return true;
}
}
题解的代码
直接记录magazine中的各个字符,用ransomNote中的字符去减,如果有小于零的,就说明magazine中的字符不能完全组成ransomNote。
class Solution {
public boolean canConstruct(String ransomNote, String magazine) {
// 定义一个哈希映射数组
int[] record = new int[26];
// 遍历
for(char c : magazine.toCharArray()){
record[c - 'a'] += 1;
}
for(char c : ransomNote.toCharArray()){
record[c - 'a'] -= 1;
}
// 如果数组中存在负数,说明ransomNote字符串总存在magazine中没有的字符
for(int i : record){
if(i < 0){
return false;
}
}
return true;
}
}
三数之和
这道题直接看的视频讲解,之后回去看了哈希的解法,但是哈希的去重实在是没看懂,所以只写了双指针的解法。
具体逻辑还是很好理解的,但是去重的细节一定要注意。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
for (int i = 0; i < nums.length; i++) {
if (nums[i] > 0) { //如果第一个数大于零 不可能达成条件 直接返回
return res;
}
if(i > 0 && nums[i] == nums[i - 1]) { //去重,如果当前的值等于前一个的值
continue; //那么如果当前值存在一个三元组,前一个值也会存在一个三元组
} //并且和这个重复
int left = i + 1;
int right = nums.length - 1;
while (right > left) {
int sum = nums[i] + nums[left] + nums[right];
if (sum > 0) {
right--;
} else if (sum < 0) {
left++;
} else {
res.add(Arrays.asList(nums[i], nums[left], nums[right]));
//去重
while (right > left && nums[left] == nums[left + 1]) {
left++;
}
while (right > left && nums[right] == nums[right - 1]) {
right--;
}
right--;
left++;
}
}
}
return res;
}
}
四数之和
与三数之和是一样的思路,在外面再多加一层循环即可。
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
for (int k = 0; k < nums.length; k++) {
if (nums[k] > target && nums[k] >= 0) {
break;
}
if(k > 0 && nums[k] == nums[k - 1]) {
continue;
}
for (int i = k + 1; i < nums.length; i++) {
if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) {
break;
}
if (i > k + 1 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = nums.length - 1;
while (right > left) {
long sum = (long) nums[k] + nums[i] + nums[left] + nums[right];
if (sum > target) {
right--;
} else if (sum < target) {
left++;
} else {
res.add(Arrays.asList(nums[k], nums[i], nums[left], nums[right]));
while (right > left && nums[right] == nums[right - 1]) {
right--;
}
while (right > left && nums[left] == nums[left + 1]) {
left++;
}
right--;
left++;
}
}
}
}
return res;
}
}