刪除元素

給定一個數組和一個值，在原地刪除與值相同的數字，返回新數組的長度。

元素的順序可以改變，并且對新的數組不會有影響。

樣例

給出一個數組?[0,4,4,0,0,2,4,4]，和值 4

返回 4 并且4個元素的新數組為[0,0,0,2]

?

思路就是把除刪除元素外的元素移到最左邊，r是最右端非被刪除元素，可以在一開始就找到最右端非被刪除元素。用i遍歷數組，找到最左端被刪除元素，把r所指元素賦到i位置即可。代碼里我是先找最左端被刪除元素，然后再找最右端非被刪除元素。時間復雜度是相同的。

?

1 public class Solution { 2 /** 3 *@param A: A list of integers 4 *@param elem: An integer 5 *@return: The new length after remove 6 */ 7 public int removeElement(int[] A, int elem) { 8 int r = A.length - 1; 9 if(r < 0) return 0; 10 int i = 0; 11 12 while(r > i) { 13 if(A[i] == elem) { 14 while(A[r] == elem && r > 0) r--; 15 if(r > i) { 16 A[i] = A[r]; 17 r--; 18 } 19 } 20 i++; 21 } 22 23 if(A[r] == elem) r--; 24 25 return r + 1; 26 } 27 } View Code

?

?

?

轉載于:https://www.cnblogs.com/FJH1994/p/5022094.html

總結

以上是生活随笔為你收集整理的删除元素（LintCode）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。