题目描述

长 100厘米的细长直杆子上有 n 只蚂蚁。它们的头有的朝左，有的朝右。

每只蚂蚁都只能沿着杆子向前爬，速度是 1 厘米 / 秒。

当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。

这些蚂蚁中，有 1 只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。

请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

输入格式

第一行输入一个整数 n(1<n<50) 表示蚂蚁的总数。

接着的一行是 n 个用空格分开的整数 Xi(−100<Xi<100)，Xi​ 的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现 0 值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。

输出格式

要求输出 1个整数，表示最后感冒蚂蚁的数目。

输入输出样例

输入 #1

3
5 -2 8

输出 #1

1

输入 #2

5
-10 8 -20 12 25

输出 #2

3

思路：

可知，不管第一只蚂蚁开始往哪边走，只要第一只蚂蚁左边的蚂蚁向右走就一定会感冒， 第一只蚂蚁右边的蚂蚁向左走就一定会感冒。所以只要出左边和右边符合要求的蚂蚁数，ans=left+right+1;

但是会有特殊情况1.当第一只蚂蚁朝左走，左边没有朝右走的蚂蚁 ，此时感冒的就只能是自己，其他的不会被传染，同理，2.当第一只蚂蚁朝右走，右边没有朝左走的蚂蚁 ，此时感冒的就只能是自己，其他的不会被传染。

代码：

#include <bits/stdc++.h>
#define ll long long
#define endl "\n"
using namespace std;
int main()
{
	int n,ans=0,left=0,right=0;
	cin>>n;
	vector<int> v(n+5,0);
	for(int i=0;i<n;i++) cin>>v[i];
	for(int i=1;i<n;i++)
	{
//		if(v[0]<0&&(abs(v[i])<abs(v[0]))&&v[i]>0)
//			left++;
//		else if(v[0]<0&&(abs(v[i])>abs(v[0]))&&v[i]<0) 
//			right++;
//		if(v[0]>0&&(abs(v[i])<abs(v[0]))&&v[i]>0)
//			left++;
//		else if(v[0]>0&&(abs(v[i])>abs(v[0]))&&v[i]<0) 
//			right++;

		if((abs(v[i])<abs(v[0]))&&v[i]>0)
			left++;
		else if((abs(v[i])>abs(v[0]))&&v[i]<0) 
			right++;
	}
	if(left==0&&v[0]<0) { //朝左走，左边没有朝右走的蚂蚁 
		cout<<1<<endl;return 0;
	}else if(right==0&&v[0]>0) {//朝右走，右边没有朝左走的蚂
		cout<<1<<endl;return 0;
	}
	ans=1+left+right;
	cout<<ans<<endl;
	return 0;	
} 

纯享简洁版：

#include <bits/stdc++.h>
#define ll long long
#define endl "\n"
using namespace std;
int main()
{
	int n,left=0,right=0;
	cin>>n;
	vector<int> v(n+5,0);
	for(int i=0;i<n;i++) cin>>v[i];
	for(int i=1;i<n;i++)
	{
		if((abs(v[i])<abs(v[0]))&&v[i]>0)	left++;
		else if((abs(v[i])>abs(v[0]))&&v[i]<0)	right++;
	}
	if(left==0&&v[0]<0) { 
		cout<<1<<endl;return 0;
	}else if(right==0&&v[0]>0) {
		cout<<1<<endl;return 0;
	}
	cout<<1+left+right<<endl;
	return 0;	
}