Improve default search UI organization

[ang-zeyu]

Is your request related to a problem?

Currently, searches for a term that has multiple occurences in a document display like so:

Describe the solution you'd like
This can be improved by organizing search results that belong to the same page, like how algolia does:

We can follow this intuitive design, appending keywords to the right side if any

--page name--
  | --heading 1-- | --heading 1--
                    --keyword 1--
                    --keyword 2--
                         ...
  | --heading 2-- | --heading 2--
  | --heading 3-- | --heading 3--
  | --heading 4-- | --heading 4--

As an improvement, the typeahead can also only highlight the matched words on the right side, instead of both sides as seen in the image.

Describe alternatives you've considered
na