AcWing 1211. 蚂蚁感冒

题目描述

原题链接

分析

思路参考大佬博客
将蚂蚁碰撞后掉头,看做穿过
先考虑一般情况:
以第一只感冒的蚂蚁$a$作为分界线, 不管$a$向左或是向右走
$a$左边向右走的蚂蚁一定会被感染,数量记为$left$
$a$右边向左走的蚂蚁一定会被感染,数量记为$right$
则感染的数量为 $left+right+1$
但是存在一种特殊情况:
$a$左边没有向右走的蚂蚁或者$a$右边没有向左走的蚂蚁
即$left$或者$right$的数量为$0$,则感染的数量为$1$

Y总视频讲解(需要权限)

实现

#include <iostream>
#include <cmath>
using namespace std;
const int N = 59;
int x[N];
int n;
int Left; // left 即 第一只蚂蚁左边向右走的蚂蚁的数量
int Right; // right 即 第一只蚂蚁右边向左走的蚂蚁的数量
int main()
{
    cin >> n;
    for(int i=1; i<=n; i++) cin >> x[i];
    for(int i=2; i<=n; i++)
    {
        if(abs(x[i]) < abs(x[1]) && x[i] > 0) Left ++;
        if(abs(x[i]) > abs(x[1]) && x[i] < 0) Right ++;
    }
    if(Left == 0 || Right == 0) cout << "1" << endl;
    else cout << Left + Right + 1 << endl;
    return 0;
}