Users With Two Purchases Within Seven Days — LeetCode 2228 Python Solution

MediumLeetCode PremiumDatabase
Problem
#2228
Reading time
4 min

Table schema

SQL
Table: Purchases +---------------+------+ | Column Name | Type | +---------------+------+ | purchase_id | int | | user_id | int | | purchase_date | date | +---------------+------+ purchase_id contains unique values. This table contains logs of the dates that users purchased from a certain retailer.

Example

SQL
+---------------+------+
| Column Name   | Type |
+---------------+------+
| purchase_id   | int  |
| user_id       | int  |
| purchase_date | date |
+---------------+------+
purchase_id contains unique values.
This table contains logs of the dates that users purchased from a certain retailer.

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("""WITH
    t AS (
        SELECT
            user_id,
            DATEDIFF(
                purchase_date,
                LAG(purchase_date, 1) OVER (
                    PARTITION BY user_id
                    ORDER BY purchase_date
                )
            ) AS d
        FROM Purchases
    )
SELECT DISTINCT user_id
FROM t
WHERE d <= 7
ORDER BY user_id;""").df()

Complexity

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

Related problems

Frequently asked questions

How hard is LeetCode 2228. Users With Two Purchases Within Seven Days?
LeetCode 2228. Users With Two Purchases Within Seven Days is rated Medium on LeetCode.
What topics does LeetCode 2228. Users With Two Purchases Within Seven Days cover?
LeetCode 2228. Users With Two Purchases Within Seven Days is tagged Database on LeetCode.
Is LeetCode 2228. Users With Two Purchases Within Seven Days a premium problem?
Yes. LeetCode 2228. Users With Two Purchases Within Seven Days 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