AcWing 1211. 蚂蚁感冒 题解

AcWing 1211. 蚂蚁感冒

原题链接:AcWing 1211. 蚂蚁感冒 .

题目描述

长 100 厘米的细长直杆子上有 n 只蚂蚁。

它们的头有的朝左,有的朝右。

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

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

这些蚂蚁中,有 1 只蚂蚁感冒了。

并且在和其它蚂蚁碰面时,会把感冒传染给碰到的蚂蚁。

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

输入格式
第一行输入一个整数 n, 表示蚂蚁的总数。

接着的一行是 n 个用空格分开的整数 Xi, Xi 的绝对值表示蚂蚁离开杆子左边端点的距离。

正值表示头朝右,负值表示头朝左,数据中不会出现 0 值,也不会出现两只蚂蚁占用同一位置。

其中,第一个数据代表的蚂蚁感冒了。

输出格式
输出1个整数,表示最后感冒蚂蚁的数目。

数据范围
1<n<50,
0<|Xi|<100
输入样例1:
3
5 -2 8
输出样例1:
1
输入样例2:
5
-10 8 -20 12 25
输出样例2:
3

题目分析

首先需要将蚂蚁碰面后会掉头进行转化,转化成相遇后不掉头而是直接穿过。
因为两只未感冒的蚂蚁相遇后仍不感冒,且速度不变可以看成蚂蚁方向未变,两只中有蚂蚁感冒了相遇后两只蚂蚁都感冒,仍可以看成是每只蚂蚁的方向未变。

假设第一只感冒蚂蚁向右走
则第一批感冒的蚂蚁就是该蚂蚁右边向左走的蚂蚁,数量为cnt。
第二批是右边被感染的蚂蚁去感染左边向右走的蚂蚁,第二批存在的前提是cnt > 0。
初始感冒蚂蚁向左同理可得。

C++代码实现

#include <iostream>
#include <cmath>
using namespace std;

const int N = 110;
int num[N];

int main() {
    int n;
    cin>>n;
    for(int i = 0; i < n; ++i) cin>>num[i];
    int key = abs(num[0]);
    int left = 0, right = 0;//left:记录num[0]左边向右的蚂蚁数,right: 记录num[0]右边向左的蚂蚁数
    for(int i = 1; i < n; ++i) {
        if(key > abs(num[i]) && num[i] > 0) left++;
        if(key < abs(num[i]) && num[i] < 0) right++;
    }
    
    if((num[0]>0 && right == 0) || (num[0]<0 && left == 0) ) cout<<1<<endl;
    else cout<<left+right+1<<endl;
    
    return 0;
}

猜你喜欢

转载自blog.csdn.net/mwl000000/article/details/108216828
今日推荐