cp_library/math/table/factor_cnts_cls.py

• View this file on GitHub
• Last update: 2025-03-30 20:17:47+09:00

Code

class FactorCounts(list[tuple[int,int]]):
    def __init__(self, N: int):
        pairs = []
        d = 2
        while d*d<=N:
            while N % d == 0:
                match pairs:
                    case [*_, (f,cnt)] if f == d:
                        pairs[-1] = (f,cnt+1)
                    case _:
                        pairs.append((d, 1))
                N //= d
            d += 1
        super().__init__(pairs)

class FactorCounts(list[tuple[int,int]]):
    def __init__(self, N: int):
        pairs = []
        d = 2
        while d*d<=N:
            while N % d == 0:
                match pairs:
                    case [*_, (f,cnt)] if f == d:
                        pairs[-1] = (f,cnt+1)
                    case _:
                        pairs.append((d, 1))
                N //= d
            d += 1
        super().__init__(pairs)

Back to top page