Math + Logic Heavy DSA Problem (Onsite Coding)

A math- and logic-heavy DSA coding problem during the onsite coding round. Requires deriving the logic and implementing a working solution. Note: Missing exact statement/IO/constraints, so it cannot be reconstructed.