蚂蚁感冒
长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左,有的朝右。 每只蚂蚁都只能沿着杆子向前爬,速度是1厘米/秒。当两只蚂蚁碰面时,它们会同时掉头往相反的方向爬行。这些蚂蚁中,有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时,会把感冒传染给碰到的蚂蚁。请你计算,当所有蚂蚁都爬离杆子时,有多少只蚂蚁患上了感冒。
【数据格式】
第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。
接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值,表示蚂蚁离开杆子左边端点的距离。
正值表示头朝右,负值表示头朝左,数据中不会出现0值,也不会出现两只蚂蚁占用同一位置。
其中,第一个数据代表的蚂蚁感冒了。要求输出1个整数,表示最后感冒蚂蚁的数目。
例如,
输入:
35 -2 8
程序应输出:
1
输入:
5-10 8 -20 12 25
程序应输出:
3
思路:蚂蚁相碰的后,可以理解为蚂蚁互相穿过身体继续向原方向继续前进,因为如果相碰的蚂蚁如果有一只感冒,结果两只都会感冒,就变成互相穿过身体的两只感冒的蚂蚁,如果两只都没有感冒,其结果任然是两只没有感冒的蚂蚁。处理就遍历数组,用位置和方向判断是否会相碰,相碰加一,同时需要判断会不会有反方向,任然需要判断相碰。
#include <cstdio> #include <string.h> #include <iostream> #include <algorithm> #include <sstream> #include <math.h> using namespace std; const int inf=0x7fffffff; const long long mod=1e9+7; const double PI=acos(-1); bool vis[105]; int a[105]; int n,m,k; long long ans; int main() { cin>>n; for(int i=0;i<n;i++){ cin>>a[i]; } int t=a[0]; //t表示感冒蚂蚁的位置,正负代表方向 if(t<0){ //初始感冒蚂蚁为负方向 ans++; //初始蚂蚁感冒 for(int i=0;i<n;i++){ if(abs(a[i])<-t&&a[i]>0){ //遍历数组,每次碰到感冒蚂蚁左边 且反向的蚂蚁 都会被感染 ans++; } } if(ans>1){ //代表 初始感冒蚂蚁左边有反向蚂蚁 会转向 for(int i=0;i<n;i++){ //转向后,每次碰到初始蚂蚁的右边的 且同向的蚂蚁 都会被传染 if(abs(a[i])>-t&&a[i]<0){ ans++; } } } } else{ //初始方向为正的同理 ans++; for(int i=0;i<n;i++){ if(abs(a[i])>t&&a[i]<0){ ans++; } } if(ans>1){ for(int i=0;i<n;i++){ if(abs(a[i])<t&&a[i]>0){ ans++; } } } } printf("%d",ans); return 0; }