蚂蚁感冒

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 double array[50];//蚂蚁位置 
 4 int array1[50];//表示蚂蚁有没有离开0表示没有1表示有 
 5 int cold[50];//表示蚂蚁有没有感冒0没有1有 
 6 int main()
 7 {
 8      int n;
 9      cin >> n;
10      memset(array,0,sizeof(array)); 
11      memset(array1,0,sizeof(array1)); 
12      memset(cold,0,sizeof(cold)); 
13      int sum=0;
14      cold[0]=1;//第一个感冒了 
15      for(int i=0;i<n;i++){
16          cin >> array[i];
17      }
18      int num=n;
19      while(true){
20          for(int i=0;i<n;i++){
21              if(array1[i]==0){
22              
23                  array[i]+=0.5;//因为速度不会影响蚂蚁感冒数量，不妨设速度为0.5 ，方便计算 
24              
25              } 
26          }
27          for(int i=0;i<n;i++){
28              if(array1[i]==0){//蚂蚁还没离开棍子 
29                  if(array[i]>=100 || array[i]==0){
30                  array1[i]=1;//表示已经离开 
31                  num--;
32              }
33              }
34         //     if(array[i]==array[i+1])
35          } 
36           for(int i=0;i<n-1;i++){
37              if(array1[i]==0 && array1[i]==0){
38                  if(abs(array[i])==abs(array[i+1])){//在树上的蚂蚁相遇， 
39                  array[i]*=(-1);//改变方向 
40                  if(cold[i]==1 || cold[i+1]==1){//如果有一只感冒，则都感冒 
41                      cold[i]=cold[i+1]=1;
42                  }
43              }
44              }
45          } 
46          if(num==0) break;//都离开完 
47      }
48         for(int i=0;i<n;i++){
49             sum+=cold[i];
50         }
51         cout << sum ;
52 }