Day7
今天的四道题目前两道非常简单,通过哈希表计数就可以解决
后两个求和比较难,二刷了还是忘记了剪枝的判断,总体框架掌握了。还要继续努力。
454. 4Sum II
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> mp;
int ans = 0;
for(int i : nums1)
{
for(int j : nums2)
{
++mp[i+j];
}
}
for(int i : nums3)
{
for(int j : nums4)
{
ans += mp[0 - i - j];
}
}
return ans;
}
};
383. Ransom Note
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
if(magazine.length() < ransomNote.length())
return false;
unordered_map<char, int> mp;
for(char c : magazine)
++mp[c];
for(char c : ransomNote)
{
if(--mp[c] < 0)
return false;
}
return true;
}
};
15. 3Sum
忘了双指针,还在用哈希表
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ans;
sort(nums.begin(), nums.end());
for(int i=0; i<nums.size(); ++i)
{
if(nums[i] > 0)
return ans;
if(i > 0 && nums[i] == nums[i-1])
continue;
int left = i+1;
int right = nums.size()-1;
while(left < right)
{
if(nums[right] + nums[left] + nums[i] > 0)
{
right--;
}
else if(nums[right] + nums[left] + nums[i] < 0)
{
left++;
}
else
{
ans.push_back({nums[i], nums[left], nums[right]});
while(left < right && nums[right] == nums[right-1])
--right;
while(left < right && nums[left] == nums[left+1])
++left;
++left;
--right;
}
}
}
return ans;
}
};
18. 4Sum
剪枝没做好,忘记了复数+复数可能比target小,轻易判断只要>target就剪掉了
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
vector<vector<int>> ans;
for(int i=0; i<nums.size(); ++i)
{
if(nums[i] > target && nums[i] >= 0)
break;
if(i > 0 && nums[i] == nums[i-1])
continue;
for(int j=i+1; j<nums.size(); ++j)
{
if(nums[i] + nums[j] > target && nums[i] + nums[j] >= 0)
break;
if(j>i+1 && nums[j] == nums[j-1])
continue;
int left = j+1;
int right = nums.size()-1;
while(left < right)
{
if((long)nums[i] + nums[j] + nums[left] + nums[right] > target)
{
--right;
}
else if((long)nums[i] + nums[j] + nums[left] + nums[right] < target)
{
++left;
}
else
{
ans.push_back({nums[i], nums[j], 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 ans;
}
};