poj-坠落的蚂蚁

转载自：点击打开链接

描述

一根长度为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!”

样例输入

4
10 1
90 0
95 -1
98 -1

样例输出

98

解题思路：

不好想到要用排序。首先将球的位置排序，设静止的球对应的下标为index，位于该球左边但向右行驶的球的个数为s，同理位于该球右边准备向左行驶的球的个数为t，则有：

如果s=t，那么球将永远不会掉下来

如果s>t，那么球行驶的时间为100-ls[s-t-1]

如果s<t，那么球行驶的时间为rs[s]

（其中ls为index左边但准备向右行驶的球的位置，rs为index右边准备向左行驶的球的位置）

# include<stdio.h>
# include<iostream>
# include<algorithm>
using namespace std;
struct E
{
	int x;
	int dir;
	bool operator < (const E &b) const
	{
		return x<b.x;//升序
	}
}ant[101];

int ls[101];
int rs[101];

int main()
{
	int n,i;
	int index;
	while(scanf("%d",&n)!=EOF)
	{
		for(i=0;i<n;i++)
		{
			scanf("%d%d",&ant[i].x,&ant[i].dir);
		}
		sort(ant,ant+n);

		for(i=0;i<n;i++)
		{
			if(ant[i].dir==0)
			{
				index=i;
				break;
			}
		}

		int s=0;
		for(i=0;i<index;i++)
		{
			if(ant[i].dir==1)
		       ls[s++]=ant[i].x;//左边
		}

		int t=0;
		for(i=index+1;i<n;i++)
		{
			if(ant[i].dir==-1)
			   rs[t++]=ant[i].x;//右边
		}

		if(s==t){
			printf("Cannot fall!\n");
		}
		else if(s>t)
		{
		    printf("%d\n",100-ls[s-t-1]);
		}
		else
		{
		   printf("%d\n",rs[s]);
		}
	}
	return 0;
}