蓝桥杯 蚂蚁感冒

蚂蚁感冒

题目描述

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

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

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

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

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

题目解析

我们可以简单画一下图，如上所示。有一只蚂蚁感冒了，现在它可以向左或向右走，我们分这两种情况讨论。

• 感冒的蚂蚁向左走

如果感冒的蚂蚁向左走，那么在该蚂蚁后面，如果有蚂蚁向前走，它肯定会被感染；然后判断该蚂蚁是否会转头（只有该蚂蚁感染其他蚂蚁后才会掉头），如果掉头，那么该蚂蚁后面向左走的蚂蚁都会感染。

•   感冒的蚂蚁向右走

向右走的情况同向左走一样，思路一样。

代码如下

#include <iostream>
using namespace std;
int n;
int arr[100];
int ans = 1;
int main() {
    cin >> n;
    for(int i = 1; i <= n; i ++){
        cin >> arr[i];
    }

    if(arr[1] > 0){ // 蚂蚁先向右走
        for(int i = 1; i <= n; i ++){
            //该蚂蚁后面有蚂蚁向左走
            if(arr[i] < 0 && abs(arr[i]) > arr[1] ){
                ans++;
            }
        }
//        如果蚂蚁掉头
        if(ans != 0){
            //该蚂蚁前面有蚂蚁向右走
            for(int i = 1; i <= n; i ++){
                if(arr[i] > 0 && arr[i] < arr[1]){
                    ans ++;
                }
            }
        }
    }else{//蚂蚁先向左走
        for(int i = 1; i <= n; i ++){
            if(arr[i] > 0 && abs(arr[1]) > arr[i]){
                ans ++;
            }
        }
        if(ans != 0){
            for(int i = 1; i <= n; i ++){
                if(arr[i] < 0 && abs(arr[1]) < abs(arr[i])){
                    ans ++;
                }
            }
        }
    }
    cout << ans << endl;
    return 0;
}