算法训练 删除数组零元素

算法训练 删除数组零元素

从键盘读入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>
using namespace std;
int CompactIntegers(int arr[],int n) {
int count =n;
for(int i=0;i<n;i++) {
if(arr[i]==0) {
count--;
}
}
return count;
}

int main() {
int n;
cin>>n;
int *a=new int
;
for(int i=0;i<n;i++) {
cin>>a[i];
}

cout<<CompactIntegers(a,n)<<endl;
for(int i=0;i<n;i++) {
if(a[i]!=0)
cout<<a[i]<<" ";
}
return 0;
}