长 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
做法:首先需要一个思想,我们可以认为蚂蚁碰面后就是直接穿过,这样就比较方便计算,不用去模拟。
这样我们可以先不管感冒蚂蚁的方向,如果在它右边,往右边走,那么速度一样,一定追不上,所以只要统计在它右边往左边走的。这样计算出左和右的蚂蚁,但是需要对特殊情况进行处理。
#include <iostream>
#include <cmath>
using namespace std;
int n;
int main(){
cin >> n;
int pivot, left = 0, right = 0;
cin >> pivot;
for(int i = 1; i < n; i++){
int x;
cin >> x;
if(abs(x) < abs(pivot) && x > 0) right++;
if(abs(x) > abs(pivot) && x < 0) left++;
}
if((pivot < 0 && right == 0)|| pivot > 0 && left == 0){
cout << "1";
}else{
cout << left + right + 1;
}
return 0;
}