面试题 17.13. 恢复空格(C++)---Tire(字典树) + 动态规划 解题

题目详情

哦，不！你不小心把一个长篇文章中的空格、标点都删掉了，并且大写也弄成了小写。像句子"I reset the computer. It still didn’t boot!"已经变成了"iresetthecomputeritstilldidntboot"。在处理标点符号和大小写之前，你得先把它断成词语。当然了，你有一本厚厚的词典dictionary，不过，有些词没在词典里。假设文章用sentence表示，设计一个算法，把文章断开，要求未识别的字符最少，返回未识别的字符数。

注意：本题相对原题稍作改动，只需返回未识别的字符数

 

示例：

输入：
dictionary = ["looked","just","like","her","brother"]
sentence = "jesslookedjustliketimherbrother"
输出： 7
解释： 断句后为"jess looked just like tim her brother"，共7个未识别字符。

提示：

• 0 <= len(sentence) <= 1000
• dictionary中总字符数不超过 150000。
• 你可以认为dictionary和sentence中只包含小写字母。

 

——题目难度:中等


 

---

分析

先将词典中所有的单词「反序」插入字典树中

——来源：力扣官方题解

然后定义 dp[i] 表示考虑前 i 个字符最少的未识别的字符数量，从前往后计算 dp 值，最后返回 dp[n] 即可。

 

---

-下面代码

class Tire {
public:
	Tire* next[26];
	bool isEnd;
	
	Tire() {
		memset(next, 0, 26 * sizeof(Tire*));
		isEnd = false;
	}
	
	void insert(string s) {
		Tire* curPos = this;
		for(int i = s.size() - 1; i >= 0; i--) {
			int t = s[i] - 'a';
			if (!curPos->next[t]) {
				curPos->next[t] = new Tire();
			}
			curPos = curPos->next[t];
		}
		curPos->isEnd = true;
	}
};
	
class Solution {
public:
    int respace(vector& dictionary, string sentence) {
		int n = sentence.size();
		
		Tire* root = new Tire();
		for(string& word : dictionary) {
			root->insert(word);
		}
		
		vector dp(n + 1);
		dp[0] = 0;
		for(int i = 1; i <= n; i++) {
			dp[i] = dp[i - 1] + 1;
			
			Tire* curPos = root;
			for(int j = i; j >=1; j--) {
				int t = sentence[j - 1] - 'a';
				if (!curPos->next[t]) {
					break;
				}
				
				if (curPos->next[t]->isEnd) {
					dp[i] = min(dp[i], dp[j - 1]);
				}
				
				if (dp[i] == 0) {
					break;
				}
				
				curPos = curPos->next[t];
			}
		}
		
		return dp[n];
    }
};

结果