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

时间限制: 1Sec 内存限制: 128MB 提交: 188 解决: 97

题目描述

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

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

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

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

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

输入

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

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

输出

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

样例输入

5 
-10  8  -20  12  25 

样例输出

3

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
int e=0;
struct Node{
    int p;
    int r;
    int flag;
}F[2][100];
int have;
void print(int n)
{
    printf("新的状态：\n");
    for(int i=0;i<n;i++)
    {
        printf("ANT :%d  p=%d  r=%d flag=%d\n",i+1,F[e][i].p,F[e][i].r,F[e][i].flag);
    }
}
int main(void)
{
    int n;
    cin >> n;
    have=n;
    for(int i=0;i<n;i++)
    {
        cin>>F[e][i].p;
        if(F[e][i].p>0) F[e][i].r=1;
        else 
        {
            F[e][i].p=-F[e][i].p;
            F[e][i].r=0;
        }
        
        if(i==0) F[e][i].flag=1;
        else F[e][i].flag=0;
    }
//    print(n); 
    while(have!=0)//如果还有剩余的蚂蚁 
    {
        e=1-e;//新的状态 
        for(int i=0;i<n;i++) //枚举每一只蚂蚁的新位置，包括已经出去的蚂蚁 
        {
            //上一次的状态 ：方向，位置，生病状态 
            int r=F[1-e][i].r;  
            int p=F[1-e][i].p;
            int flag=F[1-e][i].flag;
            if(p<=0 || p>=100) //判断是否出范围 
            {
                F[e][i]=F[1-e][i];
                if(F[e][i].r==0) F[e][i].p--;
                else F[e][i].p++; 
                continue;
            }
            if(r==1)//如果没有出界 
            {
                F[e][i].p=p+1;
                F[e][i].r=1;
                F[e][i].flag=flag;    
                for(int j=0;j<n;j++)//判断是否相撞 
                {
                    if((F[1-e][j].p==p+1 || F[1-e][j].p==p+2) && F[1-e][j].r==0)
                    {
                        if(F[1-e][j].flag==1 || F[1-e][i].flag==1)
                        {
                            F[e][i].flag=1;
                            break;
                        }    
                    }
                }
                if(p+1==100) have--;
            } 
            else if(r==0)
            {
                F[e][i].p=p-1;
                F[e][i].r=0;
                F[e][i].flag=flag;
                for(int j=0;j<n;j++)
                {
                    if((F[1-e][j].p==p-1 && F[1-e][j].r==1)||(F[1-e][j].p==p-2 && F[1-e][j].r==1)) 
                    {
                        if(F[1-e][j].flag==1 || F[1-e][i].flag==1)
                        {    
                            F[e][i].flag=1;
                            break;
                        }
                    }
                }
                if(p-1==0) have--;
            }
        }
        //print(n);
    }
    int cnt=0;
    for(int i=0;i<n;i++)
    {
        if(F[e][i].flag==1) cnt++;
    }
    printf("%d",cnt);
    return 0;    
}