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

从键盘读入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

思路：
一开始想用C来写，可是发现自己好像不会删除数组元素（报错），算了算了，我还是用C++里面的vector比较好吧。可是，奈何vector我也用的不熟，vector的erase执行之后，it指针会变成野指针，必须要给it重新赋值it=a.erase(it);这是一个坑。后面我才发现执行一次for循环只能删除一个元素，噗，那就只能从头到尾遍历一遍了，加个while(n–)循环，总算AC了。

AC代码：

#include <iostream>
#include <vector>
using namespace std;

vector<int> a;

int CompactIntegers(int n){
	while(n--)//没有循环的话，一次只会删除一个0 
		for(vector<int>::iterator it=a.begin();it!=a.end();it++){ 
			if(*it==0)
				it=a.erase(it);//注意给it指针赋值，防止变成野指针 
			if(it==a.end()) it=a.begin();//防止变成野指针异常
		} 
	n=a.size();
	return n;
} 

int main(){
	int n,num,i,cnt=0;
	cin>>n;
	for(i=0;i<n;i++){
		int m;
		cin>>m;
		a.push_back(m);
		if(m==0) cnt++;
	}
	if(cnt==n){
		cout<<"0";
		return 0;
	}
	num=CompactIntegers(n);
	cout<<num<<"\n";
	if(num==0){
		cout<<"0";
		return 0;
	}
	for(i=0;i<num;i++){
		cout<<a[i]<<" ";
	}
	return 0;
}

后来我发现其实C语言实现也比较简单，就把不是0的元素重新赋给一个新的数组里就好了。害，我竟然没想起来这种。

#include <stdio.h>
int b[10010];
int CompactIntegers(int *a,int n){
	int i,j;
	for(i=j=0;i<n;i++){
		if(a[i])
			b[j++]=a[i];
	}
	return j;
}
int main(){
	int n,i,num,a[10010];
	while(scanf("%d",&n)!=EOF){
		for(i=0;i<n;i++)
			scanf("%d",&a[i]);
		num=CompactIntegers(a,n);
		if(num==0){
			printf("0\n");
			continue;
		}
		printf("%d\n",num);
		printf("%d",b[0]);
		for(i=1;i<num;i++)
			printf(" %d",b[i]);
		printf("\n");
	}
	return 0;
}