蚂蚁（Piotr's Ants，UVa 10881）

题目链接

https://vjudge.net/problem/UVA-10881

题意

一根长度为L厘米的木棍上有n只蚂蚁，每只蚂蚁要么朝左爬，要么朝右爬，速度为1厘米/秒。当两只蚂蚁相撞时，二者同时掉头（掉头时间忽略不计）。给出每只蚂蚁的初始位置和朝向，计算T秒之后每只蚂蚁的位置。每组最后输出一个空行。

分析

把蚂蚁看作车，两只蚂蚁相撞时等效为两辆车的驾驶员互换（车的朝向不变）。这样n辆车的最终位置是唯一确定的，问题转化成了求n个驾驶员与n辆车的映射。
两辆车要相撞肯定得朝不同方向行驶，用集合l表示朝右行驶的车，用集合r表示朝左行驶的车。
对于l中的每一辆车x，穷举r中的每一辆车y与x比较看是否相撞，即是否互换驾驶员。经过穷举后，n辆车的最终驾驶员就确定下来了。

解题

（1）构造结构体数组a，a中存放n辆车的位置、朝向、驾驶员编号。
（2）对结构体按相对位置从小到大排序
（3）计算驾驶员的互换情况
（4）求出n辆车的最终位置
（5）对结构体按驾驶员编号从小到大排序
（6）输出答案

代码

#include <cstdio>
#include <cstring>
#include <stack>
#include <algorithm>
#include <map>

using namespace std;
const int maxn=1e4+100;
struct ant
{
    int num,pos,sub;
    char dir;
} a[maxn];
stack<ant> l; //朝向右边的蚂蚁
ant r[maxn]; //朝向左边的蚂蚁
map<int,int> m;
bool cmp1(const ant &t1,const ant &t2)
{
    return t1.pos<t2.pos;
}
bool cmp2(const ant &t1,const ant &t2)
{
    return t1.num<t2.num;
}

int main()
{
    int t,d=1,tot=0;
    scanf("%d",&t);
    while(t--)
    {
        int L,T,n;
        m.clear();
        tot=0;
        scanf("%d%d%d",&L,&T,&n);
        for(int i=0; i<n; i++)
        {
            scanf("%d %c",&a[i].pos,&a[i].dir);
            a[i].num=i;
        }
        sort(a,a+n,cmp1);
        for(int i=0; i<n; i++)
        {
            a[i].sub=i;
        }
        while(!l.empty()) l.pop();
        for(int i=0; i<n; i++)
            if(a[i].dir=='R')
                l.push(a[i]);
            else
                r[++tot]=a[i];

        while(!l.empty())
        {
            ant x=l.top();
            l.pop();

            for(int i=1; i<=tot; i++)
            {
                ant y=r[i];
                if(x.pos>y.pos) continue;
                if(x.pos+T>=y.pos-T)
                {
                    swap(a[x.sub].num,a[y.sub].num);
                }
            }

        }
        for(int i=0; i<n; i++)
        {
            a[i].pos+=a[i].dir=='R'?T:-T;
            m[a[i].pos]++;
        }
        sort(a,a+n,cmp2);
        printf("Case #%d:\n",d++);
        for(int i=0; i<n; i++)
        {

            if(a[i].pos>L || a[i].pos<0)
                printf("Fell off\n");

            else if(m[a[i].pos]>1)
                printf("%d %s\n",a[i].pos,"Turning");
            else
                printf("%d %c\n",a[i].pos,a[i].dir);
        }
        printf("\n");
    }
    return 0;
}