通勤车油耗(算法题)

字节跳动在北京有 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，反正是环形，都差不多