GROUP BY single and multiple columns in PostgreSQL - Time & Space Complexity

Time Complexity: GROUP BY single and multiple columns

O(n)

Understanding Time Complexity

When using GROUP BY in SQL, the database groups rows based on column values.

We want to understand how the time to group data changes as the data grows.

Scenario Under Consideration

Analyze the time complexity of these two queries:

-- Group by a single column
SELECT department, COUNT(*)
FROM employees
GROUP BY department;

-- Group by multiple columns
SELECT department, role, COUNT(*)
FROM employees
GROUP BY department, role;

These queries count employees grouped by one or two columns.

Identify Repeating Operations

Look for repeated work done by the database:

• Primary operation: Scanning all rows in the employees table once.
• How many times: Once for each query, but grouping work depends on number of rows.

How Execution Grows With Input

The database reads each row and places it into a group based on column values.

Input Size (n)	Approx. Operations
10	About 10 grouping steps
100	About 100 grouping steps
1000	About 1000 grouping steps

Pattern observation: The work grows roughly in direct proportion to the number of rows.

Final Time Complexity

Time Complexity: O(n)

This means the time to group grows linearly as the number of rows increases.

Common Mistake

[X] Wrong: "Grouping by more columns always makes the query much slower in a way that grows faster than the number of rows."

[OK] Correct: Grouping time mainly depends on scanning rows once; adding more columns changes grouping keys but does not multiply the number of rows processed.

Interview Connect

Understanding how grouping scales helps you explain query performance clearly and shows you know how databases handle data.

Self-Check

What if we added an ORDER BY after GROUP BY? How would that affect the time complexity?

Practice

1. What does the GROUP BY clause do in a PostgreSQL query?

easy

A. It updates values in the specified columns.

B. It sorts the rows in ascending order.

C. It deletes duplicate rows from the table.

D. It groups rows that have the same values in specified columns.

Solution

1. Step 1: Understand the purpose of GROUP BY

   The GROUP BY clause collects rows that share the same value in one or more columns into groups.

2. Step 2: Differentiate from other clauses

   Unlike ORDER BY which sorts rows, GROUP BY organizes rows into groups for aggregation.

3. Final Answer:

   It groups rows that have the same values in specified columns. -> Option D

4. Quick Check:

   GROUP BY = groups rows by column values [OK]

Hint: GROUP BY collects rows sharing column values into groups [OK]

Common Mistakes:

• Confusing GROUP BY with ORDER BY
• Thinking GROUP BY deletes duplicates
• Assuming GROUP BY updates data

2. Which of the following is the correct syntax to group data by two columns, category and region?

easy

A. SELECT category, region, COUNT(*) FROM sales GROUP BY category, region;

B. SELECT category, region, COUNT(*) FROM sales GROUP BY category region;

C. SELECT category, region, COUNT(*) FROM sales GROUP BY (category, region);

D. SELECT category, region, COUNT(*) FROM sales GROUP category, region;

Solution

1. Step 1: Check correct GROUP BY syntax

   In PostgreSQL, multiple columns are separated by commas in GROUP BY without parentheses.

2. Step 2: Validate each option

   SELECT category, region, COUNT(*) FROM sales GROUP BY category, region; uses commas correctly; SELECT category, region, COUNT(*) FROM sales GROUP BY category region; misses comma; SELECT category, region, COUNT(*) FROM sales GROUP BY (category, region); uses parentheses incorrectly; SELECT category, region, COUNT(*) FROM sales GROUP category, region; misses BY keyword.

3. Final Answer:

   SELECT category, region, COUNT(*) FROM sales GROUP BY category, region; -> Option A

4. Quick Check:

   GROUP BY columns separated by commas [OK]

Hint: Separate multiple columns with commas in GROUP BY [OK]

Common Mistakes:

• Omitting commas between columns
• Using parentheses around columns
• Forgetting the BY keyword

3. Given the table orders with columns customer_id, product_id, and quantity, what will this query return?

SELECT customer_id, SUM(quantity) FROM orders GROUP BY customer_id;

medium

A. Total quantity ordered by each customer.

B. Total quantity ordered by each product.

C. Total quantity ordered by each customer and product.

D. Syntax error due to missing GROUP BY columns.

Solution

1. Step 1: Analyze the SELECT and GROUP BY columns

   The query groups rows by customer_id and sums quantity for each group.

2. Step 2: Understand the aggregation result

   SUM(quantity) calculates total quantity per customer_id, ignoring product_id.

3. Final Answer:

   Total quantity ordered by each customer. -> Option A

4. Quick Check:

   GROUP BY customer_id sums quantity per customer [OK]

Hint: GROUP BY column matches SELECT non-aggregates [OK]

Common Mistakes:

• Assuming product_id affects grouping
• Thinking query returns per product totals
• Confusing aggregation with filtering

4. Identify the error in this query:

SELECT product_id, region, COUNT(*) FROM sales GROUP BY product_id;

medium

A. COUNT(*) cannot be used with GROUP BY.

B. Missing region in GROUP BY causes error.

C. product_id cannot be grouped alone.

D. Syntax error due to missing FROM clause.

Solution

1. Step 1: Check SELECT columns vs GROUP BY columns

   SELECT includes product_id and region, but GROUP BY only has product_id.

2. Step 2: Understand PostgreSQL grouping rules

   All non-aggregated columns in SELECT must appear in GROUP BY; region is missing.

3. Final Answer:

   Missing region in GROUP BY causes error. -> Option B

4. Quick Check:

   All SELECT non-aggregates must be in GROUP BY [OK]

Hint: Include all non-aggregated SELECT columns in GROUP BY [OK]

Common Mistakes:

• Forgetting to add all columns to GROUP BY
• Thinking COUNT(*) disallows grouping
• Ignoring SELECT and GROUP BY mismatch

5. You want to find the total sales quantity for each product in each region from the sales table with columns product_id, region, and quantity. Which query correctly groups and sums the data?

hard

A. SELECT product_id, SUM(quantity) FROM sales GROUP BY product_id;

B. SELECT product_id, region, SUM(quantity) FROM sales GROUP BY product_id;

C. SELECT product_id, region, SUM(quantity) FROM sales GROUP BY product_id, region;

D. SELECT product_id, region, SUM(quantity) FROM sales;

Solution

1. Step 1: Identify grouping needs

   We want totals per product and per region, so both columns must be grouped.

2. Step 2: Check each option's GROUP BY clause

   SELECT product_id, region, SUM(quantity) FROM sales GROUP BY product_id, region; groups by product_id and region, matching SELECT columns; others miss region or GROUP BY entirely.

3. Final Answer:

   SELECT product_id, region, SUM(quantity) FROM sales GROUP BY product_id, region; -> Option C

4. Quick Check:

   Group by all non-aggregated columns for correct totals [OK]

Hint: Group by all selected non-aggregated columns [OK]

Common Mistakes:

• Grouping by only one column when selecting multiple
• Omitting GROUP BY causing aggregation errors
• Selecting columns without aggregation or grouping