Python Code Example

from collections import defaultdict
from itertools import product
from itertools import accumulate

import numpy as np
from numpy.random import choice

import re
import random

from collections import Counter
from functools import reduce
import math

# Define ANSI escape codes for red text
RED = "\033[91m"
GREEN = "\033[92m"
YELLOW = "\033[93m"
ORANGE = "\033[38;5;208m"
L_ORANGE = "\033[38;5;214m"
RESET = "\033[0m"