Status of Flight Tickets — LeetCode 2793 Python Solution

HardLeetCode PremiumDatabase
Problem
#2793
Reading time
4 min

Table schema

SQL
Table: Flights +-------------+------+ | Column Name | Type | +-------------+------+ | flight_id | int | | capacity | int | +-------------+------+ flight_id column contains distinct values. Each row of this table contains flight id and capacity.

Example

SQL
+-------------+------+
| Column Name | Type |
+-------------+------+
| flight_id   | int  |
| capacity    | int  |
+-------------+------+
flight_id column contains distinct values.
Each row of this table contains flight id and capacity.

Python solution

Python
import duckdb
import pandas as pd

def solution(passengers: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Passengers", passengers)
    return con.execute("""SELECT
    passenger_id,
    IF(
        (
            RANK() OVER (
                PARTITION BY flight_id
                ORDER BY booking_time
            )
        ) <= capacity,
        'Confirmed',
        'Waitlist'
    ) AS Status
FROM
    Passengers
    JOIN Flights USING (flight_id)
ORDER BY passenger_id;""").df()

Complexity

MeasureComplexity
TimeO(n log n) (typical)
SpaceO(n) auxiliary

Related problems

Frequently asked questions

How hard is LeetCode 2793. Status of Flight Tickets?
LeetCode 2793. Status of Flight Tickets is rated Hard on LeetCode.
What topics does LeetCode 2793. Status of Flight Tickets cover?
LeetCode 2793. Status of Flight Tickets is tagged Database on LeetCode.
Is LeetCode 2793. Status of Flight Tickets a premium problem?
Yes. LeetCode 2793. Status of Flight Tickets is a LeetCode Premium problem, so the full statement and test cases require a paid LeetCode subscription.

Stuck on problems like this in a live interview?

Stealth Interview is a desktop app for macOS and Windows. It reads the problem off your screen and returns a working solution with a step-by-step explanation and its time and space complexity — invisible to screen sharing.

Get Stealth Interview