坠落的蚂蚁（北大复试机试题）

坠落的蚂蚁

一根长度为 1 米的木棒上有若干只蚂蚁在爬动。 它们的速度为每秒一厘米或静止不动，方向只有两种，向左或者向右。
如果两只蚂蚁碰头，则它们立即交换速度并继续爬动。 三只蚂蚁碰头，则两边的蚂蚁交换速度，中间的蚂蚁仍然静止。
如果它们爬到了木棒的边缘（0 或 100 厘米处）则会从木棒上坠落下去。 在某一时刻蚂蚁的位置各不相同且均在整数厘米处（即
1，2，3，…99 厘米），有且只有一只蚂蚁 A 速度为 0，其他蚂蚁均在向左或向右爬动。
给出该时刻木棒上的所有蚂蚁位置和初始速度，找出蚂蚁 A 从此时刻到坠落所需要的时间

输入格式

第一行包含一个整数表示蚂蚁的个数 N，之后共有 N 行，每一行描述一只蚂蚁的初始状态。
每个初始状态由两个整数组成，中间用空格隔开，第一个数字表示初始位置厘米数 P，第二个数字表示初始方向，−1 表示向左，1 表示向右，0 表示静止。

输出格式

蚂蚁 A 从开始到坠落的时间。若不会坠落，输出 Cannot fall!。

数据范围：

2≤N≤99, 1≤P≤99

输入样例：

4
10 1
90 0
95 -1
98 -1

输出样例：

98

思路

最初的思路是模拟，但后来想了想太麻烦

参考了其他几位博主的解答，思路大体相同。以下是几个要点：

• 两只蚂蚁交换速度，相当于身份互换，二者对穿。所以，两只与A无关的蚂蚁碰面，并不影响A的最终结果。
• 由上一条推知，A的左边蚂蚁若向左走，右边蚂蚁若向右走，不会影响A，且最终会掉落下去，不用考虑。
• A的左边蚂蚁向右走，A的右边蚂蚁向左走，无论其与A相遇的先后顺序，最终A的位置不变。
  （解释：想象有 B、A、C 3只蚂蚁，BA先碰面，B留在A的位置，A向右走，AC碰面，交换速度后，A向左走再次与B碰面，留在最初A的位置）
• 由上一条可推知，若A左右两边相向而行的蚂蚁数量相等，则A留在原位，不会掉下去。
• 如果两边相向而行的的蚂蚁数量不同，根据上一条，最终会影响到A的蚂蚁应该是左边蚂蚁与右边蚂蚁抵消之后剩下的蚂蚁中，最靠近A的第一只，因此，A掉下去的时间，应该就是这个第一只蚂蚁从它的最初位置走到边缘的时间。

因此，做法如下：用pair将蚂蚁的位置和速度打包成一个数据，利用set自排序的特征，用一个set存储整个蚂蚁群，再分别用两个vector存储A左边向右走的蚂蚁和A右边向左走的蚂蚁，统计两个set的蚂蚁数量，若相同，则A不会坠落，若左多，则左边第L-R只 （注意：下标为L-R-1，下同） 蚂蚁为第一只；反之右边第L+1只为第一只。

代码

#include<iostream>
#include<cstdlib>
#include<string>
#include<set>
#include<vector>
using namespace std;

/*
    题目描述：
        坠落的蚂蚁
*/

typedef pair<int,int> ant;
set<ant> a;
vector<ant> l,r;
int pos_A;

int main(){
    int n;
    int x,y;
    cin>>n;
    while(n--){
        cin>>x>>y;
        if(!y) pos_A = x;
        a.insert({x,y});//这种打包方式还不熟，借鉴了一下别人的操作
    }
    for(auto i : a){ //要注意，set插入为insert，vector插入为push_back，另外set不支持随机访问
        if(i.first < pos_A && i.second == 1) l.push_back(i);
        else if(i.first > pos_A && i.second == -1) r.push_back(i);
    }
    if(l.size() == r.size()) cout<<"Cannot fall!"<<endl;
    else if(l.size() > r.size()) cout<<100 - l[l.size()-r.size()-1].first <<endl;
    else cout<<r[l.size()].first<<endl;
    
    system("pause");
    return 0;
}