【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.

 


题解:用一个len变量记录当前数组的长度。每次遇到元素elem,就把len-1指着的元素换到elem元素所在的地方,然后把len--。要注意的地方就是elem在数组结尾的时候,len减一减到和i相等的时候,就不能再减了,应该结束遍历,返回len的值。

代码如下:

 1 class Solution {

 2 public:

 3     int removeElement(int A[], int n, int elem) {

 4         int len = n;

 5         for(int i = 0;i < len;i ++){

 6             if(A[i] == elem){

 7                 while(A[len - 1] == elem && len > i)

 8                     len--;

 9                 A[i] = A[len-1];

10                 if(len > i)

11                     len --;

12             }

13         }

14         /*for(int i= 0;i < len;i++)

15             cout <<A[i]<<endl;*/

16         return len;

17     }

18 };

 还有一种正序的想法,就是设置一个puthere指针,每次发现和elem不想等的元素,就放到puthere所指向的位置;

java代码如下:

 1 public class Solution {

 2     public int removeElement(int[] A, int elem) {

 3         int puthere = 0;

 4         int n = A.length;

 5         for(int i = 0;i < A.length;i++){

 6             if(A[i] != elem){

 7                 A[puthere] = A[i];

 8                 puthere++;

 9             }

10             else {

11                 n--;

12             }

13         }

14         return n;

15     }

16 }

 

你可能感兴趣的:(LeetCode)