Freaks City

This problem involves calculating valid colorings of a tree under certain constraints. The solution uses inclusion-exclution principle combined with polynomial multiplication via Number Theoretic Transform (NTT). Basic Approach We approach the problem by computing the complement: count arrangements where at least one node violates the coloring ...