算法训练 删除数组零元素

从键盘读入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<cstring>
#include<algorithm>
using namespace std;


int CompactIntegers(int *a,int count){
    for(int i=0;i<count;i++){
        if(a[i]==0){
            for(int j=i;j<count-1;j++){
                a[j]=a[j+1];
            }
            count=count-1;
            i=i-1;
        }
    }

    return count;
}

int main(){
    int n;
    cin>>n;
    int i;
    int *a=new int[n];
    for(i=0;i<n;i++){
        cin>>a[i];
    }
    int sum=CompactIntegers(a,n);
    cout<<sum<<endl;
    for(i=0;i<sum;i++){
        cout<<a[i]<<" ";
    }
}