当前位置:   article > 正文

贪心算法题目总结

贪心算法题目总结

1. 整数替换

看到这道题目,我们首先能想到的方法就应该是递归解法,我们来画个图

此时我们出现了重复的子问题,就可以使用递归,只要我们遇到偶数,直接将n除以2递归下去,如果是奇数,选出加1和减1中最小步数的那个继续递归下去即可,直接看代码:

  1. class Solution {
  2. public:
  3. int integerReplacement(int n) {
  4. if(n == 1)
  5. return 0;
  6. if(n & 1 == 1) // 奇数
  7. return min(integerReplacement(n+1),integerReplacement(n-1)) + 1;
  8. else
  9. return integerReplacement(n / 2) + 1;
  10. }
  11. };

然后我们去运行代码,很遗憾,此时代码没有通过,这是因为的递归过程中出现了大量的重复子问题的,这些重复的子问题其实我们已经计算过了,没必要再计算一次,所以我们可以利用记忆化搜索的备忘录来解决这个问题,此时我们递归的时候,如果此时备忘录立马存在这个值,我们就可以直接返回,如果没有这个值,我们再去递归。

  1. class Solution {
  2. unordered_map<int,int> hash;
  3. public:
  4. int integerReplacement(int n) {
  5. // 先判断是否再在备忘录中
  6. if(hash.count(n))
  7. {
  8. return hash[n];
  9. }
  10. if(n == 1)
  11. {
  12. hash[1] = 0;
  13. return 0;
  14. }
  15. if(n & 1 == 1) // 奇数
  16. {
  17. hash[n] = min(integerReplacement(n+1),integerReplacement(n-1)) + 1;
  18. return hash[n];
  19. }
  20. else
  21. {
  22. hash[n] = integerReplacement(n / 2) + 1;
  23. return hash[n];
  24. }
  25. }
  26. };

但是此时我们发现程序没有通过,通过这个测试案例我们可以知道,当n是2147483647的时候,此时是奇数,会去递归2147483647+1,此时就超过了int的范围。

此时我们需要自己来定义一个dfs函数来解决这个问题。

  1. class Solution {
  2. unordered_map<int,int> hash;
  3. public:
  4. int integerReplacement(int n)
  5. {
  6. return dfs(n);
  7. }
  8. int dfs(long long int n){
  9. // 先判断是否再在备忘录中
  10. if(hash.count(n))
  11. {
  12. return hash[n];
  13. }
  14. if(n == 1)
  15. {
  16. hash[1] = 0;
  17. return 0;
  18. }
  19. if(n & 1 == 1) // 奇数
  20. {
  21. hash[n] = min(dfs(n+1),dfs(n-1)) + 1;
  22. return hash[n];
  23. }
  24. else
  25. {
  26. hash[n] = dfs(n / 2) + 1;
  27. return hash[n];
  28. }
  29. }
  30. };

此时我们的代码就能通过,我们再来写另一种解法。

直接上手代码:

  1. class Solution {
  2. public:
  3. int integerReplacement(int n) {
  4. return dfs(n);
  5. }
  6. int dfs(long long int n)
  7. {
  8. if(n == 1)
  9. return 0;
  10. if(n % 2 == 0) // 偶数
  11. {
  12. return dfs(n/2) + 1;
  13. }
  14. else
  15. {
  16. if(n == 3)
  17. {
  18. return dfs(n-1) + 1;
  19. }
  20. else if(n % 4 == 1)
  21. {
  22. return dfs(n - 1) + 1;
  23. }
  24. else if(n % 4 == 3)
  25. {
  26. return dfs(n + 1) + 1;
  27. }
  28. }
  29. // 这里有返回值是因为要让所有的路径都有返回值
  30. return 0;
  31. }
  32. };

