The Number of Rich Customers — LeetCode 2082 Python Solution

EasyLeetCode PremiumDatabase
Problem
#2082
Reading time
2 min

Table schema

SQL
Table: Store +-------------+------+ | Column Name | Type | +-------------+------+ | bill_id | int | | customer_id | int | | amount | int | +-------------+------+ bill_id is the primary key (column with unique values) for this table. Each row contains information about the amount of one bill and the customer associated with it.

Example

SQL
+-------------+------+
| Column Name | Type |
+-------------+------+
| bill_id     | int  |
| customer_id | int  |
| amount      | int  |
+-------------+------+
bill_id is the primary key (column with unique values) for this table.
Each row contains information about the amount of one bill and the customer associated with it.

Python solution

Python
import duckdb
import pandas as pd

def solution(store: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Store", store)
    return con.execute("""SELECT
    COUNT(DISTINCT customer_id) AS rich_count
FROM Store
WHERE amount > 500;""").df()

Complexity

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

Related problems

Frequently asked questions

How hard is LeetCode 2082. The Number of Rich Customers?
LeetCode 2082. The Number of Rich Customers is rated Easy on LeetCode.
What topics does LeetCode 2082. The Number of Rich Customers cover?
LeetCode 2082. The Number of Rich Customers is tagged Database on LeetCode.
Is LeetCode 2082. The Number of Rich Customers a premium problem?
Yes. LeetCode 2082. The Number of Rich Customers 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