Improve the speed of atom mapping search