其实我们这里也可以不使用递归来解决这个题目。

  1. class Solution {
  2. public:
  3. int integerReplacement(int n) {
  4. int ret = 0;
  5. while(n > 1)
  6. {
  7. if(n % 2 == 0) // 偶数
  8. {
  9. n /= 2;
  10. ret++;
  11. }
  12. else // 奇数
  13. {
  14. if(n == 3)
  15. {
  16. n = 1;
  17. ret += 2;
  18. }
  19. else if(n % 4 == 1)
  20. {
  21. n -= 1;
  22. n /= 2;
  23. ret += 2;
  24. }
  25. else if(n % 4 == 3)
  26. {
  27. n /= 2; // 这里需要先除2,防止n+1溢出
  28. n += 1;
  29. ret += 2;
  30. }
  31. }
  32. }
  33. return ret;
  34. }
  35. };

2. 俄罗斯套娃信封问题

首先我们看到这个题目,就和我们之前的最长递增子序列是一样的,我们需要对左端点进行排序,然后看是否满足俄罗斯套娃的条件,既然是找最长的套娃个数,此时我们可以使用动态规划来解决这个问题,我们可以假设dp[i]位置表示i位置之前的此时最长的套娃个数,当我们到i的位置的时候,我们需要看看0到i-1位置的最长的套娃个数,如果有一个条件满足俄罗斯套娃的条件,那么最长的套娃个数就可以+1,此时就可以使用动态规划来解决这个问题。

  1. class Solution {
  2. public:
  3. int maxEnvelopes(vector<vector<int>>& envelopes) {
  4. // 动态规划
  5. sort(envelopes.begin(), envelopes.end());
  6. int n = envelopes.size();
  7. vector<int> dp(n, 1);
  8. int ret = 1;
  9. for(int i = 1; i < n; i++)
  10. {
  11. for(int j = 0; j < i; j++)
  12. {
  13. if(envelopes[i][0] > envelopes[j][0] &&
  14. envelopes[i][1] > envelopes[j][1])
  15. {
  16. dp[i] = max(dp[i], dp[j]) + 1;
  17. }
  18. }
  19. ret = max(ret, dp[i]);
  20. }
  21. return ret;
  22. }
  23. };

但是此时我们的代码会超时,所以此时我们就需要来使用另一种策略

  1. class Solution {
  2. public:
  3. int maxEnvelopes(vector<vector<int>>& envelopes) {
  4. // 排序
  5. sort(envelopes.begin(), envelopes.end(), [&](const vector<int>& v1, const vector<int>& v2)
  6. {
  7. // 左端点不同的时候
  8. return v1[0] != v2[0] ? v1[0] < v2[0] : v1[1] > v2[1];
  9. });
  10. vector<int> ret;
  11. ret.push_back(envelopes[0][1]);
  12. for(int i = 1; i < envelopes.size(); i++)
  13. {
  14. int b = envelopes[i][1];
  15. if(b > ret.back())
  16. {
  17. ret.push_back(b);
  18. }
  19. else
  20. {
  21. int left = 0, right = ret.size() - 1;
  22. while(left < right)
  23. {
  24. int mid = (left + right) / 2;
  25. if(ret[mid] >= b) right = mid;
  26. else left = mid + 1;
  27. }
  28. ret[left] = b;
  29. }
  30. }
  31. return ret.size();
  32. }
  33. };

3. 可被三整除的最大和

正难则反: 我们可以先把所有的数累加在⼀起,然后根据累加和的结果,贪心的删除⼀些数。

