题目描述
长 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
题目分析
首先需要将蚂蚁碰面后会掉头进行转化,转化成相遇后不掉头而是直接穿过。
因为两只未感冒的蚂蚁相遇后仍不感冒,且速度不变可以看成蚂蚁方向未变,两只中有蚂蚁感冒了相遇后两只蚂蚁都感冒,仍可以看成是每只蚂蚁的方向未变。
假设第一只感冒蚂蚁向右走
则第一批感冒的蚂蚁就是该蚂蚁右边向左走的蚂蚁,数量为cnt。
第二批是右边被感染的蚂蚁去感染左边向右走的蚂蚁,第二批存在的前提是cnt > 0。
初始感冒蚂蚁向左同理可得。
C++代码实现
#include <iostream>
#include <cmath>
using namespace std;
const int N = 110;
int num[N];
int main() {
int n;
cin>>n;
for(int i = 0; i < n; ++i) cin>>num[i];
int key = abs(num[0]);
int left = 0, right = 0;//left:记录num[0]左边向右的蚂蚁数,right: 记录num[0]右边向左的蚂蚁数
for(int i = 1; i < n; ++i) {
if(key > abs(num[i]) && num[i] > 0) left++;
if(key < abs(num[i]) && num[i] < 0) right++;
}
if((num[0]>0 && right == 0) || (num[0]<0 && left == 0) ) cout<<1<<endl;
else cout<<left+right+1<<endl;
return 0;
}