Onto Functions — Solution & Editorial

Back to the Problem

Inclusion–exclusion: ∑_j=0..K (−1)^j C(K, j)(K−j)^N. Precompute factorials for the binomials.

Complexity: O(K log N)