长 100 厘米的细长直杆子上有 n 只蚂蚁。
它们的头有的朝左,有的朝右。
每只蚂蚁都只能沿着杆子向前爬,速度是 1 厘米/秒。
当两只蚂蚁碰面时,它们会同时掉头往相反的方向爬行。
这些蚂蚁中,有 1 只蚂蚁感冒了。
并且在和其它蚂蚁碰面时,会把感冒传染给碰到的蚂蚁。
请你计算,当所有蚂蚁都爬离杆子时,有多少只蚂蚁患上了感冒。
输入格式
第一行输入一个整数 n, 表示蚂蚁的总数。
接着的一行是 n 个用空格分开的整数 Xi, Xi 的绝对值表示蚂蚁离开杆子左边端点的距离。
正值表示头朝右,负值表示头朝左,数据中不会出现 0 值,也不会出现两只蚂蚁占用同一位置。
其中,第一个数据代表的蚂蚁感冒了。
输出格式
输出1个整数,表示最后感冒蚂蚁的数目。
数据范围
1<n<50
0<|Xi|<100
输入样例1:
3
5 -2 8
输出样例1:
1
输入样例2:
5
-10 8 -20 12 25
输出样例2:
3
思路:
本题最大重点:蚂蚁碰头转换方向可以视为穿过。
想明白这一点以后,我们再来思考如何计算感冒蚂蚁的数量:
第一种情况,如果感冒的蚂蚁往左走,那么如果存在它左边往右走的蚂蚁一定会被传染,那么在感冒蚂蚁右边往左走的蚂蚁也会被传染。如果不存在它左边往右走的蚂蚁,那么只有第一只蚂蚁会感冒,因为所有蚂蚁移动的速度都是一样的。
第二种情况,如果感冒的蚂蚁往右走,那么如果存在它右边往左走的蚂蚁一定会被传染,那么在感冒蚂蚁左边往右走的也会被传染。如果不存在,同理也就只有第一只蚂蚁感冒。
综上,我们每一次都需要统计这只蚂蚁左边往右走的(记为left),和右边往左走的数量(记为rigth),然后再分开讨论:
如果蚂蚁是往右走的,那么如果有右边往左走的(right不为0),如果蚂蚁是往左走的,同理(left不为0) 那么上述两种情况的答案均为 left + right + 1。
如果蚂蚁往右走 ,right为0;或者蚂蚁往左走,left为0,那么答案均为 1 。
思路实现:
1.统计左右蚂蚁的数量
for(int i = 1; i < n; i++)
{
int tmp;
cin >> tmp;
if(tmp > 0 && abs(tmp) < abs(ill)) left++;
if(tmp < 0 && abs(tmp) > abs(ill)) right++;
}
2.计算感冒蚂蚁的数量
if(ill > 0 && !right || ill < 0 && !left)
cout << '1' << endl;
else
cout << left + right + 1 << endl;
完整代码(C++):
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
int main()
{
int n, ill;
cin >> n >> ill;
int left = 0, right = 0;
for(int i = 1; i < n; i++)
{
int tmp;
cin >> tmp;
if(tmp > 0 && abs(tmp) < abs(ill)) left++;
if(tmp < 0 && abs(tmp) > abs(ill)) right++;
}
if(ill > 0 && !right || ill < 0 && !left)
cout << '1' << endl;
else
cout << left + right + 1 << endl;
return 0;
}