2014第五届蓝桥杯C++B组第八题：蚂蚁感冒

题目：

蚂蚁感冒 
长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左，有的朝右。 
每只蚂蚁都只能沿着杆子向前爬，速度是1厘米/秒。
当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。
这些蚂蚁中，有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。
请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。
【数据格式】
第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。
接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。
要求输出1个整数，表示最后感冒蚂蚁的数目。
例如，输入：
3
5 -2 8
程序应输出：
1


再例如，输入：
5
-10 8 -20 12 25
程序应输出：
3
资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms
请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。
所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>， 不能通过工程设置而省略常用头文件。
提交时，注意选择所期望的编译器类型。

思路：模拟整个走的过程，一直走直到所有蚂蚁都走下去，每走一步处理下当前状态。

代码：

// 蚂蚁感冒
#include <bits/stdc++.h>
using namespace std;

int N;
int *an;  // 存放蚂蚁的状态(距离和方向) 
long cnt(1);

bool *sick;   // 存放蚂蚁是否感冒 

void init()
{
    cin >> N;
    an = new int[N+1];
    sick = new bool[N+1];
    memset(an,0,sizeof(int)*(N+1));
    memset(sick,0,sizeof(bool)*(N+1));
    sick[1]=true;  // 第一只蚂蚁感冒 
    for(int i=1;i<=N;++i) cin >> an[i];
}

bool over()
{
    for(int i=1;i<=N;++i) {
        if(an[i]!=0 && an[i]!=100) return false;
    }
    return true;
}

// 走完之后处理目前的状态（处理相遇，判断是否传染，掉头） 
void f()
{
    for(int i=1;i<=N-1;++i) {
        if(an[i]==0||an[i]==100) continue;
        for(int j=i+1;j<=N;++j) {
            if(an[j]==0||an[j]==100) continue;
            if(abs(an[i])==abs(an[j])) {
                if(sick[i]||sick[j]) {
                    if(sick[i]+sick[j]==1) cnt += 1;
                    sick[i]=sick[j]=true;
                }
                an[i] *= -1;
                an[j] *= -1;
            }
        }
    } 
}

void print()
{
    for(int i=1;i<=N;++i) cout << an[i] << ends;
    cout << endl;
}

// 一直走下去，直到所有蚂蚁都走下去 
void walk()
{
    while(1) {
        if(over()) return;  
        for(int i=1;i<=N;++i) {
            if(an[i]==0||an[i]==100) continue;
            an[i]+=1;   
        }
        //print();
        f();        
    }

} 



int main()
{
    init();
    walk();
    cout << cnt << endl;    

    return 0;
} 

贴上两位大佬的解法：https://blog.youkuaiyun.com/u014552756/article/details/50574260

https://blog.youkuaiyun.com/qq_34594236/article/details/61210194