当前位置:   article > 正文

Leetcode Part2

Leetcode Part2

27--移除元素

        最开始想到的是单指针,但是无法处理连续相同的目标值,故想到双指针法,设立首尾指针,先判断数组是否为空,其次判断尾指针所指元素是否为目标值,是则使用vector的内置函数去除尾元素,并让尾指针重新指向末尾元素;否则判断首指针的元素是否为目标值,若不是,使首指针后移;若是,交换首尾指针所指元素,去除尾元素,重新赋值尾指针。在跳出循环后,判断数组大小,若数组只有一个,且元素为目标值,可能原数组所有元素都相等,直接返回0,反之返回数组大小。

  1. class Solution {
  2. public:
  3. int removeElement(vector<int>& nums, int val) {
  4. int start = 0, end = nums.size() - 1;
  5. while(start < end){
  6. if(nums.empty()){
  7. break;
  8. }
  9. if(nums[end] == val){
  10. nums.pop_back();
  11. end = nums.size() - 1;
  12. continue;
  13. }
  14. else{
  15. if(nums[start] == val){
  16. swap(nums[start], nums[end]);
  17. nums.pop_back();
  18. end = nums.size() - 1;
  19. }
  20. else{
  21. start++;
  22. }
  23. }
  24. }
  25. if(start == end && nums[start] == val){
  26. return 0;
  27. }
  28. return nums.size();
  29. }
  30. };

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/一键难忘520/article/detail/933342
推荐阅读
相关标签
  

闽ICP备14008679号