![](https://i-blog.csdnimg.cn/blog_migrate/72a800c2c534e3a806a1f3174a81c735.png)
样例输入
3
5 -2 8
5 -2 8
样例输出
1
样例输入
5
-10 8 -20 12 25
-10 8 -20 12 25
样例输出
3
一开始看到题目想的有点细导致反而让题目变复杂的,采用模拟的方法的话,各种掉头也是很难搞
看了网上一些大神的揭发啊发现,掉头本职和穿行是一样的,因为碰头后两只蚂蚁都感冒了,那掉头和穿行的效果是一样的
因此就会有碰到一只蚂蚁后两只蚂蚁开始往两边扫荡,因此除特殊情况外(不会传染别的蚂蚁的情况)
最后感冒的蚂蚁的数量 = 初代(第一只感冒的蚂蚁)左边向右走的蚂蚁数 + 初代右边向左走的蚂蚁的数量
具体一点可以这样想
首先,易知
同方向的蚂蚁由于速度相同的原因是不可能被传染的(被追上)
然后考虑
一只蚂蚁不管它是往左走还是往右走(排除特殊情况)都会和一只蚂蚁碰头,然后初代和第二代蚂蚁就会向两边扫
因此,初代左边向右走的蚂蚁肯定会被初代或是第二代传染,然后第三代的蚂蚁可能传染的蚂蚁数已经在上面算进去了,毕竟已经有两只不同方向的蚂蚁再扫荡了。
而特殊的情况就只有两种
1.初代向左走,而其左边的蚂蚁都往左走
2.初代向右走,而其右边的蚂蚁都向右走
这两种情况包含了,初代左/右走,左/右边没蚂蚁 和 所有蚂蚁同方向走的几中特殊情况
以下为AC代码
#include<iostream>
#include<string>
#include<set>
#include<vector>
#include<algorithm>
#include<fstream>
using namespace std;
bool cmp(int a,int b){
return abs(a)<abs(b);
}
int main(){
//ifstream cin("aaa.txt");
int n;
cin>>n;
int ans = 0;
int cold;
vector<int>v;
for(int i =0;i<n;i++){
int t;
cin>>t;
v.push_back(t);
if(i==0)cold = t;
else{
if((abs(t)<abs(cold)&&t>0)||abs(t)>abs(cold)&&t<0)ans++;//感冒的蚂蚁左边往右走的 + 感冒的蚂蚁右边往左走的
}
}
sort(v.begin(),v.end(),cmp);
int k = 0;
//找到排序后蚂蚁的位置
for(int i = 0;i<v.size();i++){
if(v[i]==cold){
k=i;
break;
}
}
//以下两种为感冒不会被传染的例子,一个就是蚂蚁往右走,但它右边的蚂蚁都往右走
if(cold>0){
for(int i=k+1;i<v.size();i++){
if(v[i]<0){//表明有右边的蚂蚁往左走,则按上面统计的蚂蚁个数为结果
cout<<ans+1;//1表示最开始那只蚂蚁本身
return 0;
}
}
cout<<1;// 往右走的蚂蚁右边的蚂蚁都往右走,因此不传染
return 0;
}
//蚂蚁往左走,但它左边的蚂蚁都往左走
if(cold<0){
for(int i=0;i<k;i++){
if(v[i]>0){//同上
cout<<ans+1;
return 0;
}
}
cout<<1;
return 0;
}
return 0;
}