问题 1454: [蓝桥杯][历届试题]蚂蚁感冒

题目描述

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

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

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

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

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

输入

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

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

输出

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

样例输入

5

-10 8 -20 12 25

样例输出

3

暴力 超时

发现一个规律

既然是一个会传染 那么 肯定是2边向中间跑的才会 传染

如图所示 红点为传染源 蓝点为其他正常蚂蚁



那么不管 传染源初始位置是向左还是向右， 那么在（存在有向右侧行驶的蚂蚁的情况下）传染源左侧的蚂蚁如果向右走就会传染， 右侧同理

所以 只要计算 左侧向右走的 和 右侧向左走的蚂蚁数量加1 即为结果

但有一种特殊情况

即 左侧都向左走 右侧都向右走 ，那么结果 即只有 传染源蚂蚁 一个

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include <bits/stdc++.h>
using namespace std;
bool cmp(int x, int y)
{
return abs(x)<abs(y);
}
int main()
{
int n;
int a[1003];
scanf("%d",&n);
for (int i=0; i<n; i++)
{
scanf("%d",&a[i]);
}
int s=a[0];
sort(a, a+n, cmp);
int t;
for (int i=0; i<n; i++)
{
if (a[i]==s)
{
t = i;
break;
}
}
int num1=0, num2=0;
for (int i=t-1; i>=0; i--)
{
if (a[i]>0)
{
num1++;
}
}
for (int i=t+1; i<n; i++)
{
if (a[i]<0)
{
num2++;
}
}
if (a[t]<0&&num1==0||a[t]>0&&num2==0)
printf ("1\n");
else
printf ("%d\n",num1+num2+1);
return 0;
}