题目
思路
“两蚂蚁碰面会掉头,若其中一只蚂蚁感冒了,会把感冒传染给碰到的蚂蚁”,这句话看作是“两蚂蚁碰面会互相穿过,只是把感冒的状态传给了另一只蚂蚁”,因为哪只蚂蚁感冒了并不是题目的重点,重点是有多少蚂蚁感冒了。因此本题可以理解为:感冒的蚂蚁一直往一个方向走,求共感染了几只蚂蚁?
通过观察可以发现,若以第一只感冒蚂蚁的位置来分左右界,
1. 若感冒蚂蚁向右走,由于蚂蚁速度大小相同:
在右界,向右边走的蚂蚁必定不会被感染,向左边走的蚂蚁必定被感染。
在左界,若右界有向左走的蚂蚁,那么左界向右边走的必定被感染;否则不会被感染。向左边走的不会被感染。
2. 若感冒蚂蚁向左走,由于蚂蚁速度大小相同:
在左界,向左边走的蚂蚁必定不会被感染,向右边走的蚂蚁必定被感染。
在右界,若左界有向右走的蚂蚁,那么右界向左边走的必定被感染;否则不会被感染。向右边走的不会被感染。
代码
#include<bits/stdc++.h>
using namespace std;int main()
{int n, x, x1;int rtl = 0, ltr = 0;//右界向左走的蚂蚁;左界向右走的蚂蚁cin >> n;cin >> x;if (x > 0){for (int i = 2; i <= n; i ++){cin >> x1;if (abs(x1) > abs(x) && x1 < 0) rtl ++;if (abs(x1) < abs(x) && x1 > 0) ltr ++;}//"1"表示第一个感冒的蚂蚁int res = 1 + rtl;if (rtl) res += ltr;cout << res;}else{for (int i = 2; i <= n; i ++){cin >> x1;if (abs(x1) < abs(x) && x1 > 0) ltr ++;if (abs(x1) > abs(x) && x1 < 0) rtl ++;}int res = 1 + ltr;if (ltr) res += rtl;cout << res;}return 0;
}
简化版代码
通过上面的代码可以发现,当第一只感冒的蚂蚁向右走时,只有rtl大于0,ltr才能加入到res中;若rtl为0,那么把rtl加入到res中没有意义。当第一只感冒的蚂蚁向左走时,只有ltr大于0,rtl才能加入到res中;若ltr为0,那么把ltr加入到res中没有意义。
#include<bits/stdc++.h>
using namespace std;int main()
{int n, x, x1;cin >> n;cin >> x;//以第一只感冒的蚂蚁为界,l:向左边走的蚂蚁的数量;r:向右边走的蚂蚁的数量int l = 0, r = 0;for (int i = 2; i <= n; i ++){cin >> x1;if (abs(x1) < abs(x) && x1 > 0) r ++;if (abs(x1) > abs(x) && x1 < 0) l ++;}int res = 1;if ((x < 0 && r) || (x > 0 && l)) res += (r + l);cout << res;
}