1094拼车

前言

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;
    }