学习分享算法解决数学问题 - 黑马程序员技术交流社区

输入的第一行为数据组数。每组数据的第一行为3个正整数L, T,n（0≤n≤10 000）；以下n行每行描述一只蚂蚁的初始位置，其中，整数x为蚂蚁距离木棍左端的距离（单位：厘米），字母表示初始朝向（L表示朝左，R表示朝右）。

假设你在远处观察这些蚂蚁的运动，会看到什么？一群密密麻麻的小黑点在移动。由于黑点太小，所以当蚂蚁因碰撞而掉头时，看上去和两个点“对穿而过”没有任何区别，换句话说，如果把蚂蚁看成是没有区别的小点，那么只需独立计算出每只蚂蚁在T时刻的位置即可。比如，有3只蚂蚁，蚂蚁1=(1, R)，蚂蚁2= (3, L)，蚂蚁3=(4, L)，则两秒钟之后，3只蚂蚁分别为(3,R)、(1,L)和(2,L)。

注意，虽然从整体上讲，“掉头”等价于“对穿而过”，但对于每只蚂蚁而言并不是这样。蚂蚁1的初始状态为(1,R)，因此一定有一只蚂蚁在两秒钟之后处于(3,R)的状态，但这只蚂蚁却不一定是蚂蚁1。换句话说，我们需要搞清楚目标状态中“谁是谁”。

也许读者已经发现了其中的奥妙：所有蚂蚁的相对顺序是保持不变的，因此把所有目标位置从小到大排序，则从左到右的每个位置对应于初始状态下从左到右的每只蚂蚁。由于原题中蚂蚁不一定按照从左到右的顺序输入，还需要预处理计算出输入中的第i只蚂蚁的序号order[i]。完整代码如下。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 10000 + 5;
struct Ant {
int id; //输入顺序
int p; //位置
int d; //朝向。 -1: 左; 0:转身中; 1:右
bool operator < (const Ant& a) const {
return p < a.p;
}
} before[maxn], after[maxn];
const char dirName[][10] = {"L", "Turning", "R"};
int order[maxn]; //输入的第i只蚂蚁是终态中的左数第order[i]只蚂蚁
int main() {
int K;
scanf("%d", &K);
for(int kase = 1; kase <= K; kase++) {
int L, T, n;
printf("Case #%d:\n", kase);
scanf("%d%d%d", &L, &T, &n);
for(int i = 0; i < n; i++) {
int p, d;
char c;
scanf("%d %c", &p, &c);
d = (c == 'L' ? -1 : 1);
before[i] = (Ant){i, p, d};
after[i] = (Ant){0, p+T*d, d}; //这里的id是未知的
}
//计算order数组
sort(before, before+n);
for(int i = 0; i < n; i++)
order[before[i].id] = i;
//计算终态
sort(after, after+n);
for(int i = 0; i < n-1; i++) //修改碰撞中的蚂蚁的方向
if(after[i].p == after[i+1].p) after[i].d = after[i+1].d = 0;
//输出结果
for(int i = 0; i < n; i++) {
int a = order[i];
if(after[a].p < 0 || after[a].p > L) printf("Fell off\n");
else printf("%d %s\n", after[a].p, dirName[after[a].d+1]);
}
printf("\n");
}
return 0;
}

复制代码

运行效果：