2.1.11 Remove Element

Link: https://oj.leetcode.com/submissions/detail/9751364/

這題應該很簡單,但沒過。再做。

我的思路:

先對數組排序,然後找到element。//不需要這樣做。

Approach I: 不等就複製。相等跳過。

Time: O(n)

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

Approach II: 相等就從數組最後取元素。(比方法I難理解)。

Time: O(n)

public class Solution {
    public int removeElement(int[] A, int elem) {
        int len = A.length - 1;
        for(int i = 0; i <=len; i++){
            if(A[i] == elem){
                A[i] = A[len];
                len--;//A[len] is considered
                i--;//because we have i++ in for loop
            }
        }
        return len+1;
    }
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章