Topological Sort Implementation in Python

10 min readintermediateUpdated 2026-03-01

NexusBro EditorialDeveloper Tooling ResearchUpdated 2026-03-01

Key Takeaways

✓Python's readability makes it ideal for learning Topological Sort.
✓The implementation achieves O(V + E) average time complexity.
✓Python's built-in data structures complement Topological Sort implementations.
✓Type hints improve code clarity and catch bugs early.

Topological Sort in Python: Overview

Python is an excellent language for implementing Topological Sort thanks to its readable syntax and rich standard library. It produces a linear ordering of vertices in a directed acyclic graph (DAG) such that for every edge u -> v, u appears before v. It is fundamental for dependency resolution and scheduling. In this tutorial, we implement Topological Sort in Python using idiomatic patterns, type hints, and thorough documentation. Python's dynamic typing and list comprehensions allow for concise implementations, while type hints from the typing module provide additional safety. We also leverage Python's built-in testing capabilities to verify correctness. Whether you are a beginner learning algorithms or an experienced developer preparing for interviews, this Python implementation will serve as a clear reference.

Python Implementation

Below is a clean, idiomatic Python implementation of Topological Sort. The code uses type hints for clarity and includes docstrings explaining each function's purpose. Note how Python's syntax makes the algorithm's logic transparent and easy to follow. Each function is designed to be self-contained and testable.

python

from collections import deque

def topological_sort(n: int, edges: list[tuple[int, int]]) -> list[int] | None:
    """Kahn's algorithm. Returns None if cycle exists."""
    graph: dict[int, list[int]] = {i: [] for i in range(n)}
    in_degree = [0] * n
    for u, v in edges:
        graph[u].append(v)
        in_degree[v] += 1
    queue = deque(i for i in range(n) if in_degree[i] == 0)
    result: list[int] = []
    while queue:
        node = queue.popleft()
        result.append(node)
        for neighbor in graph[node]:
            in_degree[neighbor] -= 1
            if in_degree[neighbor] == 0:
                queue.append(neighbor)
    return result if len(result) == n else None  # None if cycle

edges = [(5, 2), (5, 0), (4, 0), (4, 1), (2, 3), (3, 1)]
print(topological_sort(6, edges))  # [4, 5, 2, 0, 3, 1] or similar

Step-by-Step Explanation

Let us walk through how this Python implementation works: Step 1: Compute in-degree for each vertex — In our Python code, this corresponds to the initial setup and input validation. Step 2: Add all vertices with in-degree 0 to a queue — In our Python code, this corresponds to the core loop body where the main logic executes. Step 3: While queue is not empty: dequeue vertex, add to result — In our Python code, this corresponds to the core loop body where the main logic executes. Step 4: Decrement in-degree of all neighbors — In our Python code, this corresponds to the core loop body where the main logic executes. Step 5: If any neighbor reaches in-degree 0, enqueue it — In our Python code, this corresponds to the final return statement and result construction. Python's list slicing and built-in functions simplify several of these steps compared to lower-level languages. The implementation prioritizes clarity over micro-optimization, making it ideal for learning and interview settings.

Did You Get the Big O Right? NexusBro Will Tell You in Seconds.

Paste your algorithm. Get complexity analysis, edge cases, and optimizations.

Test My Algorithm

Complexity Analysis

This Python implementation of Topological Sort has the following complexity profile: - Time: Best O(V + E), Average O(V + E), Worst O(V + E) - Space: O(V) Python's overhead from dynamic typing and interpreted execution means constant factors are higher than compiled languages. However, for algorithm analysis, the asymptotic complexity remains the same. For performance-critical applications, consider using NumPy arrays or Cython extensions. Python's built-in sort uses TimSort, which achieves O(n log n) worst case and excels on partially sorted data. Understanding how Topological Sort compares to built-in methods helps you make informed decisions in production code.

Testing Your Implementation

Always test your Topological Sort implementation with a variety of inputs: 1. Empty input: Verify the function handles empty lists or None gracefully. 2. Single element: Ensure no errors occur with minimal input. 3. Already processed input: Check that the algorithm handles best-case scenarios correctly. 4. Reverse order: Test worst-case behavior. 5. Duplicates: Verify correct handling of repeated values. 6. Large inputs: Stress test with thousands of elements to confirm acceptable performance. Python's unittest and pytest frameworks make it easy to write comprehensive test suites. Use parameterized tests to cover many cases with minimal code duplication. Property-based testing with Hypothesis can also uncover edge cases you might not think of manually.

Python-Specific Optimizations

Python offers several ways to optimize your Topological Sort implementation: - Use collections.deque for O(1) append and pop operations from both ends. - Leverage itertools for memory-efficient iteration over large datasets. - Use functools.lru_cache for memoization in recursive implementations. - Consider array.array for type-homogeneous numeric data with lower memory overhead. - Profile with cProfile before optimizing to identify actual bottlenecks. Remember that premature optimization is the root of all evil. Start with a correct, readable implementation, measure performance, and optimize only where necessary. Python's ecosystem provides excellent profiling and benchmarking tools that make this process straightforward.

Unlock Unlimited QA Audits for $15.99/mo

