[LeetCode]Remove Element

题目描述

Given an array and a value, remove all instances of that value in place and return the new length.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

从一个数组中移除所有的指定元素并返回新的长度

解题思路

通过上图我们来分析，需要删除的是元素2.

我们将初始的len = A.Length；每次遇到2，将还未被交换的新数组尾部的元素覆盖到当前位置，然后再从当前位置继续遍历，直到遍历完长度为len的数组；

最终返回新数组长度len。

代码

public int removeElement(int[] A, int elem) {
		int len = A.length;
		
		for(int i = 0;i < len;){
			if(elem==A[i]){
				A[i] = A[len-1];
				len--;
			} else {
				i++;
			}
		}
		
		return len;
	}