蚂蚁感冒 解题报告

历届试题 蚂蚁感冒  

时间限制：1.0s   内存限制：256.0MB

   

问题描述

　　长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左，有的朝右。

　　每只蚂蚁都只能沿着杆子向前爬，速度是1厘米/秒。

　　当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。

　　这些蚂蚁中，有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。

　　请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

输入格式

　　第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。

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

输出格式

　　要求输出1个整数，表示最后感冒蚂蚁的数目。

样例输入

3
5 -2 8

样例输出

1

样例输入

5
-10 8 -20 12 25

样例输出

3

思路：这道题可以参见刘汝佳《算法竞赛入门经典训练指南》第一章例题五蚂蚁这道题。

引用原话：假设你在远处观察这些蚂蚁的运动，会看到什么？一群密密麻麻的小黑点在移动，由于黑点太小，所以当蚂蚁碰撞而

掉头时，看上去和两个点“对穿而过”没有什么区别。(a)

现在已知第一个蚂蚁感冒了，不妨设其向右运动。那么显然，该蚂蚁右边的蚂蚁若向右运动，不会感冒。而向左运动，结合(a)

可看作该两蚂蚁会碰面（看成“对穿而过”的情况，即每个蚂蚁一直保持自己的运动状态而忽略碰面的影响），即第一个蚂蚁右边

向左运动的蚂蚁会感冒。不妨将其中一个会感冒的蚂蚁记为b。现在再看第一个蚂蚁左边蚂蚁的情况，显然其若向左运动也不会感冒，而向右运动的话，若b不存在，则该蚂蚁不会感冒，否则，该蚂蚁与蚂蚁b可看成相对运动（一右一左），会碰面，从而该

蚂蚁也会感冒。这样，所有蚂蚁感冒的情况都考虑出来了。

类似的，第一个感冒的蚂蚁一开始向左运动，该条件下各蚂蚁感冒与否也能判断出来。

代码如下：

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

class State {
	boolean dir;
	int cur;
}

public class Main {
	static int n, num, pos, val, ans;
	static State arr[];

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Comparator<State> cmp = new Comparator<State>() {
			@Override
			public int compare(State o1, State o2) {
				// TODO Auto-generated method stub
				return (o1.cur - o2.cur);
			}

		};
		Scanner reader = new Scanner(System.in);
		n = reader.nextInt();
		arr = new State[n + 1];
		int poscold = 0;
		for (int i = 1; i <= n; i++) {
			num = reader.nextInt();
			arr[i] = new State();
			if (num > 0)
				arr[i].dir = true;
			else
				arr[i].dir = false;
			arr[i].cur = Math.abs(num);
			if (i == 1)
				poscold = arr[1].cur;
		}
		Arrays.sort(arr, 1, n + 1, cmp);
		for (int i = 1; i <= n; i++)
			if (poscold == arr[i].cur) {
				pos = i;
				break;
			}
		ans = 1;
		if (arr[pos].dir) {
			val = 0;
			for (int i = pos + 1; i <= n; i++)
				if (!arr[i].dir)
					val++;
			ans += val;
			if (val >= 1) {
				for (int i = 1; i <= pos - 1; i++)
					if (arr[i].dir)
						ans++;
			}
		} else {
			val = 0;
			for (int i = 1; i <= pos - 1; i++)
				if (arr[i].dir)
					val++;
			ans += val;
			if (val >= 1) {
				for (int i = pos + 1; i <= n; i++)
					if (!arr[i].dir)
						ans++;
			}
		}
		System.out.println(ans);
	}

}