454.四数相加II
再次熟悉unordered_map的用法,这道题对我来说,难点在于不会用unordered_map完成数组的数的出现次数统计。
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> record;
int ans = 0;
for (int i = 0; i < nums1.size(); i++) {
for (int j = 0; j < nums2.size(); j++) {
int sum1 = nums1[i] + nums2[j];
record[sum1]++;
}
}
for (int i = 0; i < nums3.size(); i++) {
for (int j = 0; j < nums4.size(); j++) {
int sum2 = nums3[i] + nums4[j];
if (record.find(0 - sum2) != record.end()) {
ans += record[0 - sum2];
}
}
}
return ans;
}
};
383. 赎金信
通过数组构造哈希表,思路和有效字母异位词相似。
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int hash[26];
if (ransomNote.size() > magazine.size()) {
return false;
}
for (auto i : magazine) {
hash[i - 'a']++;
}
for (auto j : ransomNote) {
hash[j - 'a']--;
}
for (int i = 0; i < 26; i++) {
if (hash[i] < 0) {
return false;
}
}
return true;
}
};
15. 三数之和
看了卡哥的视频,大概理解思路。边看答案边尝试着自己写。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> res;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] > 0) {
return res;
}
if (i > 0 && nums[i - 1] == nums[i]) {
continue;
}
int l = i + 1;
int r = nums.size() - 1;
while (l < r) {
if (nums[i] + nums[l] + nums[r] > 0) {
r--;
}
else if (nums[i] + nums[l] + nums[r] < 0) {
l++;
}
else {
res.push_back(vector<int>{nums[i], nums[l], nums[r]});
while (l < r && nums[l + 1] == nums[l]) l++;
while (l < r && nums[r - 1] == nums[r]) r--;
l++;
r--;
}
}
}
return res;
}
};
18. 四数之和
有了三树之和的基础,做这道题简单很多。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
if (nums[i] >= 0 && nums[i] > target) {
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 l = j + 1;
int r = nums.size() - 1;
while (l < r) {
if ((long)nums[i] + nums[j] + nums[l] + nums[r] < target) {
l++;
}
else if ((long)nums[i] + nums[j] + nums[l] + nums[r] > target) {
r--;
}
else {
res.push_back(vector<int>{nums[i] ,nums[j] ,nums[l] ,nums[r]});
while (l < r && nums[l + 1] == nums[l]) l++;
while (l < r && nums[r - 1] == nums[r]) r--;
l++;
r--;
}
}
}
}
return res;
}
};
总结
今天又是充实的一天。