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