概述

使用拼音或拼音首字母進行搜索匹配，可以減少輸入，提高交互效率。在查找和搜索時帶來效率提升，在 iOS中，app本地搜索功能，需要實現(xiàn)拼音或拼音首字母匹配。本文從算法流程及實現(xiàn)中遇到的問題來闡述，并附上源碼實現(xiàn)。

需求

輸入拼音或拼音首字母，匹配對應漢字字符。若原字符串中對應有非漢字，則進行原始字符匹配。
比如:中華民族 ，輸入:zhonghua或zh,匹配輸出:中華；zhan勝困難，輸入:zhansheng或zhans,匹配輸出:zhan勝；
為了減少用戶的輸入，支持不完全輸入搜索。即最后一個拼音不用完整輸入，比如::中華民族 ，輸入:zhongh,匹配輸出:中華。

算法設計

首先，需要將漢字轉化為拼音
蘋果 CFString.h 中提供了將字符轉化為帶音標的拼音的函數(shù)：
CFStringTransform(CFMutableStringRef string, CFRange *range, CFStringRef transform, Boolean reverse)
可以將漢字先轉化為帶音標的拼音，然后將音標去除：
- (NSString *)stringByFoldingWithOptions:(NSStringCompareOptions)options locale:(nullable NSLocale *)locale

第2步，進入匹配流程

1. 準備工作：
   由于中文拼音的復雜性，以及原始字符串可能是中英文混合字符串，所以構造一個輔助數(shù)組，存儲每個字符的拼音；另外構造一個數(shù)組，存儲每個字符的首個拼音字母。
   將轉化后的拼音拼接在一起，組成一個完整的字符串拼音
   用圖表示的流程如下：

   
   
   生成拼音及拼音首字母

根據(jù)原始字符串得到拼音、拼音首字母及單個字符拼音組成的數(shù)組。
比如,只含有漢字、漢字與英文混合等樣式的字符串經過轉換，可以得到下列結果。

2.2 正式匹配
首先將待匹配的字符串轉換為大寫，通過待匹配的字符串（M）的首字母，去步驟 2.1 中得到的拼音首字母字符串(LA)中定位到位置i，然后借助拼音數(shù)組SA,尋找第i個位置的字符串，從第i個位置開始尋找最少個數(shù)(len)的字符串，拼接組成的字符串以M為起始字符串。
尋找成功，則匹配的字符串位于原始字符串i，長度為len；若拼接子字符串長度大于或等于待匹配字符串，或直到LA最后一個字符串，仍然找不到拼接的字符串能以M為起始字符串。則需要在拼音首字母字符串(LA)中重新定位，在位置i后尋找新的位置，重新上一個步驟。直至遍歷到LA的末尾字符，仍然沒有找到，則匹配失敗。
以2.1中的表最后一個字符串為例，搜索: banxing.

3. 校驗并匹配
   根據(jù)生成的拼音或拼音首字母，當待匹配字符串在上述2個字符串中其中一個里面存在時，再進行匹配.

遇到的問題

之前，借助漢字轉拼音的第三方庫[NSString+Extensional]，將整個原始字符串通過函數(shù)pinyinForSort作為整體轉換得到拼音,由于該函數(shù)返回的字符串，相鄰的漢字拼音使用空格' '分割，因此可以進一步使用NSString 的函數(shù):

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

得到拼音字符數(shù)組SA。不過針對中英文混合的字符串，相鄰的英文和中文會連在一起，中間沒有空格分割。導致使用該庫的函數(shù)firstLettersForSort得到的拼音首字母字符串FS長度與SA長度不一致。給后續(xù)匹配造成困難。
下標列出全部轉換為大寫后的結果:

詳細代碼

將代碼放在了Github spellMatch,若有不足，歡迎指正。有幫助的話，不吝star。