蓝桥杯 算法训练 ALGO-79 删除数组零元素

算法训练 删除数组零元素
时间限制：1.0s 内存限制：512.0MB
从键盘读入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

分析：本题规定了子函数的返回值，函数名，参数以及具体操作，剩下的就是我们去定义这个函数并且写操作的具体代码了。
要求删除数组中0元素，其后元素向数组首端移动，这个很简单。
值得注意的是，下一次判断的时候，数组长度要减一，循环索引 $i$也要减一。
因为0元素的下一个元素前移了，如果 $i$不减一，继续++循环，会导致0元素的下一个元素没有检索到，如果下一位依然是0，就会导致错误。代码如下：

#include <iostream>
using namespace std;

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

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