[FEATURE] Incorporate indexed priority queue in query grouping logic

Is your feature request related to a problem?

A clear and concise description of what the problem is, e.g. I'm always frustrated when [...]. Currently we use a priority queue, incorporate indexed priority queue in the grouping logic to make the time complexity go from O(n) to O(logn).

What solution would you like?

A clear and concise description of what you want to happen.

What alternatives have you considered?

A clear and concise description of any alternative solutions or features you've considered.

Do you have any additional context?

Add any other context or screenshots about the feature request here.