2014第五届蓝桥杯预赛试题本科c++蚂蚁搬家

标题：蚂蚁感冒

    长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左，有的朝右。 

    每只蚂蚁都只能沿着杆子向前爬，速度是1厘米/秒。

    当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。

    这些蚂蚁中，有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。

    请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

【数据格式】

    第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。

    接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。

    要求输出1个整数，表示最后感冒蚂蚁的数目。

例如，输入：
3
5 -2 8
程序应输出：
1

再例如，输入：
5
-10 8 -20 12 25
程序应输出：
3

资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。

注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>， 不能通过工程设置而省略常用头文件。

提交时，注意选择所期望的编译器类型。

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

#define length 100
#define max 50
int f(int *a, int n)
{
	int count = 0;

	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
	}	
	if (a[0] < 0)			 //首先考虑的是，第一个感冒的向左爬
	{
		int flag = 0;		 //标记左边是否有向右爬的蚂蚁
		for (int i = 1; i < n; i++)
		{					 //如果有在第一个蚂蚁的左边的，并且向右爬的
			if (a[i] < abs(a[0]) && a[i]>0) 
			{
				count++;      
				flag = 1;    //左边有向右爬的蚂蚁之后再进行下面的判定
			}								
		}
		if (flag)
		{				    //在感染蚂蚁右边向左爬的
			for (int i = 1; i<n; i++)
			{
				if (a[i] < 0 && a[i]<a[0])
				{
					count++;
				}
			}
		}
	}
	else if (a[0] > 0)   // 如果第一个感冒的向右爬
	{
		int flag = 0;    //标记右边是否有向左爬的蚂蚁
		for (int i = 1; i < n; i++)
		{				 //如果有在第一个蚂蚁右边的，并且向左爬的
			if (abs(a[i]) > abs(a[0]) && a[i] < 0)  
			{
				count++;
				flag = 1;
			}
		}
		if (flag)
		{
			for (int i = 1; i < n; i++)
			{		 //那么如果左边的蚂蚁向右爬的	
				if (a[i] < a[0] && a[i] > 0) 
					count++;
			}
		}
	}
	return count+1;
}

int main(void)
{
	int a[max];
	int n;
	cin >> n;
	cout << f(a, n) << endl;
	return 0;
}