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下车地点小于或等于本次行程的上车地点)并将其移除。然后判断车子当前是否有足够座位载客(TreeMapvalue之和加上本次行程的乘客数是否不大于车子总座位数),如果足够则加入本次行程(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 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 locationSet = new TreeSet<>();
                locationSet.addAll(capacityMap.keySet());
                Iterator 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;
    }

你可能感兴趣的:(leetcode,java,算法)