LeetCode之Gas Station --环形加油站周游

题目

There are N gas stations along a circular route, where the amount of gas at station i is gas[i].

You have a car with an unlimited gas tank and it costs cost[i] of gas to travel from station i to its next station (i+1). You begin the journey with an empty tank at one of the gas stations.

Return the starting gas station's index if you can travel around the circuit once, otherwise return -1.

Note:
The solution is guaranteed to be unique.

解法

这道题分两步来做,第一步先确定是否能周游一圈,第二步在找从哪个加油站开始。

确定能否周游一圈

首先给出一个断言:如果gas数组元素之和大于等于cost数组元素之和,那么必然能周游一圈。证明很简单,可以这样想象下,加油站的高度为0,加油站的油高度为gas,cost代表两个加油站之间的坑深度,且油的流向为同一个方向,那么当gas之和大于等于cost之和的话,最终动态平衡后整体油面高度>=0,那么必然能周游一圈。否则不能。

从哪个加油站开始

一开始没看到题目的note,感觉可以从多个加油站开始。note提示题目有唯一解,我想如果能找到环中最大连续序列的和的话,那么就能找到其实点,但是找环中的最大连续感觉比找数组中最大连续序列和(见http://blog.csdn.net/xhu_eternalcc/article/details/24880013)要难,就直接蛮力找的:如果从某个加油站能向前遍历N步(加油站的个数),那么就是从这个加油站开始,需要进行如下优化:

det[i]=gas[i]-cost[i]

sum为从第i个加油站开始,向前遍历到某加油站时的总剩油量

1)起始加油站的det[i]>=0

2)  当从第i个加油站开始,向前step步,到达第(i+step)%N个加油站,sum<0时,这时在探测第x= i+1 至(i+step)%N个加油站能否作为开始点时,可以直接求得从加油站x到(i+step)%N 的总剩油量sum,详见代码。

代码

class Solution {
public:
    int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
        int sum=0;
        int *det=new (std::nothrow)int[gas.size()];
        for(std::vector<int>::size_type i=0;i<gas.size();++i){
            det[i]=gas[i]-cost[i];
            sum+=det[i];
        }
        if(sum<0){
            delete [] det;
            return -1;
        }
        int step=0;
        for(std::vector<int>::size_type i=0;i<gas.size();++i){
            if(step>0){//i-1加油站向前走step步后总油量开始<0
                --step;
                if(det[i]<0){
                    sum-=det[i];
                    continue;
                }else if(sum<0){
                    sum-=det[i];
                    continue;
                }
            }else{
                if(det[i]<0)
                    continue;
                sum=det[i]; 
            }
            while(sum>=0&&step<gas.size()){
                ++step;
                sum+=det[(step+i)%gas.size()];
            }
            if(step==gas.size()){
                delete [] det;
                return i;
            }
            else
                sum-=det[i];
        }
    }
};





你可能感兴趣的:(code,code,Station,gas,gas,加油站,Leet,Leet,周游,Statio)