【蓝桥杯练习--DP】蚂蚁感冒

蚂蚁感冒

蚂蚁不掉头 而是“穿过”遇到的蚂蚁 并且所有遇到的蚂蚁都会感冒

以感冒蚂蚁（向右走）为界

• 右侧
  • 向左走的蚂蚁：一定都会被感染
  • 向右走：一定不会被感染
• 左侧
  • 向左走：一定不会被感染
  • 向右走
    • 右侧有向左走的蚂蚁：一定会被感染（因为有向左移动的感冒蚂蚁）
    • 右侧无向左走的蚂蚁：一定不会被感染

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 55;

int n;
int a[N];

int main()
{
    //输入蚂蚁数据情况
    cin >> n;
    for(int i=0;i<n;i++)
        cin>>a[i];
    
    //只要统计会被感染的蚂蚁数量
    int left=0, right=0;    //分别表示左边向右走的蚂蚁数量 和右边向左走的蚂蚁数量
    for(int i=1; i<n; i++)  //第一个数据是感冒蚂蚁 所以从a[1]开始判断
        if(abs(a[i])<abs(a[0]) && a[i]>0)
            left++;
        else if(abs(a[i])>abs(a[0]) && a[i]<0)
            right++;
    
    
    if(a[0]>0 && right==0 || a[0]<0 && left==0)
        cout << 1 << endl;
    else
        cout << left + right + 1 << endl;
    
    return 0;   //返回必须为0 否则报错“None Zero Exit”
}