# Basic comprehension
squares = [x**2 for x in range(5)]  # [0, 1, 4, 9, 16]

# With condition
evens = [x for x in range(10) if x % 2 == 0]  # [0, 2, 4, 6, 8]

# Nested with transformation
matrix = [[1, 2], [3, 4]]
flattened = [num for row in matrix for num in row]  # [1, 2, 3, 4]

# Equivalent to loop (interview comparison)
result = []
for x in range(5):
    result.append(x**2)
# result = [0, 1, 4, 9, 16]  # Same as first example

# Example: Solve "Find max in list" problem step-by-step
# Step 1: Understand - Input: list of nums; Output: max value; Constraints: empty list?

def find_max(numbers):
    if not numbers:  # Step 2: Handle edge case (empty list)
        return None  # Or raise ValueError

    max_val = numbers  # Step 3: Initialize with first element
    for num in numbers[1:]:  # Step 4: Loop through rest (sub-problem: compare)
        if num > max_val:
            max_val = num
    return max_val  # Step 5: Return result

# Step 6: Test cases
print(find_max([3, 1, 4, 1, 5]))  # Output: 5
print(find_max([]))  # Output: None
print(find_max())  # Output: 10

# If stuck: Comment code to trace, or simplify (e.g., use max() built-in first to verify)

# Basic for loop over iterable (list)
fruits = ["apple", "banana", "cherry"]
for fruit in fruits:  # Iterates each element directly
    print(fruit)      # Output: apple \n banana \n cherry

# For loop with range() for numeric sequences
for i in range(3):    # Generates 0, 1, 2 (start=0, stop=3, step=1)
    print(i)          # Output: 0 \n 1 \n 2

for i in range(1, 6, 2):  # Start=1, stop=6, step=2
    print(i)              # Output: 1 \n 3 \n 5

# Index-aware with enumerate() (gets both index and value)
for index, fruit in enumerate(fruits, start=1):  # start=1 for 1-based indexing
    print(f"{index}: {fruit}")                   # Output: 1: apple \n 2: banana \n 3: cherry

# Parallel iteration with zip() (pairs multiple iterables)
names = ["Alice", "Bob", "Charlie"]
ages = [25, 30, 35]
for name, age in zip(names, ages):  # Stops at shortest iterable
    print(f"{name} is {age} years old")  # Output: Alice is 25 years old \n Bob is 30 years old \n Charlie is 35 years old

# Nested for loops (outer for rows, inner for columns; e.g., matrix)
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
for row in matrix:         # Outer: each sublist
    for num in row:        # Inner: each element in row
        print(num, end=' ')  # Output: 1 2 3 4 5 6 7 8 9 (space-separated)

# For loop in list comprehension (concise iteration with optional condition)
squares = [x**2 for x in range(5)]  # Basic comprehension
print(squares)  # Output: [0, 1, 4, 9, 16]

evens_squared = [x**2 for x in range(10) if x % 2 == 0]  # With condition (if)
print(evens_squared)  # Output: [0, 4, 16, 36, 64]

# Nested comprehension (flattens 2D list)
flattened = [num for row in matrix for num in row]  # Equivalent to nested for
print(flattened)  # Output: [1, 2, 3, 4, 5, 6, 7, 8, 9]

# Basic linear search (O(n) - naive approach)
def linear_search(arr, target):
    for i in range(len(arr)):
        if arr[i] == target:
            return i
    return -1

nums = [1, 3, 5, 7, 9]
print(linear_search(nums, 5))  # Output: 2

# Binary search (O(log n) - efficient for sorted arrays)
def binary_search(arr, target):
    left, right = 0, len(arr) - 1
    while left <= right:  # Divide range until found or empty
        mid = (left + right) // 2
        if arr[mid] == target:
            return mid
        elif arr[mid] < target:
            left = mid + 1  # Search right half
        else:
            right = mid - 1  # Search left half
    return -1

sorted_nums = [1, 3, 5, 7, 9]
print(binary_search(sorted_nums, 5))  # Output: 2
print(binary_search(sorted_nums, 6))  # Output: -1 (not found)

# Edge cases
print(binary_search([], 1))      # Output: -1 (empty list)
print(binary_search(, 1))     # Output: 0 (single element)

