前言

Weekly Contest 142的 拼车：

假设你是一位顺风车司机，车上最初有 capacity 个空座位可以用来载客。由于道路的限制，车 只能 向一个方向行驶（也就是说，不允许掉头或改变方向，你可以将其想象为一个向量）。

这儿有一份行程计划表 trips[][]，其中 trips[i] = [num_passengers, start_location, end_location] 包含了你的第 i 次行程信息：

  • 必须接送的乘客数量；
  • 乘客的上车地点；
  • 以及乘客的下车地点。

这些给出的地点位置是从你的 初始 出发位置向前行驶到这些地点所需的距离（它们一定在你的行驶方向上）。

请你根据给出的行程计划表和车子的座位数，来判断你的车是否可以顺利完成接送所用乘客的任务（当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false）。

示例1：

输入：trips = [[2,1,5],[3,3,7]], capacity = 4输出：false

示例2：

输入：trips = [[2,1,5],[3,3,7]], capacity = 5输出：true

示例3：

输入：trips = [[2,1,5],[3,5,7]], capacity = 3输出：true

示例4：

输入：trips = [[3,2,7],[3,7,9],[8,3,9]], capacity = 11输出：true

提示：

  1. 你可以假设乘客会自觉遵守 “先下后上” 的良好素质
  2. trips.length <= 1000
  3. trips[i].length == 3
  4. 1 <= trips[i][0] <= 100
  5. 0 <= trips[i][1] < trips[i][2] <= 1000
  6. 1 <= capacity <= 100000

解题思路

本题难度为中等，主要是需要注意以下几点：

  1. 多段路程中可能存在部分乘客下车的情况
  2. 需要考虑车子剩余的空座位
  3. 可能会存在多段路程的乘客同一个下车地点

我的解题思路如下：

  1. 将所有行程按照上车地点从小到大排序

  2. 遍历所有行程，模拟车子行驶，使用TreeMap记录有某个下车地点下车人数，按照以下逻辑处理

    1. 如果当前没有乘客在车上（TreeMap为空），则判断本次行程是否能够让乘客都坐下，若是将本次的行程的下车地点为key，乘客数目为value存入到TreeMap中；否则直接返回false
    2. 如果当前有乘客（TreeMap不为空），则检查所有上车乘客是否存在需要下车的人（TreeMap中的key下车地点小于或等于本次行程的上车地点）并将其移除。然后判断车子当前是否有足够座位载客（TreeMap的value之和加上本次行程的乘客数是否不大于车子总座位数），如果足够则加入本次行程（TreeMap记录下来，如果存在相同下车地点，则人数相加），否则返回false

实现代码

   /**     * 1094. 拼车     *     * @param trips     * @param capacity     * @return     */    public boolean carPooling(int[][] trips, int capacity) {        boolean flag = true;        // 根据上车地点从小到大排序        Arrays.sort(trips, Comparator.comparingInt(o -> o[1]));        // key为下车地点，value为乘客数目        TreeMap<Integer,Integer> capacityMap=new TreeMap<>();        for (int i = 0; i < trips.length; i++) {            // 乘客数量            int numPassengers = trips[i][0];            // 上车地点            int startLocation = trips[i][1];            // 下车地点            int endLocation = trips[i][2];            if(!capacityMap.isEmpty()){                // 处理java.util.ConcurrentModificationException                Set<Integer> locationSet = new TreeSet<>();                locationSet.addAll(capacityMap.keySet());                Iterator<Integer> it=locationSet.iterator();                while (it.hasNext()){                    Integer lastEndLocation=it.next();                    if(lastEndLocation<=startLocation){ // 到达终点，乘客下车                        capacityMap.remove(lastEndLocation);                    }                }                // 计算当前总乘客数                int totalCap=capacityMap.values().stream().mapToInt(Integer::intValue).sum()+numPassengers;                if(totalCap>capacity){ // 车子座位不足                    flag=false;                    break;                }                if(capacityMap.containsKey(endLocation)){ // 是否存在同一个下车地点的乘客                    capacityMap.put(endLocation,capacityMap.get(endLocation)+numPassengers);                }else{                    capacityMap.put(endLocation,numPassengers);                }            }else{                if (numPassengers > capacity) { // 车子座位不足                    flag = false;                    break;                }                capacityMap.put(endLocation,numPassengers);            }        }        return flag;    }