蚂蚁感冒

最新推荐文章于 2023-02-27 11:50:06 发布

原创最新推荐文章于 2023-02-27 11:50:06 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

算法之路专栏收录该内容

240 篇文章

订阅专栏

本文解析了一道关于两只蚂蚁在直线上相遇的问题，并提供了一个O(n)的算法解决方案。通过模拟两只蚂蚁的移动过程，我们可以得出感冒蚂蚁的数量。代码实现使用了C++，并详细解释了如何计算向左和向右移动的蚂蚁数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

nyoj990 和蓝桥杯都有这道题。

由于数据很水，模拟应该也能过。

主要说一下O(n)算法，两只蚂蚁相遇可以看做互相穿过，那么向右走的蚂蚁数量和向右走的蚂蚁数量都不会变。假设第一只感冒的蚂蚁向右走，那么在它右边向左走的蚂蚁都会感冒，如果它的右边存在向左走的蚂蚁，那么它的左边向右走的蚂蚁也会和他们相遇，总的感冒蚂蚁就算出来了。

AC代码

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <utility>
#include <string>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000") 
#define eps 1e-10
#define inf 0x3f3f3f3f
#define PI pair<int, int> 
typedef long long LL;
const int maxn = 50 + 5;
int pos[maxn];
bool cmp(int a, int b) {
	return abs(a) < abs(b); 
}
int main() {
	int n;
	while(scanf("%d", &n) == 1) {
		for(int i = 0; i < n; ++i) scanf("%d", &pos[i]);
		int fir = pos[0]; //第一只蚂蚁的位置 
		sort(pos, pos+n, cmp);
		int b = 0, c = 0;
		int tpos;
		for(int i = 0; i < n; ++i) {
			if(pos[i] == fir) {
				tpos = i;
				break;
			}
			if(pos[i] > 0) ++b;
		}
		for(int i = tpos+1; i < n; ++i) {
			if(pos[i] < 0) ++c;
		}
		int ans = 1;
		if(fir < 0) { //向左 
			ans += b;
			if(b) ans += c;
		}
		else if(fir > 0) { //向右 
			ans += c;
			if(c) ans += b;
		}
		printf("%d\n", ans);
 	}
	return 0;
}

如有不当之处欢迎指出！