考研数据结构之数组（5.1）——练习题之移动数组所有非零元素到数组前端（C表示）

题目

设数组A[0, .... , n-1]的n个元素中有多个零元素，设计一个算法，将A中所有的非零元素依次移动到A数组的前端。

分析

首先说下自己的思路：

代码写出来，时间复杂度比较大。

再说下参考书上的思路：从左到右扫描整个数组，当发现非零元素时，使其尽可能与靠左边的零元素进行交换。

这个则更加简洁，时间复杂度更小。

代码

我的思路的代码：

/* 计算数组中0的数量 */
/* a[]指的是要被计算的数组；length指的是数组中的元素个数 */ 
int countZero(int a[],int length){
	int j=0;// 0的计数器 
	for(int i=0;i<length;i++){// 循环遍历所有数组元素 
		if(a[i]==0){ // 判断数组中的元素是否等于0 
			j++;// 计数器加1 
		}
	}
	return j;
}

/* 将所有非0元素移动到A数组的前端 */
/* a[]指的是要被移动的数组；length指的是数组中的元素个数 */ 
void moveZero(int a[],int length){
	int m=0;// 计数器，记录循环次数，即要经历几轮移动 
	int j; 
	while(m<length-countZero(a,length)){// 注意循环次数，有几个非0数就需要移动几轮 
		if(a[m]==0){// 判断数组中的元素是否等于0
			// 如果等于0，则将0后面的所有元素都向前移动一个位置 
			for(j=m;j<length-1;j++){
				a[j]=a[j+1];
			}
			a[j]=0;// 然后将数组尾的元素设置为0 
		}else{
			m++;// 计数器加1 
		}
	}
}

参考书思路写成的代码：

/* 参考书思路代码：将所有非0元素移动到A数组的前端 */
/* a[]指的是要被移动的数组；a指的是数组中的元素个数 */
void moveElement(int a[],int n) {// i记录从左向右为0的元素下标；j记录从左向右非0的元素下标 
	int i=-1;// i记录零元素的下标 
	int j=0;// j记录非零元素的下标
	while(j<n){
		// 从左向右寻找非零元素 
		while(a[j]==0){
			j++;
		}
		// 从左向右寻找零元素 
		while(a[i]!=0){
			i++;
		}
		// 交换零元素与非零元素
		if(a[j]!=0&&a[i]==0){
			int temp=a[i];
			a[i]=a[j];
			a[j]=temp;
			i++;
			j++;
		} 
	} 
}

运行结果：

除此之外，还提供一种思路，这种思路更容易理解，分别两根指针i和j指向数组的第一个元素和最后一个元素，i指针从左向右寻找零元素，j指针从右向左寻找非零元素，找到后则交换它们的值。如图：

代码如下：

void moveElement2(int a[], int n) {
	int i=0,j=n-1;
	while(i<j) {
		// 从左向右寻找零元素 
		while(a[i]!=0) {
			i++;
		}
		// 从右向左寻找非零元素 
		while(a[j]==0) {
			j--;
		}
		// 交换零元素与非零元素 
		if(i<j){
			int temp=a[i];
			a[i]=a[j];
			a[j]=temp;
		}
	}
}

完整代码：

#include <stdio.h>
#include <stdlib.h>

/* 打印数组 */
/* a[]指的是要被打印的数组；length指的是数组中的元素个数 */
void printArray(int a[],int length) {
	printf("\n");
	for(int i=0; i<length; i++) {
		printf("%d\t",a[i]);
	}
	printf("\n");
}

/* 计算数组中0的数量 */
/* a[]指的是要被计算的数组；length指的是数组中的元素个数 */
int countZero(int a[],int length) {
	int j=0;// 0的计数器
	for(int i=0; i<length; i++) { // 循环遍历所有数组元素
		if(a[i]==0) { // 判断数组中的元素是否等于0
			j++;// 计数器加1
		}
	}
	return j;
}

/* 作者思路代码：将所有非0元素移动到A数组的前端 */
/* a[]指的是要被移动的数组；length指的是数组中的元素个数 */
void moveZero(int a[],int length) {
	int m=0;// 计数器，记录循环次数，即要经历几轮移动
	int j;
	while(m<length-countZero(a,length)) { // 注意循环次数，有几个非0数就需要移动几轮
		if(a[m]==0) { // 判断数组中的元素是否等于0
			// 如果等于0，则将0后面的所有元素都向前移动一个位置
			for(j=m; j<length-1; j++) {
				a[j]=a[j+1];
			}
			a[j]=0;// 然后将数组尾的元素设置为0
		} else {
			m++;// 计数器加1
		}
	}
}

/* 参考书思路代码：将所有非0元素移动到A数组的前端 */
/* a[]指的是要被移动的数组；a指的是数组中的元素个数 */
void moveElement(int a[],int n) {// i记录从左向右为0的元素下标；j记录从左向右非0的元素下标 
	int i=-1;// i记录零元素的下标 
	int j=0;// j记录非零元素的下标
	while(j<n){
		// 从左向右寻找非零元素 
		while(a[j]==0){
			j++;
		}
		// 从左向右寻找零元素 
		while(a[i]!=0){
			i++;
		}
		// 交换零元素与非零元素
		if(a[j]!=0&&a[i]==0){
			int temp=a[i];
			a[i]=a[j];
			a[j]=temp;
			i++;
			j++;
		} 
	} 
}

int main() {
	int a[]= {0,0,0,0,0,0,0,0,0,1};
	int length=10;

	printArray(a,length);

	moveZero(a,length);
//	moveElement(a,length);

	printArray(a,length);

	return 0;
}