Comment by darshanmakwana

Ahhh! section A uses the classic good old gumble max trick[1]

[1] - https://darshanmakwana412.github.io/2026/01/gumbel-max-trick...