Suggestions needed: speed up optimization

[stevengj]

@DShiu

I revised the function be based on comprehension

This will allocate an array on the heap each time it is called, which is bad for performance.