历届试题 蚂蚁感冒
时间限制: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);
}
}