实时公交到站数计算

题目：XX市实时公交项目，能得到该市的公交车每一次报站信息，比如“下一站到五道口站，请下车的乘客做好准备”，需要得到线路上每个站点的下两辆车到站情况。

抽象：

131路公交车有19站，有编号为1 2 3 4，四辆公交车，四辆车下一个站点分别是2 4 6 8。那么对于1号站点，前面没有公交车，就输出-1，-1表示没有接下来的两辆车。而3号站点，因为前面只有一个1号公交车（马上到2号站台），所以输出1，-1（第一辆公交车还有1站，没有第二辆公交车）。对于18号站点，最近的两辆公交车是6、8号站台的3 4号公交车，所以输出10，12.

求demo。

#include <vector>
#include <map>
#include <utility>
#include <cstdio>
#include <algorithm>
using namespace std;
//一个busInfo表示车号为busNum的车下一站是pos站
typedef struct{
        int busNum;
        int pos;
} busInfo;

//路线id到当前线路上所有车的busInfo映射
static map<int,vector<busInfo>> lineInfos;
//路线id到路线总站数的映射
static map<int,int> stationCount;

//输出每一站的接下来两辆车信息，格式为“站编号：第一辆车到此站站树，第二辆车到此站站数”，0为马上到，-1为没有车
void printLineInfo(int lineNum){
        int nstation = stationCount[lineNum];
        vector<busInfo> infos = lineInfos[lineNum];
        for_each(infos.begin(),infos.end(),[](const busInfo &info){printf("(%d,%d) ",info.busNum,info.pos);});
        printf("\n");
        int i = 1,now = 0;
        //now是当前车站的下一个公交，每次循环就是看当前公交是否已经不是站点的“下一个”，然后根据“下一个”信息输出当前站点公交信息
        while(i <= nstation){
                if(i == infos[now].pos){
                        while(now <= (infos.size() - 1) && i >= infos[now].pos)
                                now++;
                }
                if(now == 0)
                        printf("%d:-1,-1\n",i);
                else if(now == 1)
            printf("%d:%d,-1\n",i,i - infos[now - 1].pos);
        else if((now - 2) >= 0)
            printf("%d:%d,%d\n",i,i - infos[now - 1].pos,i - infos[now - 2].pos);
                else
            printf("%d:%d,-1\n",i,i - infos[now - 1].pos);
                i++;
        }
        printf("\n");
}

//车辆位置更新事件，车辆加入线路
void busArrived(int lineNum,const busInfo &info){
        vector<busInfo> &infos = lineInfos[lineNum];
        int nstation = stationCount[lineNum];
        bool del = false;
        if(info.pos >= nstation)
                del = true;

        //删除之前车辆已有信息
        vector<busInfo>::iterator iter = find_if(infos.begin(),infos.end(),[info](const busInfo &in){return info.busNum == in.busNum;});
        if(iter != infos.end())
                infos.erase(iter);
        //如果是到终点站，删除后直接返回
        if(del)
                return;
        //否则按顺序插入
        infos.insert(upper_bound(infos.begin(),infos.end(),info,[](const busInfo &lhs,const busInfo &rhs){return lhs.pos <= rhs.pos;}),info);
}

int main(){
        int lineNum = 100;
        //100号线路有18个站点
        stationCount[100] = 18;
        //100号线路上有4辆公交，1号马上到3号站点，2号到7号...
        vector<busInfo> infos = {{1,3},{2,7},{7,9},{10,11}};
        lineInfos[100] = infos;

        //打印线路到站信息
        printLineInfo(100);
        lineInfos[100] = infos;

        //2号巴士到达第10站台
        busInfo info = {2,10};
        busArrived(100,info);
        //再次打印
        printLineInfo(100);
}

输出：

(1,3) (2,7) (7,9) (10,11) 
1:-1,-1
2:-1,-1
3:0,-1
4:1,-1
5:2,-1
6:3,-1
7:0,4
8:1,5
9:0,2
10:1,3
11:0,2
12:1,3
13:2,4
14:3,5
15:4,6
16:5,7
17:6,8
18:7,9


(1,3) (7,9) (2,10) (10,11) 
1:-1,-1
2:-1,-1
3:0,-1
4:1,-1
5:2,-1
6:3,-1
7:4,-1
8:5,-1
9:0,6
10:0,1
11:0,1
12:1,2
13:2,3
14:3,4
15:4,5
16:5,6
17:6,7
18:7,8

符合预期