题1094、拼车

一、题目¹

二、思路

这道题其实贼简单，暴力遍历即可，因为数据量是有限的，最多也就1001次上下车。

个人感觉他麻烦的地方在于他每一条数据不是按照上车位置排序的，有点麻烦，别的也无所谓了，我的想法就是用两个哈希表，把上下车的数据存储起来，然后按照从大到小的顺序查找哈希表，上车就减座位，下车就加座位，不够就返回false，最后成功通过了。

看了大佬提交的代码，发现，其实两个哈希表可以直接合并，在同一个位置上下车的数据可以直接相减，然后正数就代表上车，负数就代表下车，把数据累加起来，中间仍然是超过座位总数就返回false。

三、代码

import java.util.HashMap;
import java.util.Map;

public class T1094 {

    public T1094(){
//        int[][] trips = {{2,1,5},{3,3,7}};      //false     4
//        int[][] trips = {{3,2,7},{3,7,9},{8,3,9}};      //true      11
//        int[][] trips = {{2,1,5},{3,5,7}};      //true      3
        int[][] trips = {{8,2,3},{4,1,3},{1,3,6},{8,4,6},{4,4,8}};      //false 12
//        int[][] trips = {{7,5,6},{6,7,8},{10,1,6}};       //false 16
        System.out.println( carPooling(trips,12) );
    }

    //大佬的代码
    public boolean carPooling(int[][] trips, int capacity) {

        //存储每一站上下车的信息
        //正数代表上车，负数代表下车
        int[] count = new int[1001];

        //遍历trips，将上下车信息存储在count里面
        for ( int[] trip : trips ){
            count[trip[1]] += trip[0];
            count[trip[2]] -= trip[0];
        }

        //判断第一次上车的人数是否超过了，座位数
        if ( count[0] > capacity )
            return false;

        //遍历count数组，将之前所有的上下车信息全部累加起来
        //判断是否超过了座位数
        for ( int i = 1; i < 1001; i++ ){
            count[i] += count[i-1];
            if (capacity<count[i])
                return false;
        }

        return true;
    }
    //本菜鸡的代码
    public boolean carPooling1(int[][] trips, int capacity) {

        //记录每一次上下车的信息
        Map<Integer,Integer> getOn = new HashMap<Integer, Integer>();
        Map<Integer,Integer> getOff = new HashMap<Integer, Integer>();

        //将上下车的信息存储在哈希表中
        for ( int[] i : trips ){

            int tmp = i[0]+ getOn.getOrDefault(i[1], 0);
            getOn.put( i[1],tmp );
            tmp = i[0]+getOff.getOrDefault(i[2], 0);

            //如果某站下车的人数超过了座位数，显然完不成任务
            if ( tmp > capacity )
                return false;
            else
                getOff.put( i[2], tmp);
        }

        //遍历所有的哈希表
        for ( int i = 0; i <= 1000 && getOn.size() > 0; i++ ){

            //将下车的座位释放，上车的座位减去
            capacity += getOff.getOrDefault(i, 0);
            capacity -= getOn.getOrDefault(i, 0);

            //将已经处理过的上车信息移除
            getOn.remove(i);

            //如果座位不够，返回false
            if ( capacity < 0 )
                return false;
        }

        return true;
    }

    public static void main(String[] args) {
        T1094 t1094 = new T1094();
    }
}

---

1. 来源：力扣（LeetCode）
   链接：https://leetcode-cn.com/problems/car-pooling
   著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。 ↩︎