题目描述
长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
思路
题目最重要的一个点就是当两只蚂蚁相撞的时候,我们可以把相撞后掉头(不是那个掉头,是调转方向)看成是两只蚂蚁穿过对方,具体解释如下:
上述只画了第二个样例的四个数据。
首先10是感冒的蚂蚁,在它左边有个8是和它方向相反的,它们会在9的时候像状态,然后将感冒传染给8,此时8和10掉头,但是其实这里两只蚂蚁都感冒了,那我们再去区分是哪一只蚂蚁已经不重要了,因为只要感冒就会传染(大家多注意身体),所以我们将掉头的10看成穿过去的8,将掉头的8看成穿过去的10
对最后传染感冒的结果是没有任何影响的,所以这个题目的核心就来了,就是将掉头看成穿过就可以了,同理两只没有感冒的蚂蚁相撞也是一样的(可以自行模拟一下上面的12和20),两只感冒的蚂蚁相撞也是一样的。那么我们在统计的时候,只需要将其分为两部分来看待就行,分别是感冒的蚂蚁的左边和右边,在左边,假如正常蚂蚁是朝右的,那么就有可能感冒,统计下来,在右边,如果正常蚂蚁是朝左的,那么也有可能感冒,统计下来,为什么是可能呢,因为可能出现这样的情况:
10还是感冒的蚂蚁,就是方向全部朝右,这样就只有一个感冒的蚂蚁,在这种情况下,不会出现感染的情况,但是呢,在我们感冒蚂蚁的右边,是不会又蚂蚁被统计进去的,所以右边是0,同理全部朝左也是,只需要判断一下就行。
代码
#include <iostream>
#include <cmath>
using namespace std;
int n;
int ill;
int main()
{
cin >> n;
cin >> ill;
int l = 0, r = 0;
for ( int i = 0; i < n - 1; i ++ )
{
int x;
cin >> x;
if ( abs(x) < abs(ill) && x > 0 ) l ++;
if ( abs(x) > abs(ill) && x < 0 ) r ++;
}
if ( ill < 0 && !l || ill > 0 && !r ) cout << 1 << endl;
else cout << (l + r + 1) << endl;
return 0;
}