蚂蚁感冒（蓝桥杯）

上题

1032: 蚂蚁感冒 时间限制: 1 Sec 内存限制: 128 MB 提交: 22 解决: 10

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

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

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

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

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

输入

每个测试点（输入文件）存在多组测试数据。

每个测试点的第一行为一个整数Task，表示测试数据的组数。

在一组测试数据中：

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

　　接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100),

Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。

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

样例输入 2 3 5 -2 8 5

-10 8 -20 12 25 样例输出 1 3

分析

第一个输入时感冒的蚂蚁，它有两种方向，向左或向右。

如果它是向左的：

那么对一组输入数据，输入完之后我们可以知道，在0-100这根直杆上，分别分布着一些蚂蚁，每只蚂蚁都有一个朝向。

那么那只感冒的蚂蚁所在直杆的位置就将直杆以它为中心分成左右两部分，

因为感冒的那只是向左的，所以它左边的直杆上的蚂蚁，只要是向右的，那么到最后都会被感冒。此时，只要它的左边的直杆上有一只向右走的蚂蚁，那么它右边的直杆上向左走的蚂蚁，走到最后肯定也会被感冒。如果它左边没有一只向右走的蚂蚁，那么最后只会有它这一只感冒。

如果它是向右走的计算方法同理。

所以走到最后会被感冒的总数就是，对第一个感冒的蚂蚁，在直杆上，它要走的方向的那一边的所有蚂蚁中，向它相反的方向走的蚂蚁的总数，如果总数不为0，就再加上它在直杆的位置的右边的直杆上的蚂蚁中和它的行动方向相同的蚂蚁的总数。最后再加上它自己本身就是answer了

（不得不承认这是复制的）

代码

#include <cstdio>
#include <algorithm>

using namespace std;
struct Node
{
int x;
int dis;
int num;
};
Node a[100];
int comp(Node a1, Node a2)
{
if (a1.x != a2.x)
return a1.x<a2.x;
}
int main(){
int T;
scanf_s("%d", &T);
int n, x;
while (T--){
scanf_s("%d", &n);
for (int i = 0; i<n; i++){
scanf_s("%d", &x);
a[i].dis = x<0 ? -1 : 1;
a[i].x = abs(x);
a[i].num = i + 1;//  用来表示初始蚂蚁的位置
}
sort(a, a + n, comp);  //排序
int tmp = 0, left = 0, right = 0, ans = 0;
for (int i = 0; i<n; i++){  //左边向右走的
if (a[i].num == 1){
tmp = i; break;// 找到第一只感冒的蚂蚁
}
if (a[i].dis == 1)
left++;
}
for (int i = tmp + 1; i<n; i++){ //右边向左走的
if (a[i].dis == -1)
right++;
}
if (a[tmp].dis == 1 && right == 0 || a[tmp].dis == -1 && left == 0)
ans = 1;
else
ans = left + right + 1;
printf("%d\n", ans);
}
return 0;
}