一、刷题
力扣15,三数之和
核心思想是使用双指针,从数组头开始遍历,后面的区间使用双指针。
十分值得注意的是去重操作。
整体代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
sort(nums.begin(), nums.end());//从小到大排序
//设置三个指针,第一个i在数组头,后两个在i+1位置和数组末尾,判断三个数和的大小并且移动后两个指针
for (int i = 0; i < nums.size(); i++) {
if (nums[i] > 0) {//第一个数大于0则一定无法满足条件
return res;
}
//去重,注意写法,nums[i]与nums[i - 1]进行对比,不能时nums[i]与nums[i + 1]进行对比
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) left++;
else if (nums[i] + nums[left] + nums[right] > 0) right--;
else {
res.push_back(vector<int>{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 res;
}
};
力扣18,四数之和
整体思路和三数之和一样,只是在外面多加了一层循环,有两个容易出错的点:一是在剪枝时一顶要判断大于等于0,因为两个负数相加是会变小的;二是在第二个循环力剪枝时不能直接return res
整体代码
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] > target && nums[i] >= 0){//这里需要注意,nums[i]一定要大于0,因为如果是负数的话,两个负数相加是会更小的
return res;
}
//去重
if (i > 0 && nums[i] == nums[i - 1]) continue;
//下面就是判断三数之和
for (int k = i + 1; k < nums.size(); k++) {
if (nums[i] + nums[k] > target && nums[i] + nums[k] >= 0) {
break;//注意这里是break,不是直接return res
}
if (k > i + 1 && nums[k] == nums[k - 1]) continue;
int left = k + 1;
int right = nums.size() - 1;
while (left < right) {
if (nums[i] + nums[k] > target - nums[left] - nums[right]) right--;
else if (nums[i] + nums[k] < target - nums[left] - nums[right]) left++;
else {
res.push_back(vector<int>{nums[i], nums[k], nums[left], nums[right]});
//对left和right进行去重
while (left < right && nums[left] == nums[left + 1]) left++;
while (left < right && nums[right] == nums[right - 1]) right--;
left++;
right--;
}
}
}
}
return res;
}
};
力扣383,赎金信
使用哈希法,这里使用一个26大小的数组即可,这样可以节省空间,
整体代码
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int record[26] = {0};
if (ransomNote.size() > magazine.size()) return false;
for (int i = 0; i < magazine.size(); i++) {
record[magazine[i] - 'a']++;
}
for (int i = 0; i < ransomNote.size(); i++) {
record[ransomNote[i] - 'a']--;
if (record[ransomNote[i] - 'a'] < 0) return false;
}
return true;
}
};