坠落的蚂蚁

坠落的蚂蚁

题目描述
一根长度为1米的木棒上有若干只蚂蚁在爬动。它们的速度为每秒一厘米或静止不动，方向只有两种，向左或者向右。如果两只蚂蚁碰头，则它们立即交换速度并继续爬动。三只蚂蚁碰头，则两边的蚂蚁交换速度，中间的蚂蚁仍然静止。如果它们爬到了木棒的边缘（0或100厘米处）则会从木棒上坠落下去。在某一时刻蚂蚁的位置各不相同且均在整数厘米处（即1，2，3，…99厘米），有且只有一只蚂蚁A速度为0，其他蚂蚁均在向左或向右爬动。给出该时刻木棒上的所有蚂蚁位置和初始速度，找出蚂蚁A从此时刻到坠落所需要的时间。
输入描述:
第一行包含一个整数表示蚂蚁的个数N（2<=N<=99），之后共有N行，每一行描述一只蚂蚁的初始状态。每个初始状态由两个整数组成，中间用空格隔开，第一个数字表示初始位置厘米数P（1<=P<=99），第二个数字表示初始方向,-1表示向左，1表示向右，0表示静止。
输出描述:
蚂蚁A从开始到坠落的时间。若不会坠落，输出“Cannot fall!”
示例1
输入

4
10 1
90 0
95 -1
98 -1

输出

98

代码：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

struct node{
    int dis;
    int dir;
};

bool cmp(node a,node b){
    return a.dis<b.dis;
}

node re[110];
vector<node> le,ri;
int idx;
int n;
int main(){
    cin >> n;
    int m;
    for(int i=0;i<n;i++){
        int d,r;
        cin >> d >> r;
        if(r==0)
            m=d;
        re[idx++]={d,r};
    }
    sort(re,re+idx,cmp);
    int l=0,r=0;
    for(int i=0;i<idx;i++){
        if(re[i].dis<m&&re[i].dir==1){
            l++;
            le.push_back(re[i]);
        }
        else if(re[i].dis>m&&re[i].dir==-1){
            r++;
            ri.push_back(re[i]);
        }
    }
    
    
    if(l==r)
        printf("Cannot fall!\n");
    else{
        if(l>r)
            printf("%d\n",100-le[le.size()-ri.size()-1].dis);
        else
            printf("%d\n",ri[le.size()]);
    }
    return 0;
}