概述
使用拼音或拼音首字母进行搜索匹配，可以减少输入，提高交互效率。在查找和搜索时带来效率提升，在 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步，进入匹配流程
-
准备工作：
由于中文拼音的复杂性，以及原始字符串可能是中英文混合字符串，所以构造一个辅助数组，存储每个字符的拼音；另外构造一个数组，存储每个字符的首个拼音字母。
将转化后的拼音拼接在一起，组成一个完整的字符串拼音
用图表示的流程如下：
生成拼音及拼音首字母
根据原始字符串得到拼音、拼音首字母及单个字符拼音组成的数组。
比如,只含有汉字、汉字与英文混合等样式的字符串经过转换，可以得到下列结果。
| 字符串（M） | 拼音（Str） | 首字母（LA） | 拼音组成的数组（SA） |
|---|---|---|---|
| 夜来风 | 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个长度的字符串。即: 半醒 |
- 校验并匹配
根据生成的拼音或拼音首字母，当待匹配字符串在上述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。