# For loop: Use for fixed iterations over iterables (e.g., processing lists)
fruits = ["apple", "banana", "cherry"]
for fruit in fruits:  # Iterates each element
    print(fruit)      # Output: apple \n banana \n cherry

for i in range(3):    # Numeric sequence (start=0, stop=3)
    print(i)          # Output: 0 \n 1 \n 2

# While loop: Use when iterations depend on a dynamic condition (e.g., user input, convergence)
count = 0
while count < 3:      # Runs as long as condition is True
    print(count)
    count += 1       # Increment to avoid infinite loop! Output: 0 \n 1 \n 2

# Nested loops: Use for 2D data (e.g., matrices, grids); outer for rows, inner for columns
matrix = [[1, 2], [3, 4]]
for row in matrix:    # Outer: each sublist
    for num in row:   # Inner: elements in row
        print(num)    # Output: 1 \n 2 \n 3 \n 4

# Control statements: break (exit loop), continue (skip iteration)
for i in range(5):
    if i == 2:
        continue    # Skip 2
    if i == 4:
        break       # Exit at 4
    print(i)        # Output: 0 \n 1 \n 3

# List comprehension: Concise for loop alternative (use for simple transformations/filtering)
squares = [x**2 for x in range(5) if x % 2 == 0]  # Even squares
print(squares)  # Output: [0, 4, 16]

import collections  

# defaultdict - Eliminate key errors with auto-initialization  
from collections import defaultdict  
gradebook = defaultdict(int)  
gradebook['Alice'] += 95  
print(gradebook['Alice'])  # Output: 95  
print(gradebook['Bob'])    # Output: 0  

# defaultdict for grouping operations  
anagrams = defaultdict(list)  
words = ["eat", "tea", "tan"]  
for w in words:  
    key = ''.join(sorted(w))  
    anagrams[key].append(w)  
print(anagrams['aet'])  # Output: ['eat', 'tea']  

# Counter - Frequency analysis in one line  
from collections import Counter  
text = "abracadabra"  
freq = Counter(text)  
print(freq['a'])          # Output: 5  
print(freq.most_common(2)) # Output: [('a', 5), ('b', 2)]  

# Counter arithmetic for problem-solving  
inventory = Counter(apples=10, oranges=5)  
sales = Counter(apples=3, oranges=2)  
print(inventory - sales)  # Output: Counter({'apples': 7, 'oranges': 3})  

# namedtuple - Self-documenting data structures  
from collections import namedtuple  
Employee = namedtuple('Employee', 'name role salary')  
dev = Employee('Alex', 'Developer', 95000)  
print(dev.role)           # Output: Developer  
print(dev[2])             # Output: 95000  

# deque - Optimal for BFS and sliding windows  
from collections import deque  
queue = deque([1, 2, 3])  
queue.append(4)  
queue.popleft()  
print(queue)              # Output: deque([2, 3, 4])  
queue.rotate(1)  
print(queue)              # Output: deque([4, 2, 3])  

# OrderedDict - Track insertion order (LRU cache essential)  
from collections import OrderedDict  
cache = OrderedDict()  
cache['A'] = 1  
cache['B'] = 2  
cache.move_to_end('A')  
cache.popitem(last=False)  
print(list(cache.keys())) # Output: ['B', 'A']  

# ChainMap - Manage layered configurations  
from collections import ChainMap  
defaults = {'theme': 'dark', 'font': 'Arial'}  
user_prefs = {'theme': 'light'}  
settings = ChainMap(user_prefs, defaults)  
print(settings['font'])   # Output: Arial  

# Practical Interview Tip: Anagram detection  
print(Counter("secure") == Counter("rescue"))  # Output: True  

# Pro Tip: Sliding window maximum  
def max_sliding_window(nums, k):  
    dq, result = deque(), []  
    for i, n in enumerate(nums):  
        while dq and nums[dq[-1]] < n:  
            dq.pop()  
        dq.append(i)  
        if dq[0] == i - k:  
            dq.popleft()  
        if i >= k - 1:  
            result.append(nums[dq[0]])  
    return result  
print(max_sliding_window([1,3,-1,-3,5,3,6,7], 3))  # Output: [3,3,5,5,6,7]  

