less comparision when a_elem == b_elem

[lengyijun]

When a_elem == b_elem, alway from the diagonal value.

Moreover, this repo needs cargo fmt for the convenience of further pull requests.