Solução Balões

Solução por Roger Benet, comentário de João Guilherme

Inicialmente salvamos as alturas em um set e também guardamos sua frequência. Então para cada altura, vemos se a frequência dela é maior que 0, se for nós aumentamos a resposta e vemos quais balões serão estourados por essa agulha, reduzindo sua frequência.

Segue código para melhor entendimento.

https://gist.github.com/jogu99/4fb01c7a7a72ced06aff5678c79ae4ff