您的位置:首页 > 其它

LeetCode Gas Station

2015-03-19 10:55 351 查看
题意:

给定一个[0,n)的gas station,每个站可以加gas[i],然后才从每个站到下一个站需要cost[i]的gas cost,问从哪个节点可以绕转一圈

解法:

首先既然是可以转圈的,那么唯一不成立的情况就是gas和比cost和小,否则都能够“转”出结果来。可能略微抽象,大家仔细想想。证明我也不会囧……

然后就是很明显的贪心了,到了一个gas station就把gas[i]加满,然后再扣掉下面的cost[i],累加,只要转一圈能大于0即可。

这么一听是不是感觉很像最大区间和呢?不过这里不用求最大区间和,而是最大区间和能否大于0并且横跨整个长度l。

吐槽:

题目中没有说明顺时针还是逆时针,做法其实是一样的。我试了一下,都是默认顺时针的结果,不过为了更加的trivial,我都写了~

public class Solution {
int clockwise[];
int counterclockwise[];
int l;

public int canCompleteCircuit(int[] gas, int[] cost) {
l = gas.length;
int s = 0;
for (int i = 0; i < l; i++)
s += gas[i] - cost[i];
if (s < 0)
return -1;

clockwise = new int[l];
counterclockwise = new int[l];
for (int i = 0; i < l; i++)
clockwise[i] = gas[i] - cost[i];
for (int i = 1; i < l; i++)
counterclockwise[i] = gas[i] - cost[i - 1];
counterclockwise[0] = gas[0] - cost[l - 1];

int sum = 0, index = 0;
for (int i = 0; i < l * 2; i++) {
if (i - l == index)
return index % l;
if (sum + clockwise[i % l] < 0) {
index = (i + 1) % l;
sum = 0;
} else {
sum += clockwise[i % l];
}
}

sum = 0;
index = 2 * l - 1;
for (int i = l * 2 - 1; i >= 0; i--) {
System.out.println(i + " --- " + index + " " + sum);
if (i + l == index) {
return index % l;
}
if (sum + counterclockwise[i % l] < 0) {
index = (i - 1) % l;
i--;
sum = 0;
} else {
sum += counterclockwise[i % l];
}
}
return -1;
}

public static void main(String[] args) {
int a[] = new int[] { 2, 3, 1 };
int b[] = new int[] { 3, 1, 2 };
Solution s = new Solution();
System.out.println(s.canCompleteCircuit(a, b));
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: