移除元素remove-element

移除元素remove-element

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1:

给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。

示例 2:

给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。

方法一：双指针

package remove_element;

public class Solution {
	
    public int removeElement(int[] nums, int val) {
	    if (nums.length == 0) return 0;
	    int j = 0;
	    for (int i = 0; i < nums.length; i++) {
			if(nums[i]!=val) {
				nums[j]=nums[i];
				j++;
			}
		}
	    return j;
    }
    
	//打印数组
	private static void printArry(int len,int[] nums) {
		System.out.println();
		String ss = "[";
		for (int i = 0; i < len; i++) {
			if(i+1==len) {
				ss+=nums[i];
			}else {
				ss+=nums[i]+",";
			}
		}
		ss+="]";
		System.out.println(ss);
	}
	
    public static void main(String[] args) {
		//int[] nums = {1,1,2};
		//int[] nums = {0,1,2,2,3,0,4,2};
		int[] nums = {0,1,2,2,3,3,2};
		Solution solu = new Solution();
		printArry(nums.length,nums);
		int len = solu.removeElement(nums,2);
		printArry(len,nums);
	}
}

方法二：双指针 —— 当要删除的元素很少时

当我们遇到 nums[i] = val时，我们可以将当前元素与最后一个元素进行交换，并释放最后一个元素。这实际上使数组的大小减少了 1。

public int removeElement(int[] nums, int val) {
    int i = 0;
    int n = nums.length;
    while (i < n) {
        if (nums[i] == val) {
            nums[i] = nums[n - 1];
            // reduce array size by one
            n--;
        } else {
            i++;
        }
    }
    return n;
}

Glang

双指针

func removeElement(nums []int, val int) int {
    left := 0
    for _, v := range nums { //v 即nums[right]
        if  v != val {
            nums[left] = v
            left++
        }
    }
    return left
}

双指针优化, 当要删除的元素很少时

func removeElement(nums []int, val int) int {
    left, right := 0, len(nums)
    for left < right {
        if nums[left] == val {
            nums[left] = nums[right-1]
            right--
        } else {
            left++
        }
    }
    return left
}