蚂蚁感冒（思维题）

题目
例如这个图红色点代表感冒的蚂蚁，在他左边与他背向的一定不会被感染，在他右边与他方向相反一定被感染，在他右边与他同向的肯定不会感染，在他左边他与同向的会不会感染要看他右边有没有与他行径方向相反的蚂蚁。

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const   int N=55;
    int n; bool cmp(int a,int b)
{
   
    return abs(a)<abs(b);
}
int a[N]; int main()
{
   

    int t;
    int ans=0;
    cin>>n;
    for(int i=0;i<n;i++)
    cin>>a[i];
    t=a[0];
    sort(a,a+n,cmp);
    for(int i=0;i<n;i++) if(a[i]==t) {
   t=i;break;}
    if(a[t]>0)
   {
    for(int i=t+1;i<n;i++)
    if(a[i]<0)  ans++;
    for(int i=0;i<t;i++)
    if(ans&&a[i]>0) ans++; 
 }
 else
 {
   
     for(int i=0;i<t;i++)
     if(a[i]>0) ans++;
     for(int i=t+1;i<n;i++)
     if(ans&&a[i]<0)    ans++;
 }
 printf("%d\n",ans+1);
}