Product's Price for Each Store — LeetCode 1777 Python Solution

EasyLeetCode PremiumDatabase
Problem
#1777
Reading time
3 min

Table schema

SQL
Table: Products +-------------+---------+ | Column Name | Type | +-------------+---------+ | product_id | int | | store | enum | | price | int | +-------------+---------+ In SQL, (product_id, store) is the primary key for this table. store is a category of type ('store1', 'store2', 'store3') where each represents the store this product is available at.

Example

SQL
+-------------+---------+
| Column Name | Type    |
+-------------+---------+
| product_id  | int     |
| store       | enum    |
| price       | int     |
+-------------+---------+
In SQL, (product_id, store) is the primary key for this table.
store is a category of type ('store1', 'store2', 'store3') where each represents the store this product is available at.
price is the price of the product at this store.

Python solution

Python
import duckdb
import pandas as pd

def solution(products: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Products", products)
    return con.execute("""SELECT
    product_id,
    SUM(IF(store = 'store1', price, NULL)) AS store1,
    SUM(IF(store = 'store2', price, NULL)) AS store2,
    SUM(IF(store = 'store3', price, NULL)) AS store3
FROM Products
GROUP BY 1;""").df()

Complexity

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

Related problems

Frequently asked questions

How hard is LeetCode 1777. Product's Price for Each Store?
LeetCode 1777. Product's Price for Each Store is rated Easy on LeetCode.
What topics does LeetCode 1777. Product's Price for Each Store cover?
LeetCode 1777. Product's Price for Each Store is tagged Database on LeetCode.
Is LeetCode 1777. Product's Price for Each Store a premium problem?
Yes. LeetCode 1777. Product's Price for Each Store 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