赞
踩
最开始想到的是单指针,但是无法处理连续相同的目标值,故想到双指针法,设立首尾指针,先判断数组是否为空,其次判断尾指针所指元素是否为目标值,是则使用vector的内置函数去除尾元素,并让尾指针重新指向末尾元素;否则判断首指针的元素是否为目标值,若不是,使首指针后移;若是,交换首尾指针所指元素,去除尾元素,重新赋值尾指针。在跳出循环后,判断数组大小,若数组只有一个,且元素为目标值,可能原数组所有元素都相等,直接返回0,反之返回数组大小。
- class Solution {
- public:
- int removeElement(vector<int>& nums, int val) {
- int start = 0, end = nums.size() - 1;
- while(start < end){
- if(nums.empty()){
- break;
- }
- if(nums[end] == val){
- nums.pop_back();
- end = nums.size() - 1;
- continue;
- }
- else{
- if(nums[start] == val){
- swap(nums[start], nums[end]);
- nums.pop_back();
- end = nums.size() - 1;
- }
- else{
- start++;
- }
- }
- }
- if(start == end && nums[start] == val){
- return 0;
- }
- return nums.size();
-
- }
- };
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。