蓝桥杯练习系统-删除数组零元素

从键盘读入n个整数放入数组中，编写函数CompactIntegers，删除数组中所有值为0的元素，其后元素向数组首端移动。注意，CompactIntegers函数需要接受数组及其元素个数作为参数，函数返回值应为删除操作执行后数组的新元素个数。输出删除后数组中元素的个数并依次输出数组元素。
样例输入: （输入格式说明：5为输入数据的个数，3 4 0 0 2 是以空格隔开的5个整数）
5
3 4 0 0 2
样例输出:（输出格式说明：3为非零数据的个数，3 4 2 是以空格隔开的3个非零整数）
3
3 4 2
样例输入:
7
0 0 7 0 0 9 0
样例输出:
2
7 9
样例输入:
3
0 0 0
样例输出:
0

#include<iostream>
#include<string>
#include <cstring>
using namespace std;
void Com(int array[], int *n)
{
    int i,j;
    int k = *n;//k为数组中元素个数
    
    for (i = 0; i < k; i ++)
    {
        if (array[i] == 0)
        {
            for (j = i; j < k - 1; j++)
            array[j] = array[j + 1];//覆盖
            array[k] = 0;
            i--;
            k--;
        }
    }
    *n = k;
}

int main()
{
    int n,i;
    cin >> n;
    int count[n];
    for (i = 0; i < n; i ++)
        cin >> count[i];

    Com(count, &n);
    cout << n << endl;
    
    for (i = 0 ; i < n; i ++)
        cout << count[i] << " ";
    
    return 0;
    
}