坠落的蚂蚁【北京大学复试上机题】

题目描述

    一根长度为1米的木棒上有若干只蚂蚁在爬动。它们的速度为每秒一厘米或静止不动，方向只有两种，向左或者向右。如果两只蚂蚁碰头，则它们立即交换速度并继续爬动。三只蚂蚁碰头，则两边的蚂蚁交换速度，中间的蚂蚁仍然静止。如果它们爬到了木棒的边缘（0或100厘米处）则会从木棒上坠落下去。在某一时刻蚂蚁的位置各不相同且均在整数厘米处（即1，2，3，…99厘米），有且只有一只蚂蚁A速度为0，其他蚂蚁均在向左或向右爬动。给出该时刻木棒上的所有蚂蚁位置和初始速度，找出蚂蚁A从此时刻到坠落所需要的时间。

输入描述:

    第一行包含一个整数表示蚂蚁的个数N（2<=N<=99），之后共有N行，每一行描述一只蚂蚁的初始状态。每个初始状态由两个整数组成，中间用空格隔开，第一个数字表示初始位置厘米数P（1<=P<=99），第二个数字表示初始方向,-1表示向左，1表示向右，0表示静止。

输出描述:

    蚂蚁A从开始到坠落的时间。若不会坠落，输出“Cannot fall!”

示例1

输入

复制

4
10 1
90 0
95 -1
98 -1

输出

复制

98

首先想一个问题。如果两只蚂蚁相碰，从微观上看，他们交换速度；但从宏观上看，还是可以看成两只蚂蚁按原速度行驶。所以在蚂蚁A左侧，且向左走的蚂蚁，不管和别的蚂蚁怎么相遇，最后都会掉下去，（可能是别的蚂蚁与他交换速度，替他掉了下去），同理，在蚂蚁A右侧，且向右走的蚂蚁，最后也会掉下去·，这两种情况的蚂蚁可以忽略，宏观上不会对A产生影响。

步骤

1、对蚂蚁位置进行排序

2、找到原来静止的蚂蚁A

3、统计A左边向右走的蚂蚁数l,和右边向左走的蚂蚁数r

4、如果l=r，两边的蚂蚁相互抵消，最终蚂蚁A静止，掉不下去

5、如果l>r，左边r个蚂蚁和右边r个抵消，（从中间一对一对的抵消）最后左边第 l-r  个蚂蚁把速度传给了A，下标为l-r-1(从0开始）A向右走，直到掉下去

6、如果l<r，左边l个蚂蚁和右边l个抵消，（从中间一对一对的抵消）最后右边第r-l个蚂蚁把速度传给了A，下标为2*l(从0开始）A向左走，直到掉下去

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

struct ant{
	int location;
	int director;  //-1,0,1分别表示向左，静止，向右 
}a[101];

bool cmp(ant a1,ant a2){
	return a1.location<a2.location;
}

int main(){
	int n,aim=-1;
	int b[101]; 
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d%d",&a[i].location,&a[i].director);
	}
	sort(a,a+n,cmp);  //题目里边没说按位置顺序输入 
	
	int l=0,r=0,k=0; 
	for(int i=0;i<n;i++){
		if(aim==-1&&a[i].director==1){   //还在蚂蚁A左边，并且速度向右 
		    l++;
			b[k++]=a[i].location;
			
		}
		if(a[i].director==0){   //找到蚂蚁A 
			aim=i;
		}
		else if(aim!=-1&&a[i].director==-1){  //在蚂蚁A右边，并且速度向左 
			r++; 
			b[k++]=a[i].location;
			
		}
	}
	if(l==r){
		printf("Cannot fall!");
	}
	else if(l>r){
		printf("%d",100-(b[l-r-1]));
	}
	else if(r>l){
		printf("%d",b[2*l]);
	}
	return 0;
}