蚂蚁感冒 思维

问题 1454: [蓝桥杯][历届试题]蚂蚁感冒

时间限制: 1Sec 内存限制: 128MB 提交: 1623 解决: 687

题目描述

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

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

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

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

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

输入

第一行输入一个整数n  (1  <   n  <   50),  表示蚂蚁的总数。 

接着的一行是n个用空格分开的整数  Xi  (-100  <   Xi  <   100),  Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数  据代表的蚂蚁感冒了。

输出

要求输出1个整数，表示最后感冒蚂蚁的数目。 

样例输入

5 
-10  8  -20  12  25 

样例输出

3

思路：

蚂蚁碰面后再都掉头反方向走，是不是相当于两只蚂蚁不掉头继续走？因为这两只蚂蚁一样啊，都感冒或者是不感冒！

就相当永不掉头的情况。

只需要特判同一个方向的情况，为1.

然后只要判断a1和ai的情况，就可以了。

代码：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=100+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int a[maxn];
int vis[maxn];
int vis2[maxn];
int main()
{
    int n;
    cin>>n;
    rep(i,1,n)cin>>a[i];
    int ans=0;
    rep(i,1,n-1)
    {
        if(a[i]*a[i+1]>0)
            ans++;
    }
    if(ans==n-1)
        ans=1;
    else
    {
        ans=1;
        rep(i,2,n)
        {
            if(a[i]>0)
            {
                if(a[i]<abs(a[1]))
                    ans++;
            }
            else if(abs(a[i])>abs(a[1]))
                ans++;
        }
    }
    cout<<ans<<endl;
    //5
//20 4 -3 -10 -22
}