用C++實現：刪除數組零元素

問題描述：

從鍵盤讀入n個整數放入數組中，編寫函數CompactIntegers，刪除數組中所有值為0的元素，其后元素向數組首端移動。注意，CompactIntegers函數需要接受數組及其元素個數作為參數，函數返回值應為刪除操作執行后數組的新元素個數。輸出刪除后數組中元素的個數並依次輸出數組元素。

輸入格式：

第一行輸入一個整數，代表數組中有多少個數，第二行輸入數組的元素。

輸出格式：

第一行輸出不為0的元素個數，第二行輸出不為0的元素。

測試樣例1：

輸入：

5

3 4 0 0 2

輸出：

3

3 4 2（3 4 2是以空格隔開的3個數）

測試樣例2：

輸入：

3

0 0 0

輸出：

0

測試樣例3：

輸入：

7

0 0 7 0 0 9 0

輸出：

2

7 9

思路：直接對整個數組進行遍歷，遇到0就把0后面的元素往前移，然后減少數組的元素個數。

代碼如下：

#include<iostream>
using namespace std;
int CompactIntegers(int *arr,int n)
{
    //直接從頭到尾遍歷整個數組，是0就將n減1，同時數組下表往前移
    for (int i = 0; i < n; i++)
    {
        if (arr[i] == 0)
        {
            //將所有元素向前移
            int j = i;
            for (; j < n - 1; j++)
            {
                arr[j] = arr[j + 1];
            }
            //將要判斷的元素的下標前移
            i = i - 1;
            //非0元素個數減1
            n--;
        }
    }
    return n;
}

int main(void)
{
    int n;
    cin >> n;
    int* arr = new int[n];
    for (int i = 0; i < n; i++)
    {
        cin >> arr[i];
    }
    int t = CompactIntegers(arr, n);
    cout << t << endl;
    if (t == 0)
    {
        return 0;
    }
    else
    {
        for (int i = 0; i < t - 1; i++)
        {
            cout << arr[i] << " ";
        }
        cout << arr[t - 1];
    }
    return 0;
}