通勤车油耗(算法题)

字节跳动在北京有 N 个工区,形成一个环状,Bytebus 是往返在各个工区的通勤车,按工区的顺序行驶,其中第 i 个工区有汽油 gas[i] 升。你有一辆油箱容量无限的 Bytebus,从第 i 个工区开往第 i+1 个工区需要消耗汽油 cost[i] 升。你从其中的一个工区出发,开始时油箱为空,可以使用当前工区的汽油 gas[i]升。如果你可以绕环路行驶一周,则返回出发时工区的编号,否则返回 -1。

输入:gas = [1,2,3,4,5] cost = [3,4,5,1,2]

输出:3

遇到一个面经上的算法题,觉得很有趣,遂答之,这个应该也就是leetcode简单级别

public static void main(String[] args) {
        int[] cost = {3,4,5,1,2};
        int[] gas = {1,2,3,4,5};
        System.out.println(byteDance(cost,gas,3));
    }

    public static int byteDance(int[] cost,int[] gas,int num)
    {
        int sum = gas[num] - cost[num];
        if(sum < 0) return -1;
        int next = num == gas.length -1 ? 0 : num + 1;
        while(num != next) {
            sum = sum + gas[next] - cost[next];
            if (sum < 0) return -1;
            next = next == gas.length -1 ? 0 : next + 1;
        }
        return num;
    }

当然,测试用例不够,我也只是把题目中的数据都改了一遍,因为题目有些歧义,所以我默认工区号是0,1,2,3,4,反正是环形,都差不多

上一篇:【Dijstra+平均点权和最大+路径】1087 All Roads Lead to Rome (30 分)


下一篇:设计模式之装饰者模式