那么我们该如何寻找到最小的和次小的值呢?当然我们可以进行一下sort排序取出最小的和次小的值,但是我们知道sort的底层使用的是快速排序,时间复杂度是O(N*logN),所以此时并不是最优的选择,我们可以使用O(N)的时间复杂度来解决这个问题:

  1. class Solution {
  2. public:
  3. int maxSumDivThree(vector<int>& nums) {
  4. int x1 = 0x3f3f3f3f3f;
  5. int x2 = 0x3f3f3f3f3f;
  6. int y1 = 0x3f3f3f3f3f;
  7. int y2 = 0x3f3f3f3f3f;
  8. int sum = 0;
  9. // 求出总和,最小和次小的值
  10. for(int i = 0; i < nums.size(); i++)
  11. {
  12. sum += nums[i];
  13. if(nums[i] % 3 == 1) // 判断当值余数是否为1
  14. {
  15. if(nums[i] <= x1)
  16. x2 = x1, x1 = nums[i];
  17. else if(nums[i] > x1 && nums[i] < x2)
  18. x2 = nums[i];
  19. }
  20. else if(nums[i] % 3 == 2) // 判断当值余数是否为2
  21. {
  22. if(nums[i] <= y1)
  23. y2 = y1, y1 = nums[i];
  24. else if(nums[i] > y1 && nums[i] < y2)
  25. y2 = nums[i];
  26. }
  27. }
  28. // 分类讨论
  29. if(sum % 3 == 0) return sum;
  30. else if(sum % 3 == 1) return max(sum - x1, sum - y1 - y2);
  31. else return max(sum - y1, sum - x1 - x2);
  32. }
  33. };

4. 距离相等的条形码

这道题目比较简单,我们只需要每次处理一批相同的数字,往 n 个空里面摆放; 每次摆放的时候,隔一个格子摆放一个数,但是我们需要优先处理出现次数最多的那个数,其余的数处理的顺序就没什么要求了。

  1. class Solution {
  2. public:
  3. vector<int> rearrangeBarcodes(vector<int>& barcodes) {
  4. unordered_map<int, int> hash; // 统计每个数出现的次数
  5. int maxVal = 0; // 最大的值
  6. int maxCount = 0; // 最大值的项数
  7. for(auto& e : barcodes)
  8. {
  9. if(maxCount < ++hash[e])
  10. {
  11. maxCount = hash[e];
  12. maxVal = e;
  13. }
  14. }
  15. vector<int> v(barcodes.size(), 0);
  16. // 先处理出现次数最多的数
  17. int index = 0;
  18. for(int i = 0; i < maxCount; i++)
  19. {
  20. v[index] = maxVal;
  21. index += 2;
  22. }
  23. // 处理剩下的数,不用管顺序,直接防即可
  24. hash.erase(maxVal);
  25. for(auto& [x, y] : hash)
  26. {
  27. for(int i = 0; i < y; i++)
  28. {
  29. // 超过最大位置,需要重新从头开始摆放
  30. if(index >= barcodes.size())
  31. index = 1;
  32. v[index] = x;
  33. index += 2;
  34. }
  35. }
  36. return v;
  37. }
  38. };

5. 重构字符串

我们读完题目会发现这道题目和上一道题目是类似的,只不过上一道题目是重排数字,而这道题目是重排字符串,思路基本上都是一致的,但是上一个题目明确告诉了我们所有数字都是可以重排列的,但是我们这个题目不一定,因此我们需要先判断一下,如果出现最多的字符的个数大于(n + 1) / 2 就无法重排,其他情况下就可以直接重排列,那我们直接上代码啦!

  1. class Solution {
  2. public:
  3. string reorganizeString(string s) {
  4. unordered_map<char, int> hash;
  5. char maxVal = ' ';
  6. int maxCount = 0;
  7. for(auto& x : s)
  8. {
  9. if(maxCount < ++hash[x])
  10. {
  11. maxCount = hash[x];
  12. maxVal = x;
  13. }
  14. }
  15. if(maxCount > ((s.size() + 1) / 2))
  16. return "";
  17. // 先处理出现次数最多的那个字符
  18. string ret;
  19. ret.resize(s.size());
  20. int index = 0;
  21. for(int i = 0; i < maxCount; i++)
  22. {
  23. ret[index] = maxVal;
  24. index += 2;
  25. }
  26. hash.erase(maxVal); // 删除最多元素项的值
  27. // 处理剩余的值
  28. for(auto& [x, y] : hash)
  29. {
  30. for(int i = 0; i < y; i++)
  31. {
  32. if(index >= ret.size()) index = 1;
  33. ret[index] = x;
  34. index += 2;
  35. }
  36. }
  37. return ret;
  38. }
  39. };
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/空白诗007/article/detail/776856
推荐阅读
相关标签
  

闽ICP备14008679号