蓝桥杯-蚂蚁感冒-模拟

题目
长 100 厘米的细长直杆子上有 n 只蚂蚁。
它们的头有的朝左，有的朝右。
每只蚂蚁都只能沿着杆子向前爬，速度是 1 厘米/秒。
当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。
这些蚂蚁中，有 1 只蚂蚁感冒了。
并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。
请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

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

输出格式
输出1个整数，表示最后感冒蚂蚁的数目。
数据范围
1<n<50,
0<|Xi|<100
输入样例1：
3
5 -2 8
输出样例1：
1
输入样例2：
5
-10 8 -20 12 25
输出样例2：
3
题解： 这题思维和模拟的做法都可以，但是前者比较巧妙，代码量也小（这里贴上大佬的做法 蚂蚁感冒-思维-经典题）。
我使用的是模拟做法，定义蚂蚁的对象，它们具有移动和掉头的行为，开一个数组存所有蚂蚁。判断两只蚂蚁碰头有两种情况：1、方向相反，坐标相差1，即刚好接触。2：方向相反，坐标相等，处于同一位置。
然后就可以开始模拟，已经爬出范围内的蚂蚁从数组里删掉就好。

#include<bits/stdc++.h>
using namespace std;

struct Ant{
	int pos, dis, ill;
	Ant(){};
	Ant(int p, int d, int i) : pos(p), dis(d), ill(i){};
	void mov(){
		if(dis == 0)
			pos--;
		else
			pos++;
	}
	bool jud(){	//	判断在是否杆子上
		if(pos <= 0 || pos >= 100)
			return true;
		return false;
	}
	void change(){	//	掉头
		if(dis == 0)	dis = 1;
		else	dis = 0;
	}
};
bool cmp(Ant a, Ant b){
	return a.pos < b.pos;
}
int n;
vector<Ant> ant;	//	存放所有在杆子上的蚂蚁

int main(){
	cin >> n;
	int pos;
	for(int i = 0; i < n; ++i){
		cin >> pos;
		ant.push_back(Ant(abs(pos), pos < 0? 0:1, 0));
	}
	ant[0].ill = true;
	sort(ant.begin(), ant.end(), cmp);

	int res = 1;
	while(1){
		for(int i = 0; i < (int)ant.size()-1; ++i){
			if(abs(ant[i].pos - ant[i+1].pos) <= 1 && (ant[i].dis > ant[i+1].dis)){//碰头
				if(ant[i].ill != ant[i+1].ill)
					res++;
				ant[i].change();
				ant[i+1].change();	//	掉头
				if(ant[i].ill || ant[i+1].ill)	//	传染
					ant[i].ill = ant[i+1].ill = true;
			}
		}

		for(auto it = ant.begin(); it != ant.end();){	//	删除出去的蚂蚁，以及让所有蚂蚁移动
			if(it->jud())
				it = ant.erase(it);
			else{
				it->mov();
				 ++it;
			}
		}
		if(ant.size() == 0)
			break;
	}
	cout << res << endl;
	return 0;
}