行星碰撞算法

给定一个整数数组，每一个元素，其绝对值表示行星的大小，正负表示行星移动的方向，正表示往右移动，负表示向左移动，他们移动的速度相同。

碰撞规则：1、如果两个行星碰撞，较小的行星会爆炸。2、如果大小相同，则两颗都会爆炸。3、两颗移动方向相同的行星永远不会发生碰撞。

 

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

using namespace std;

int* collision(int* data, int dataSize, int* retSize) {
	int n = 0;
	while (1) {
		int per = 0;
		int next = 1;

		while (next < dataSize) {
			//判断数组左右元素是否异号
			if (data[per] * data[next] < 0) {
				if (data[per] < 0) { //左边的元素小于0，会往右边走不会相撞,per移向next的位置，next向前移动一步
					per = next;
					next++;
					continue;
				}

				if (abs(data[per]) > abs(data[next])) {//左边元素的绝对值大于右边元素，右边元素被置为0
					data[next] = 0;
					n++;//碰撞置0数加1
				}
				else if (abs(data[per]) < abs(data[next])) {//右边元素的绝对值大于左边元素，左边元素被置为0
					data[per] = 0;
					n++;//碰撞置0数加1
				}
				else {//两个元素的绝对值一样大，被同时置为0
					data[per] = 0;
					data[next] = 0;
					n += 2;//碰撞置0数加2
				}
				break;
			}

			//当左右两个元素不异号或相乘为0时
			 if (data[next] == 0) {//如果右边元素为0,则next向右移动一步,per不动
				next++;
			}
			else {//如果左边元素为0或者两元素同号时，per移向next的位置，next向前移动一步
				per = next;
				next++;
			}
		}

		//当next的值大于等于dataSize时，说明没有要碰撞的元素，跳出循环
		if (next >= dataSize) {
			break;
		}
	}

	*retSize = dataSize - n; //*retSize为剩下元素的个数，等于原来个数减去置0个数
	int* retArray = (int*)malloc(*retSize * sizeof(int));//分配一块新的空间用来存储剩余的元素
	for (int i = 0,k=0; i < dataSize; i++) {
		if (data[i]) {//当data的元素不为0的时候将值赋给新的数组
			retArray[k++] = data[i];//后置++,先赋值再执行++操作
		}
	}
	return retArray;
}

int main(void) {
	int array[] = {26,2,-5,-36,20,18,-16,-20,13};
	int count = sizeof(array) / sizeof(array[0]);
	int retSize;

	int *ret=collision(array, count, &retSize);

	for (int i = 0; i < retSize; i++) {
		cout << ret[i] << " ";
	}
	
	return 0;
}