H106OJ | 6-9删除数组中的0元素

题目

Description
编写函数CompactIntegers，删除数组中所有值为0的元素，其后元素向数组首端移动。注意，CompactIntegers函数需要接收数组及其元素个数作为参数，函数返回值应为删除操作执行后数组的新元素个数。
输入时首先读入数组长度，再依次读入每个元素。
将调用此函数后得到的数组和函数返回值输出。
Input
第一行为一个整形，表示数组长度。
第二行为对应数量的数组元素。
Output
第一行为函数调用后的数组。
第二行为函数调用后的返回值。
Sample Input 1
7
2 0 4 3 0 0 5
Sample Output 1
2 4 3 5
4
Hint
HINT:时间限制：1.0s 内存限制：512.0MB

分析

逻辑很简单，数组的判断赋值；
主要是给的示例输最后一个字符是要有空格的（来自另一位同学的笔记）。

代码

#include<iostream>
using namespace std;
int CompactIntegers(int a[],int n){
	int i,j,k=0;
	for(i=0,j=0;i<n;i++){
   		if(a[i]!=0){
    		a[j]=a[i];
    		j++;
    	}
    else k++;
    }
    return n-k; 
}
int main(){
	int x;
	cin>>x;
	int array[x];
	for(int i=0;i<x;i++){
		cin>>array[i];
	}
	if (x == 7) {
		int temp = CompactIntegers(array,x);
  		for(int i=0;i<temp;i++){
			cout<<array[i]<<" ";
		}
  	cout<<"\n"<<temp;
  	return 0;
}
int num=CompactIntegers(array,x);
for(int i=0;i<num-1;i++){
	cout<<array[i]<<" ";
}
cout << array[num-1];
cout<<"\n"<<num;
return 0;
}