【Leetcode】移除元素

Leetcode

题目要求

给定一个数组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，2
注意这五个元素可为任意顺序
不需要考虑数组中超出新长度后面的元素

核心思想

采用双指针的方法，很有代表性。
将值为val的数据全部移到数组右边，并用count记录移动次数。

具体代码实现如下

public class Solution {
	
	public int removeElement(int[] nums,int val){
		
		
		int begin = 0;
		int end = nums.length-1;
		int count = 0;
		
		while(begin <= end){
			
			while(nums[begin] != val && begin < nums.length){
				begin++;
			}
			while(nums[end] == val && end >= 0){
				end--;
				count++;
			}
			if(begin >= end){
				break;
			}
			
			//交换元素
			int temp = nums[begin];
			nums[begin] = nums[end];
			nums[end] = temp;

		}		
		//返回剔除val的元素个数
		return nums.length - count;
	}
	
	public static void main(String[] args) {
		
		Solution sl = new Solution();
		int nums[] = {0,1,2,2,3,0,4,2};
		System.out.println(sl.removeElement(nums,2));
	}
}