454.四数相加II
题目链接
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0。
为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间,最终结果不会超过 2^31 - 1 。
例如:
输入:
A = [ 1, 2]
B = [-2,-1]
C = [-1, 2]
D = [ 0, 2]
输出:
2
解释:
两个元组如下:
(0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
(1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0
重点
本题要计算的是有多少组,直接暴力解法,时间复杂度O(n^4)。
- 为什么用map?
有效字母异位词,只有二十六个字母,是可控的,所以用了数组。但本题数字范围不可控,而且不仅要记录数值,还要统计出现过的次数,所以用map类型。 - 本题解题步骤:
- 首先定义 一个unordered_map,key放a和b两数之和,value 放a和b两数之和出现的次数。
- 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中。 定义int变量count,用来统计 a+b+c+d = 0
- 出现的次数。 在遍历大C和大D数组,找到如果 0-(c+d)
- 在map中出现过的话,就用count把map中key对应的value也就是出现次数统计出来。 最后返回统计值 count 就可以了
代码
之前有想过统计一个countab,统计一个countcd,最后元组数就是countab*countcd,这样也行,但是这样就要AB、CD遍历统计完之后,再遍历一遍countcd,更复杂了。
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int> map;
for(int i=0;i<nums1.size();i++){
for(int j=0;j<nums2.size();j++){
if(map.find(nums1[i]+nums2[j])==map.end()){
map.insert(pair<int,int>(nums1[i]+nums2[j],1));
}
else{
map[nums1[i]+nums2[j]]++;
}
}
}
int count=0;
for(int i=0;i<nums3.size();i++){
for(int j=0;j<nums4.size();j++){
if(map.find(0-(nums3[i]+nums4[j]))!=map.end()){
//不能是count++;
count+=map[0-(nums3[i]+nums4[j])];
}
}
}
return count;
}
};
写完后再看官方答案,真的简洁了不是一点半点,for (int a : A)这种写法:
class Solution {
public:
int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
unordered_map<int, int> umap; //key:a+b的数值,value:a+b数值出现的次数
// 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中
for (int a : A) {
for (int b : B) {
umap[a + b]++;
}
}
int count = 0; // 统计a+b+c+d = 0 出现的次数
// 在遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就把map中key对应的value也就是出现次数统计出来。
for (int c : C) {
for (int d : D) {
if (umap.find(0 - (c + d)) != umap.end()) {
count += umap[0 - (c + d)];
}
}
}
return count;
}
};
时间复杂度O(n^2)。
383. 赎金信
题目链接
给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串,判断第一个字符串 ransom 能不能由第二个字符串 magazines 里面的字符构成。如果可以构成,返回 true ;否则返回 false。
(题目说明:为了不暴露赎金信字迹,要从杂志上搜索各个需要的字母,组成单词来表达意思。杂志字符串中的每个字符只能在赎金信字符串中使用一次。)
注意:
你可以假设两个字符串均只含有小写字母。
canConstruct(“a”, “b”) -> false
canConstruct(“aa”, “ab”) -> false
canConstruct(“aa”, “aab”) -> true
重点
这道题和有效字母异位非常像,首先考虑用数组。
代码
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int record[26]={0};
for(char str:ransomNote){
record[str-'a']++;
}
for(char str:magazine){
record[str-'a']--;
}
for(int count:record){
if(count>0) return false;
}
return true;
}
};
15. 三数之和
题目链接
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
重点
本题要求的三元组是不能重复的,用哈希法需要去重,比较麻烦。
-
双指针法:
要对数组进行排序,for循环确定a,left指针确定b,right指针确定c。
如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。
如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。这里就很能体现排序的好处。 -
有关去重
a如果重复了,应该直接跳过去,但这里有一个细节,就是判断 nums[i] 与nums[i + 1]是否相同,还是判断 nums[i] 与nums[i-1] 是否相同。如果是判断 nums[i] 与nums[i + 1]是否相同,那就把三元组中出现重复元素的情况直接pass掉了。 例如{-1, -1 ,2} 这组数据,当遍历到第一个-1 的时候,判断下一个也是-1,那这组数据就pass了。
b,c也按同样逻辑去重。
代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
//一定要先排序
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++){
if(nums[i]>0) break;//a大于零,肯定不能三数和为零
//if(i>0&&nums[i]==nums[i-1]) continue;//a的去重
int left=i+1;
int right=nums.size()-1;
while(left<right){
if(nums[i]+nums[left]+nums[right]>0){
right--;
}
else if(nums[i]+nums[left]+nums[right]<0){
left++;
}
else{
result.push_back(vector<int>{nums[i],nums[left],nums[right]});
// 去重逻辑应该放在找到一个三元组之后,对b 和 c去重
while(left<right&&nums[left]==nums[left+1]) left++;
while(left<right&&nums[right]==nums[right-1]) right--;
left++;
right--;
}
}
//把a的去重加到这里也行,这样abc的去重方法就统一了
//但要注意i的取值范围,防止下标越界
while(i<nums.size()-1&&nums[i]<=0&&nums[i]==nums[i+1]){
i++;
}
}
return result;
}
};
时间复杂度: O(n^2)
18. 四数之和
题目链接
题意:给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:
答案中不可以包含重复的四元组。
示例: 给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。 满足要求的四元组集合为: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]
重点
三数之和 的双指针解法是一层for循环num[i]为确定值,然后循环内有left和right下标作为双指针,找到nums[i] + nums[left] + nums[right] == 0。
四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值,依然是循环内有left和right下标作为双指针,找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况,三数之和的时间复杂度是O(n^2),四数之和时间复杂度是n的三次方。
那么一样的道理,五数之和、六数之和等等都采用这种解法。
代码
tips:
long sum=(long)(nums[a]+nums[b]+nums[left]+nums[right])会超出int范围,而long sum=(long)nums[a]+nums[b]+nums[left]+nums[right]不会,因为前者是把四个数先相加保存为int,再转为long,就超出了;而后者是把nums[a]变为long,long和int相加时会做整形提升,最终的和是long。
还有注意剪枝的处理:target是任意值。比如:数组是[-4, -3, -2, -1],target是-10,不能因为-4 > -10而跳过。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
sort(nums.begin(),nums.end());
for(int a=0;a<nums.size();a++){
//一级剪枝
if (nums[a]>target && nums[a] >= 0) break;
if(a>0&&nums[a]==nums[a-1]) continue;
for(int b=a+1;b<nums.size();b++){
//二级剪枝
if((nums[a]+nums[b])>target&&(nums[a]+nums[b])>=0) break;
if(b>a+1&&nums[b]==nums[b-1]) continue;
int left=b+1;
int right=nums.size()-1;
while(left<right){
// nums[k] + nums[i] + nums[left] + nums[right] > target 会溢出
//long sum=(long)(nums[a]+nums[b]+nums[left]+nums[right]);
long sum=(long)nums[a]+nums[b]+nums[left]+nums[right];
if(sum>target){
right--;
}
else if(sum<target){
left++;
}
else{
result.push_back(vector<int>{nums[a],nums[b],nums[left],nums[right]});
while(left<right&&nums[left]==nums[left+1]) left++;
while(left<right&&nums[right]==nums[right-1]) right--;
left++;
right--;
}
}
}
}
return result;
}
};