Gas Station

interview_workbook/leetcode/greedy /app/src/interview_workbook/leetcode/greedy/gas_station.py
View Source

Algorithm Notes

Summary: Gas Station — notes not yet curated.
Time: Estimate via loops/recurrences; common classes: O(1), O(log n), O(n), O(n log n), O(n^2)
Space: Count auxiliary structures and recursion depth.
Tip: See the Big-O Guide for how to derive bounds and compare trade-offs.

Big-O Guide

Source

"""
Gas Station

Problem: Gas Station
LeetCode link: https://leetcode.com/problems/gas-station/
Description: Given two integer arrays gas and cost, determine if there exists a starting gas station index from which you can travel around the circuit once in the clockwise direction. Return -1 if not possible.
"""

from src.interview_workbook.leetcode._registry import register_problem
from src.interview_workbook.leetcode._types import Category, Difficulty


class Solution:
    def solve(self, *args) -> int:
        """
        Greedy solution to the Gas Station problem.
        Returns the starting index if possible, otherwise -1.
        """
        gas, cost = args
        if sum(gas) < sum(cost):
            return -1

        start = 0
        tank = 0
        for i in range(len(gas)):
            tank += gas[i] - cost[i]
            if tank < 0:
                start = i + 1
                tank = 0
        return start


def demo():
    """Run a demo for the Gas Station problem."""
    solver = Solution()
    gas = [1, 2, 3, 4, 5]
    cost = [3, 4, 5, 1, 2]
    result = solver.solve(gas, cost)
    return str(result)


register_problem(
    id=134,
    slug="gas_station",
    title="Gas Station",
    category=Category.GREEDY,
    difficulty=Difficulty.MEDIUM,
    tags=["array", "greedy"],
    url="https://leetcode.com/problems/gas-station/",
    notes="",
)