sub group_hero_better {
    use bigint;
    
    my @nums = sort { $a <=> $b } @_;
    
    my $sum = 0;
    for my $i ( 0 .. $#nums ) {
        my $s = 0;
        for my $j ( 0 .. $i - 1 ) {
            $s += $nums[ $j ] * 2 ** ( $i - $j - 1 )
        }
        $sum += ( $s + $nums[ $i ]) * $nums[ $i ] ** 2
    }
    return $sum
}