代码随想录算法训练营第6天|454. 四数相加II、383. 赎金信、15. 三数之和、18. 四数之和
454. 四数相加II
提交代码(暴力)
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
if(nums1.size() == 0) return 0;
vector<int> sum1;
vector<int> sum2;
int result = 0;
for(int i = 0; i < nums1.size(); i++)
for(int j = 0; j < nums2.size(); j++)
sum1.push_back(nums1[i] + nums2[j]);
for(int i = 0; i < nums3.size(); i++)
for(int j = 0; j < nums4.size(); j++)
sum2.push_back(nums3[i] + nums4[j]);
for(int i = 0; i < sum1.size(); i++)
{
for(int j = 0; j < sum2.size(); j++)
{
if(sum1[i] + sum2[j] == 0)
result++;
}
}
return result;
}
};
解答代码(方法)
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> map;
int result = 0;
for(int num1:nums1)
{
for(int num2:nums2)
{
map[num1 + num2]++;
}
}
for(int num3:nums3)
{
for(int num4:nums4)
{
if(map.count( - num3 - num4))
result += map[- num3 - num4];
}
}
return result;
}
};
383. 赎金信
提交代码(哈希)
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
vector<int> vec(26, 0);
for(int i = 0; i < magazine.size(); i++)
vec[magazine[i] - 'a']++;
for(int i = 0; i < ransomNote.size(); i++)
{
vec[ransomNote[i] - 'a']--;
if(vec[ransomNote[i] - 'a'] < 0)
return false;
}
return true;
}
};
解答代码(方法)
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
if (ransomNote.size() > magazine.size()) {
return false;
}
vector<int> cnt(26);
for (auto & c : magazine) {
cnt[c - 'a']++;
}
for (auto & c : ransomNote) {
cnt[c - 'a']--;
if (cnt[c - 'a'] < 0) {
return false;
}
}
return true;
}
};
15. 三数之和
提交代码(方法)
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() - 2; i++)
{
if(nums[i] > 0)
break;
if(i != 0 && nums[i] == nums[i - 1])
continue;
int left = i + 1;
int right = nums.size() - 1;
while(left < right)
{
if(left != i + 1 && nums[left] == nums[left - 1])
{
left++;
continue;
}
if(right != nums.size() - 1 && nums[right] == nums[right + 1])
{
right--;
continue;
}
if(nums[i] + nums[left] + nums[right] > 0)
right--;
else if(nums[i] + nums[left] + nums[right] < 0)
left++;
else
{
vector<int> vec{nums[i], nums[left], nums[right]};
result.push_back(vec);
left++;
right--;
}
}
}
return result;
}
};
解答代码(方法)
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() - 2; i++)
{
if(nums[i] > 0)
break;
if(i != 0 && nums[i] == nums[i - 1])
continue;
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
{
vector<int> vec{nums[i], nums[left], nums[right]};
result.push_back(vec);
while(left < right && nums[left] == nums[left + 1])
left++;
while(left < right && nums[right] == nums[right - 1])
right--;
left++;
right--;
}
}
}
return result;
}
};
18. 四数之和
提交代码(双指针)
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
if(nums.size() < 4)
return result;
sort(nums.begin(), nums.end());
for(int j = 0; j < nums.size() - 3; j++)
{
if(nums[j] > 0 && nums[j] > target)
break;
if(j != 0 && nums[j] == nums[j - 1])
continue;
for(int i = j + 1; i < nums.size() - 2; i++)
{
if((long)nums[i] + nums[j] > 0 && (long)nums[i] + nums[j] > target)
break;
if(i != j + 1 && nums[i] == nums[i - 1])
continue;
int left = i + 1;
int right = nums.size() - 1;
while(left < right)
{
if((long long) nums[j] + nums[i] + nums[left] + nums[right] > target)
right--;
else if((long long) nums[j] + nums[i] + nums[left] + nums[right] < target)
left++;
else
{
vector<int> vec{nums[j], nums[i], nums[left], nums[right]};
result.push_back(vec);
while(left < right && nums[left] == nums[left + 1])
left++;
while(left < right && nums[right] == nums[right - 1])
right--;
left++;
right--;
}
}
}
}
return result;
}
};
解答代码(方法)
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> quadruplets;
if (nums.size() < 4) {
return quadruplets;
}
sort(nums.begin(), nums.end());
int length = nums.size();
for (int i = 0; i < length - 3; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
if ((long) nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) {
break;
}
if ((long) nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) {
continue;
}
for (int j = i + 1; j < length - 2; j++) {
if (j > i + 1 && nums[j] == nums[j - 1]) {
continue;
}
if ((long) nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) {
break;
}
if ((long) nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target) {
continue;
}
int left = j + 1, right = length - 1;
while (left < right) {
long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
if (sum == target) {
quadruplets.push_back({nums[i], nums[j], nums[left], nums[right]});
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
left++;
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
right--;
} else if (sum < target) {
left++;
} else {
right--;
}
}
}
}
return quadruplets;
}
};
总结
日期: 2023 年 3 月 21 日
学习时长: 2 h 0 m
难度:
★
\bigstar
★
★
\bigstar
★
★
\bigstar
★
累计完成题目数量: 20
距离目标还有数量: 280
小结:
- 第二题最先做的,比较简单,和之前判断两个字符串能不能相互变换的思路一致。
- 第一题先是暴力,时间复杂度是N^2,超出时间限制了,我以为是N的平方,但实际上最后两个数组的长度都是N的平方,所以总的时间复杂度其实是N的四次方。然后想用哈希表的话,按照答案写一下把,哈希表其实也是N ^2的复杂度,。
- 三数之和写了一大堆代码来去重,最终打败19%哈哈哈,看下答案把。答案的去重能更优雅一些,是用的while循环,并且不用判断left和right是不是初始值。
- 第四题会出现整数溢出的情况,所以要转换为long
- 第四题的判断条件答案的感觉更多一些,它是前四个数和大于0就break,感觉能比我这个大于零的条件筛选掉更多的情况。还有计算当前和最后三个的和,如果小于target就continue。