蓝桥杯PREV-27 蚂蚁感冒 【思维 / 模拟】

时间限制：1.0s   内存限制：256.0MB

问题描述

　　长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

 解题思路

这题可以一秒一秒的模拟，但是很麻烦。 可以这么做： 假设最初感冒的那只蚂蚁往右走，则他右边被传染的蚂蚁数量就等于他右边往左走的蚂蚁数量（这些蚂蚁并不一定是最初向左走的蚂蚁），如果他右边有向左走的蚂蚁（如果没有，那么被感冒的蚂蚁就只有他自己），那么这只最初感冒的蚂蚁会再一次相遇后往左走，他左边所有被传染的蚂蚁数量就等于他左边所有往右走的蚂蚁的数量。最后加上他自己，就是所有感冒的蚂蚁的数量。

AC代码

#include <bits/stdc++.h>
using namespace std;
struct Ant {
	int x,l;
}a[51];
bool cmp(Ant a1,Ant a2)
{
	return a1.x < a2.x;
}
int main()
{
	int n,t;
	cin >> n;
	for(int i=0;i<n;i++) {
		cin >> t;
		a[i].x = abs(t);
		a[i].l = t/abs(t);	//+left   -right
	}
	int loc = a[0].x;	
	sort(a,a+n,cmp);

	for(int i=0;i<n;i++) {
		if(a[i].x == loc) {
			loc = i;	//loc 现在代-表最初感冒的那只蚂蚁为排好序后的第几只 
			break;
		}
	}
	int tot=0;
	if(a[loc].l == 1) {
		for(int i=loc+1;i<n;i++) 
			if(a[i].l == -1)
				tot++;
		if(tot != 0) {
			for(int i=loc-1;i>=0;i--)
			if(a[i].l == 1)
				tot++;
		}
	}
	else {	//a[loc].l = -1;
		for(int i=0;i<loc;i++)
			if(a[i].l == 1)
				tot++;
		if(tot != 0) {
			for(int i=loc+1;i<n;i++)
				if(a[i].l == -1)
					tot++;
		}
	}
	cout << tot+1 << endl;
	return 0;
}