概述

使用拼音或拼音首字母进行搜索匹配，可以减少输入，提高交互效率。在查找和搜索时带来效率提升，在 iOS中，app本地搜索功能，需要实现拼音或拼音首字母匹配。本文从算法流程及实现中遇到的问题来阐述，并附上源码实现。

需求

输入拼音或拼音首字母，匹配对应汉字字符。若原字符串中对应有非汉字，则进行原始字符匹配。
比如:中华民族 ，输入:zhonghua或zh,匹配输出:中华；zhan胜困难，输入:zhansheng或zhans,匹配输出:zhan胜；
为了减少用户的输入，支持不完全输入搜索。即最后一个拼音不用完整输入，比如::中华民族 ，输入:zhongh,匹配输出:中华。

算法设计

首先，需要将汉字转化为拼音
苹果 CFString.h 中提供了将字符转化为带音标的拼音的函数：
CFStringTransform(CFMutableStringRef string, CFRange *range, CFStringRef transform, Boolean reverse)
可以将汉字先转化为带音标的拼音，然后将音标去除：
- (NSString *)stringByFoldingWithOptions:(NSStringCompareOptions)options locale:(nullable NSLocale *)locale

第2步，进入匹配流程

1. 准备工作：
   由于中文拼音的复杂性，以及原始字符串可能是中英文混合字符串，所以构造一个辅助数组，存储每个字符的拼音；另外构造一个数组，存储每个字符的首个拼音字母。
   将转化后的拼音拼接在一起，组成一个完整的字符串拼音
   用图表示的流程如下：

   
   生成拼音及拼音首字母

根据原始字符串得到拼音、拼音首字母及单个字符拼音组成的数组。
比如,只含有汉字、汉字与英文混合等样式的字符串经过转换，可以得到下列结果。

字符串	拼音	首字母	拼音组成的数组
夜来风	YELAIFENG	YLF	YE,LAI,FENG
ALC花	ALCHUA	ALCH	A,L,C,HUA
半梦半醒	BANMENGBANXING	BMBX	BAN,MENG,BAN,XING

2.2 正式匹配
首先将待匹配的字符串转换为大写，通过待匹配的字符串（M）的首字母，去步骤 2.1 中得到的拼音首字母字符串(LA)中定位到位置i，然后借助拼音数组SA,寻找第i个位置的字符串，从第i个位置开始寻找最少个数(len)的字符串，拼接组成的字符串以M为起始字符串。
寻找成功，则匹配的字符串位于原始字符串i，长度为len；若拼接子字符串长度大于等于待匹配字符串，或直到LA最后一个字符串，仍然找不到拼接的字符串能以M为起始字符串。则需要在拼音首字母字符串(LA)中重新定位，在位置i后寻找新的位置，重新上一个步骤。直至遍历到LA的末尾字符，仍然没有找到，则匹配失败。
以2.1中的表最后一个字符串为例，搜索: banxing.

步骤	操作内容
1	banxing 转换为大写-> BANXING
2	BANXING 取首字母 -> B
3	BMBX 中匹配 B，得到位置NSRange r1 =（0,1）
4	(BAN,MENG,BAN,XING)第一个位置起，匹配含有 BANXING 的最短拼接字符串
4.1	取第一个拼音字符串，一个元素，得到 BAN，不包含BANXING
4.2	取第一个拼音字符串，2个元素，得到 BANMENG，不包含BANXING的前缀
4.3	判断拼接的拼音子字符串BANMENG长度，为7，与BANXING的长度相等，结束此轮匹配
5	BMBXBFS 中，从第一个B后面，即第二个位置开始匹配 B，得到位置NSRange r2 =（1,1），新的起始位置为 r1.location+r2.location+r2.length=0+1+1=2
5.1	(BAN,MENG,BAN,XING)第2个位置起，匹配含有 BANXING 的最短拼接字符串
5.2	取第2个拼音字符串(一个元素)，得到 BAN，不包含BANXING
5.3	取第一个拼音字符串，2个元素，得到 BANXING，
5.4	判断拼接的拼音子字符串 BANXING 的长度，与待匹配字符串长度一致，继续。
5.5	拼接的 字符串 BANXING 包含 BANXING 的前缀。 匹配的长度为2
5.6	返回原字符串r3=（2,2）,即从第2个元素开始2个长度的字符串。即: 半醒

3. 校验并匹配
   根据生成的拼音或拼音首字母，当待匹配字符串在上述2个字符串中其中一个里面存在时，再进行匹配.

遇到的问题

之前，借助汉字转拼音的第三方库[NSString+Extensional]，将整个原始字符串通过函数pinyinForSort作为整体转换得到拼音,由于该函数返回的字符串，相邻的汉字拼音使用空格' '分割，因此可以进一步使用NSString 的函数:

- (NSArray<NSString *> *)componentsSeparatedByString:(NSString *)separator

得到拼音字符数组SA。不过针对中英文混合的字符串，相邻的英文和中文会连在一起，中间没有空格分割。导致使用该库的函数firstLettersForSort得到的拼音首字母字符串FS长度与SA长度不一致。给后续匹配造成困难。
下标列出全部转换为大写后的结果:

	库函数结果	修改后函数结果
原始字符串	YDB4测试服7	YDB4测试服7
字符串拼音	YDB4CESHIFU7	YDB4CESHIFU7
字符串拼音首字母	YDB4CSF7	YDB4CSF7
字符串拼音数组	YDB4CE,SHI,FU7	Y,D,B,4,CE,SHI,FU,7

详细代码

将代码放在了Github spellMatch,若有不足，欢迎指正。有帮助的话，不吝star。