# Expert Move: Custom LRU Cache implementation  
class LRUCache:  
    def __init__(self, capacity):  
        self.cache = OrderedDict()  
        self.capacity = capacity  
    def get(self, key):  
        if key not in self.cache:  
            return -1  
        self.cache.move_to_end(key)  
        return self.cache[key]  
    def put(self, key, value):  
        if key in self.cache:  
            del self.cache[key]  
        self.cache[key] = value  
        if len(self.cache) > self.capacity:  
            self.cache.popitem(last=False)  
cache = LRUCache(2)  
cache.put(1, 10)  
cache.put(2, 20)  
cache.get(1)  
cache.put(3, 30)  
print(list(cache.cache.keys()))  # Output: [2, 1, 3] → Wait! Correction: Should be [1, 3] (capacity=2 triggers eviction of '2')  

# Bonus: Multiset operations with Counter  
primes = Counter([2, 3, 5, 7])  
odds = Counter([1, 3, 5, 7, 9])  
print(primes | odds)  # Output: Counter({3:1, 5:1, 7:1, 2:1, 9:1, 1:1})  

# SQLAlchemy Setup - The industry standard ORM
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, relationship

# Configure database connection
engine = create_engine('sqlite:///company.db', echo=True)
Base = declarative_base()
Session = sessionmaker(bind=engine)
session = Session()

# Model Definition - Translate tables to Python classes
class Department(Base):
    __tablename__ = 'departments'
    id = Column(Integer, primary_key=True)
    name = Column(String(50), nullable=False)
    # One-to-Many relationship
    employees = relationship("Employee", back_populates="department")

class Employee(Base):
    __tablename__ = 'employees'
    id = Column(Integer, primary_key=True)
    name = Column(String(100))
    email = Column(String(100), unique=True)
    # Foreign Key
    department_id = Column(Integer, ForeignKey('departments.id'))
    # Relationship back-reference
    department = relationship("Department", back_populates="employees")

# Create tables in database
Base.metadata.create_all(engine)

# CRUD Operations - Core interview competency
# CREATE
hr = Department(name="HR")
session.add(hr)
session.commit()

alice = Employee(name="Alice", email="alice@company.com", department=hr)
session.add(alice)
session.flush()  # Assigns ID without committing
print(alice.id)  # Output: 1

# READ
employee = session.query(Employee).filter_by(name="Alice").first()
print(employee.department.name)  # Output: "HR"

# UPDATE
employee.email = "alice.smith@company.com"
session.commit()

# DELETE
session.delete(employee)
session.commit()

# Advanced Querying - Solve complex data challenges
from sqlalchemy import or_, and_, func

# Filter combinations
active_employees = session.query(Employee).filter(
    Employee.name.like('A%'),
    or_(Employee.email.endswith('@company.com'), Employee.id < 10)
)

# Aggregation
dept_count = session.query(
    Department.name, 
    func.count(Employee.id)
).join(Employee).group_by(Department.id).all()
print(dept_count)  # Output: [('HR', 1), ('Engineering', 5)]

# Pagination (critical for web apps)
page_2 = session.query(Employee).limit(10).offset(10).all()

# Relationship Handling - Avoid N+1 query disasters
# LAZY LOADING (default - causes N+1 problem)
for dept in session.query(Department):
    print(dept.employees)  # Triggers separate query per department

# EAGER LOADING (interview gold)
from sqlalchemy.orm import joinedload

depts = session.query(Department).options(
    joinedload(Department.employees)
).all()
print(len(session.identity_map))  # Output: 6 (1 query for all data)

# Many-to-Many Relationships - Real-world schema design
# Association table
employee_projects = Table('employee_projects', Base.metadata,
    Column('employee_id', Integer, ForeignKey('employees.id')),
    Column('project_id', Integer, ForeignKey('projects.id'))
)

class Project(Base):
    __tablename__ = 'projects'
    id = Column(Integer, primary_key=True)
    name = Column(String(100))
    # Many-to-Many
    members = relationship("Employee", secondary=employee_projects)

# Add employee to project
project = Project(name="AI Initiative")
project.members.append(alice)
session.commit()

# Transactions - Atomic operations for data integrity
from sqlalchemy.exc import SQLAlchemyError

try:
    with session.begin():
        alice = Employee(name="Alice", email="alice@company.com")
        session.add(alice)
        # Automatic rollback if error occurs
        raise ValueError("Simulated error")
except ValueError:
    print(session.query(Employee).count())  # Output: 0 (no partial data)