有一个环形公路上有n个加油站,第i个加油站的油量为ai。假设有一辆邮箱体积无穷大的汽车,初始邮箱是空的,汽车从加油站i行驶到加油站i+1需耗油g[i]。
问是否能够选出某个加油站作为起点,使汽车能够绕环形公路行驶一圈返回到该加油站。
实现函数int selectGasStation(int a[], int g[], int n)
,如果存在满足条件的加油站,返回该加油站的序号(0-based)。否则返回-1。
提示:n可能达到106,O(n2)的枚举算法会超出时间限制
据说是google在08年的一道题,哎,好难啊,在面试时那么几分钟我肯定想不出来的,哎。
主要就是变形成为查找是否有一个n长的前缀和全大于0的序列。
#include <vector>
using namespace std;
int selectGasStation(const vector<int> &a, const vector<int> &g) {
vector<int> line(a.size()<<1);
int n=line.size();
for(int i=0;i<n;i++)
line[i]=a[i%a.size()]-g[i%a.size()];
int start=0;
int oil=0;
for(int i=0;i<n;i++)
{
if ( oil<0 )
{
start=i;
oil=line[i];
}
else
{
oil+=line[i];
if ( i-start>=a.size() )
return start;
}
}
return -1;
}