Free: 5 audits/day. Pro $15.99/mo: 50/day + 250 pages. Pro Max $99/mo: unlimited audits, 10K pages, API access.

See Plans

Frequently Asked Questions

Is Python good for implementing Topological Sort?

Yes, Python is excellent for learning and implementing Topological Sort. Its readable syntax makes the algorithm logic clear, and its standard library provides useful supporting data structures. While Python is slower than compiled languages, the asymptotic complexity is identical, making it perfect for understanding and interviews.

How does Python's built-in sort compare to Topological Sort?

Python's built-in sort uses TimSort, a hybrid merge-sort and insertion-sort algorithm with O(n log n) worst case. Depending on Topological Sort's complexity class, it may be faster or slower for specific inputs. Built-in sort is highly optimized in C, so it will outperform pure Python implementations.

Should I use type hints in my Topological Sort Python code?

Yes, type hints improve code readability, enable better IDE support, and help catch type-related bugs early. They are especially valuable in algorithm implementations where the types of inputs and outputs should be clear to readers.

Can I use Topological Sort in Python for large datasets?

For large datasets, consider the algorithm's complexity. If Topological Sort has O(V + E) worst case, it may be slow for very large inputs. Python's NumPy and Pandas libraries offer optimized C-based alternatives for data-heavy operations.

What Python version should I use for Topological Sort?

Use Python 3.10 or later for the best experience. Recent versions offer structural pattern matching, improved type hints, and performance improvements that benefit algorithm implementations.

Share this article

X LinkedIn Reddit WhatsApp

Topological Sort Complete Guide Topological Sort Typescript Topological Sort Java Depth First Search Complete Guide

Unlock Unlimited QA Audits for $15.99/mo

Free: 5 audits/day. Pro $15.99/mo: 50/day + 250 pages. Pro Max $99/mo: unlimited audits, 10K pages, API access.

See Plans

Noizz helps you discover and compare the best new products and tools. Try it free →

Is YOUR site's SEO this optimized?

Find out in 60 seconds with a free QA audit.

Free SEO Check

Is your site built to last?

Run a free QA audit and get your Site Health Score in seconds.

Check Your Site Free

No signup required

QA Score Checker·Compare Sites·Industry Benchmarks

Topological Sort in Python: Overview

Python Implementation

python

from collections import deque

def topological_sort(n: int, edges: list[tuple[int, int]]) -> list[int] | None:
    """Kahn's algorithm. Returns None if cycle exists."""
    graph: dict[int, list[int]] = {i: [] for i in range(n)}
    in_degree = [0] * n
    for u, v in edges:
        graph[u].append(v)
        in_degree[v] += 1
    queue = deque(i for i in range(n) if in_degree[i] == 0)
    result: list[int] = []
    while queue:
        node = queue.popleft()
        result.append(node)
        for neighbor in graph[node]:
            in_degree[neighbor] -= 1
            if in_degree[neighbor] == 0:
                queue.append(neighbor)
    return result if len(result) == n else None  # None if cycle

edges = [(5, 2), (5, 0), (4, 0), (4, 1), (2, 3), (3, 1)]
print(topological_sort(6, edges))  # [4, 5, 2, 0, 3, 1] or similar

Step-by-Step Explanation

Complexity Analysis

Testing Your Implementation

Python-Specific Optimizations

Frequently Asked Questions

Is Python good for implementing Topological Sort?

How does Python's built-in sort compare to Topological Sort?

Should I use type hints in my Topological Sort Python code?

Can I use Topological Sort in Python for large datasets?

What Python version should I use for Topological Sort?

Use Python 3.10 or later for the best experience. Recent versions offer structural pattern matching, improved type hints, and performance improvements that benefit algorithm implementations.

Key Takeaways

Topological Sort in Python: Overview

Python Implementation

Step-by-Step Explanation

Did You Get the Big O Right? NexusBro Will Tell You in Seconds.

Complexity Analysis

Testing Your Implementation

Python-Specific Optimizations

Unlock Unlimited QA Audits for $15.99/mo

Frequently Asked Questions

Share this article

Related Articles

Unlock Unlimited QA Audits for $15.99/mo

Is your site built to last?

How does your site compare?

Explore More Topics

The Complete Guide to Python Variables

The Complete Guide to TypeScript Strict Mode

The Complete Guide to Swift Optionals

The Complete Guide to SELECT Queries

The Definitive Git Basics Guide for Developers

The Complete PHP Fundamentals Developer Guide

Key Takeaways

Topological Sort in Python: Overview

Python Implementation

Step-by-Step Explanation

Did You Get the Big O Right? NexusBro Will Tell You in Seconds.

Complexity Analysis

Testing Your Implementation

Python-Specific Optimizations

Unlock Unlimited QA Audits for $15.99/mo

Frequently Asked Questions

Share this article

Related Articles

Unlock Unlimited QA Audits for $15.99/mo

Is your site built to last?

How does your site compare?

Explore More Topics

The Complete Guide to Python Variables

The Complete Guide to TypeScript Strict Mode

The Complete Guide to Swift Optionals

The Complete Guide to SELECT Queries

The Definitive Git Basics Guide for Developers

The Complete PHP Fundamentals Developer Guide