The Users That Are Eligible for Discount — LeetCode 2230 Python Solution

EasyLeetCode PremiumDatabase
Problem
#2230
Reading time
3 min

Table schema

SQL
Table: Purchases +-------------+----------+ | Column Name | Type | +-------------+----------+ | user_id | int | | time_stamp | datetime | | amount | int | +-------------+----------+ (user_id, time_stamp) is the primary key (combination of columns with unique values) for this table. Each row contains information about the purchase time and the amount paid for the user with ID user_id.

Example

SQL
+-------------+----------+
| Column Name | Type     |
+-------------+----------+
| user_id     | int      |
| time_stamp  | datetime |
| amount      | int      |
+-------------+----------+
(user_id, time_stamp) is the primary key (combination of columns with unique values) for this table.
Each row contains information about the purchase time and the amount paid for the user with ID user_id.

Python solution

Python
import duckdb
import pandas as pd

def solution(purchases: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Purchases", purchases)
    return con.execute("""CREATE PROCEDURE getUserIDs(startDate DATE, endDate DATE, minAmount INT)
BEGIN
    SELECT DISTINCT user_id
    FROM Purchases
    WHERE amount >= minAmount AND time_stamp BETWEEN startDate AND endDate
    ORDER BY user_id;
END;""").df()

Complexity

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

Related problems

Frequently asked questions

How hard is LeetCode 2230. The Users That Are Eligible for Discount?
LeetCode 2230. The Users That Are Eligible for Discount is rated Easy on LeetCode.
What topics does LeetCode 2230. The Users That Are Eligible for Discount cover?
LeetCode 2230. The Users That Are Eligible for Discount is tagged Database on LeetCode.
Is LeetCode 2230. The Users That Are Eligible for Discount a premium problem?
Yes. LeetCode 2230. The Users That Are Eligible for Discount 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