leetcode1094. 拼车

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

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

必须接送的乘客数量；
乘客的上车地点；
以及乘客的下车地点。
这些给出的地点位置是从你的 初始 出发位置向前行驶到这些地点所需的距离（它们一定在你的行驶方向上）。

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

示例 1：

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

代码

class Solution {
    public boolean carPooling(int[][] trips, int capacity) {

        Map<Integer,Integer> add=new HashMap<>();
        Map<Integer,Integer> sub=new HashMap<>();
       TreeSet<Integer> treeSet=new TreeSet<>();
        for(int[] c:trips)
        {
            treeSet.add(c[1]);//用treeset记录上下车的时间节点
            treeSet.add(c[2]);
            add.put(c[1],add.getOrDefault(c[1],0)+c[0]);//记录上车的时间点和人数
            sub.put(c[2],sub.getOrDefault(c[2],0)+c[0]);
        }
         for(int c:treeSet)//模拟上下车
        {
            if(sub.containsKey(c)) capacity+=sub.get(c);
            if(add.containsKey(c)) capacity-=add.get(c);
            if(capacity<0) return false;
            
        }
        return true;
    }
}

代码

class Solution {
    public boolean carPooling(int[][] trips, int capacity) {

        int[] track=new int[1001];//记录每一个节点的改变量
        for(int[] c:trips)
        {
            track[c[1]]-=c[0];
            track[c[2]]+=c[0];      
        }
         for(int c:track)//遍历所有节点，如果容量出现不满足的情况，则返回false
        {
          capacity+=c;
         if(capacity<0) return false;
        }
        return true;
    }
}