四数相加Ⅱ
代码随想录
454. 四数相加 II - 力扣(LeetCode)
v1.0:直接超时,完全可以把0 - nums1[i]-nums2[j]作为一组,避免了三重循环!
class Solution {
public:int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {unordered_map<int,int> map;int n = nums1.size();for(int i = 0;i<n;i++){for(int j = 0;j<n; j++){for(int k = 0;k<n;k++){int key = 0 - nums1[i]-nums2[j]-nums3[k];map[key]++;}}}int count = 0;for(int l=0;l<n;l++){if(map.find(nums4[l])!=map.end()){count+=map[nums4[l]];}}return count;}};v2.0:直接改进 nums[i]和nums[j]一组,避免三重循环
class Solution {
public:int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {unordered_map<int,int> map;int n = nums1.size();for(int i = 0;i<n;i++){for(int j = 0;j<n; j++){int key = 0 - nums1[i]-nums2[j];map[key]++;}}int count = 0;for(int k = 0;k<n;k++){for(int l=0;l<n;l++){if(map.find(nums3[k]+nums4[l])!=map.end()){count+=map[nums3[k]+nums4[l]];}}} return count;}
};v3.0:改用rangefor循环加速
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:nums1){for(int j :nums2){int key = 0 - i - j;map[key]++;}}int count = 0;for(int k:nums3){for(int l:nums4){if(map.find(k+l)!=map.end()){count+=map[k+l];}}} return count;}
};
赎金信
383. 赎金信 - 力扣(LeetCode)
代码随想录
v1.0:直接使用数组来统计字母出现次数,然后减去相应的字母次数即可 初见秒
class Solution {
public:bool canConstruct(string ransomNote, string magazine) {if(ransomNote.size()>magazine.size())return false;int chars[26] = {0};for(char c: magazine){chars[c-'a']++;}for(char c: ransomNote){chars[c-'a']--;if(chars[c-'a']<0){return false;}}return true;}
};
三数之和
15. 三数之和 - 力扣(LeetCode)
代码随想录
v1.0:
这道题只需要返回数值而不是下标,所以可以先对数组进行排序
使用三指针方法:最外层一个for循环,i是最小的值
然后对i进行判断,如果i>0直接返回,不用在找了
再使用两个指针,一个指向i+1,一个指向数组最后,分别是中间值和最大值,
然后固定i的情况下移动left和right指针,和大了就把right变小(左移),小了就把left变大(右移)
最后找到了符合的值后,将结果压入result数组,然后再对left和right可能重复的情况进行排除,例如:
-2,-1,-1,-1,2,2,2,2这种情况
class Solution {
public:vector<vector<int>> threeSum(vector<int>& nums) {vector<vector<int>> result;sort(nums.begin(), nums.end());//sort默认是从小到大排序for(int i = 0; i < nums.size(); i++){if(nums[i]>0)return result;//如果最小的数已经大于0,那么不可能等于0//这里需要排除重复的三元组而不是重复的元素,所以找的是i是否已经等于过某个数//如果使用nums[i]==nums[i+1],那么会把left=i+1时满足三数和等于0的情况漏掉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[right]+nums[left]<0){left++;}else{result.push_back(vector<int>{nums[i], nums[left], nums[right]});while(right>left&&nums[right]==nums[right-1]){right--;}while(right>left&&nums[left]==nums[left+1]){left++;}right--;left++;}} }return result;}
};
四数之和
代码随想录
18. 四数之和 - 力扣(LeetCode)
这道题跟三数之和一个思路,只是外层多一层循环,以此类推五数之和、六数之和都是这样做
但是需要注意的是,这里的target不是固定的,所以剪枝的时候要确定值大于等于0
v1.0:
class Solution {
public:vector<vector<int>> fourSum(vector<int>& nums, int target) {vector<vector<int>> result;sort(nums.begin(), nums.end());for(int i = 0; i< nums.size();i++){//不要判断nums[k] > target 就返回了,//三数之和 可以通过 nums[i] > 0 就返回了,因为 0 已经是确定的数了//四数之和target是任意值。//比如:数组是[-4, -3, -2, -1],target是-10,不能因为-4 > -10而跳过。if(nums[i]>target&&nums[i]>=0){return result;}if(i>0 && nums[i]==nums[i-1]){continue;}for(int j = i+1; j < nums.size();j++){if(nums[j] + nums[i]>target&&nums[j] + nums[i]>=0){break;//这里不能直接return result,因为外层还有循环!}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{result.push_back(vector<int>{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 result;}
};