代码随想录训练营第七天
leetcode454:四数相加II
先遍历前两个数组,并求对应的和,并把该和的值作为一个键值保存在一个unordered_map中,其中对应的value值是键值出现的次数,再遍历后两个数组,将保存有0-C[k]-D[u]的value值全部相加。
代码如下:
class Solution
{
public:
int countfunction(vector<int>num1, vector<int>num2, vector<int>num3, vector<int>num4)
{
unordered_map<int,int> map;
int count_num = 0;
for (int i=0;i<num1.size();i++)
{
for (int j=0;j<num2.size();j++)
{
int sum_AB = num1[i] + num2[j];
if (map.find(sum_AB) == map.end())
{
map.insert(pair<int, int>(sum_AB, 1));
}
else
{
map[sum_AB]++;
}
}
}
for (int i = 0; i < num3.size(); i++)
{
for (int j = 0; j < num4.size(); j++)
{
int sum = num3[i] + num4[j];
if (map.find(0 - sum) != map.end())
{
count_num += map[-sum];
}
}
}
return count_num;
}
};
leetcode15:三数相加
通过两次循环,先固定第一个数,将三数相加变为两数相加,从而获得和为0的三元组,其中每轮循环前需要进行去重处理,而与两数相加一样,遍历到暂时还不符合条件的元素时,存储在unordered_set保存,方便一次查询。
代码如下:
class Solution//三数之和
{
public:
vector<vector<int>> threesum(vector<int>num, int target)
{
vector<vector<int>>result;
sort(num.begin(), num.end());
for (int i=0;i<(num.size()-2);i++)
{
if(num[i]>0)
continue;
if (i > 0 && num[i - 1] == num[i])
{
continue;
}
unordered_set<int>set;
for (int j = i + 1; j < num.size(); j++)
{
if (j > i + 2 && num[j] == num[j - 1] && num[j - 1] == num[j - 2])
{
continue;
}
int c = 0 - num[i] - num[j];
if (set.find(c) != set.end())
{
result.push_back({ num[i],num[j],c });
set.erase(c);
}
else
{
set.insert(num[j]);
}
}
}
return result;
}
};
leetcode18:四数之和
四数之和当首先固定四元组中的第一个值时,便转化为三数之和的情况,所以需要使用三个循环,并且使用unordered_set保存暂时不符合条件的元素
代码如下:
class Solution7//四数之和
{
public:
vector<vector<int>> fourNum(vector<int>num)
{
vector<vector<int>> result;
sort(num.begin(), num.end());
for (int i=0;i<(num.size()-3);i++)
{
if(i>0&&num[i]==num[i-1])
continue;
for (int j=i+1;j<(num.size()-2);j++)
{
unordered_set<int> set;
if(j>i+1&&num[j]==num[j-1])
continue;
for (int k=j+1;k<num.size();k++)
{
if (k > j + 2 && num[k] == num[k - 1] && num[k - 1] == num[k - 2])
{
continue;
}
int c = 0 - num[i] - num[j] - num[k];
if (set.find(c) != set.end())
{
result.push_back({ num[i],num[j],num[k],c });
set.erase(c);
}
else
{
set.insert(num[k]);
}
}
}
}
}
};
leetcode383:赎金信
与有效字母异位词题目相同,将两个字符串的每个字符映射到长度为26的整型数组上,如果第一个字符串对应的整型数组上的各元素值小于或等于第二个字符串对应的整型数组各元素值,则说明可以替换。
代码如下:
class Solutuon7
{
public:
bool canConstruct(string ransomNote, string magazine)
{
vector<int>arr(26, 0);
if (ransomNote.size() > magazine.size())
return false;
for (int i=0;i<ransomNote.size();i++)
{
int a = int(ransomNote[i] - 'a');
arr[a]++;
}
for (int i=0;i<magazine.size();i++)
{
int a = int(magazine[i] - 'a');
arr[a]--;
}
for (int i=0;i<26;i++)
{
if (arr[i]>0)
{
return false;
}
}
return true;
}
};