【面試題】將一個無序整形數組的全部0移到末尾，並保持其它非0元素的相對順序不變

問題描述： 將一個無序整形數組的全部0移到末尾，並保持其它非0元素的相對順序不變。如：a={3,0,1,0,0,5,2,0,4,0}，輸出：{3,1,5,2,4,0,0,0,0,0}。
解決思路： 在Java中，一般數組是不可以直接增刪元素的，可以轉換成ArrayList再進行增刪操作。而在Python中，可以將np.array數組轉換成list，再進行增刪操作。先逐個遍歷數組元素，判斷是否爲0，爲0則從ArrayList/list中移出，並用一個zero_list保存這些0，最後把兩個列表合併。
Java代碼如下：

import java.util.*;
import java.util.stream.Collectors;

public class Main {
    public static void main(String[] args) {
        int[] a = new int[]{3,0,1,0,0,5,2,0,4,0};
        System.out.print("原數組：");
        System.out.println(Arrays.toString(a));
        a = moveZeroToEnd(a);
        System.out.print("處理後的數組：");
        System.out.println(Arrays.toString(a));
    }

    public static int[] moveZeroToEnd(int[] arr) {
        // 將數組轉換成list
        List<Integer> arr_list = Arrays.stream(arr).boxed().collect(Collectors.toList());
        List<Integer> zero_list = new ArrayList<>();
        for(int i = 0; i < arr_list.size();) {
            // 如果元素爲0，則把它添加到zero，並把它從list中移出出來
            if(arr_list.get(i) == 0) {
                zero_list.add(0);
                arr_list.remove(i);

            }
            else { //否則，跳過，再判斷下一個元素
                i++;
            }
        }
        arr_list.addAll(zero_list); //合併兩個list
        // 將list轉換成數組
        int[] res = arr_list.stream().mapToInt(Integer::valueOf).toArray();
        return res;
    }
}

執行結果：

原數組：[3, 0, 1, 0, 0, 5, 2, 0, 4, 0]
處理後的數組：[3, 1, 5, 2, 4, 0, 0, 0, 0, 0]

Python代碼如下：

import numpy as np

def move_zero_to_end(arr):
    left_list = [] # 保存去掉0之後的list
    zero_list = [] # 保存提取出來的全爲0的list
    for i in arr:
        if i == 0:
            zero_list.append(i)
        else:
            left_list.append(i)
    return np.array(left_list + zero_list)

a = np.array([3,0,1,0,0,5,2,0,4,0])
print('原數組：')
print(a)
after_arr = move_zero_to_end(a)
print('處理後的數組：')
print(after_arr)

執行結果：

原數組：[3 0 1 0 0 5 2 0 4 0]
處理後的數組：[3 1 5 2 4 0 0 0 0 0]

注意：Python中的數組是通過np.array定義的，而直接定義a = [3,0,1,0,0,5,2,0,4,0]是list，而不是數組。