【回溯 字典树（前缀树）】212. 单词搜索 II

最新推荐文章于 2024-08-28 23:27:41 发布
最新推荐文章于 2024-08-28 23:27:41 发布
阅读量613 收藏 24
点赞数 25
分类专栏： 数据结构与算法 文章标签： 算法 力扣 c++ 字典树 回溯 单词 前缀树
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/he_zhidan/article/details/138540626
版权

本文涉及知识点

回溯 字典树（前缀树）

LeetCode212. 单词搜索 II

给定一个 m x n 二维字符网格 board 和一个单词（字符串）列表 words， 返回所有二维网格上的单词 。
单词必须按照字母顺序，通过 相邻的单元格 内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。
示例 1：
在这里插入图片描述

输入：board = [[“o”,“a”,“a”,“n”],[“e”,“t”,“a”,“e”],[“i”,“h”,“k”,“r”],[“i”,“f”,“l”,“v”]], words = [“oath”,“pea”,“eat”,“rain”]
输出：[“eat”,“oath”]
示例 2：
在这里插入图片描述

输入：board = [[“a”,“b”],[“c”,“d”]], words = [“abcb”]
输出：[]

提示：

m == board.length
n == board[i].length
1 <= m, n <= 12
board[i][j] 是一个小写英文字母
1 <= words.length <= 3 * 104
1 <= words[i].length <= 10
words[i] 由小写英文字母组成
words 中的所有字符串互不相同

回溯

用字典树存储所有words。
枚举起点，时间复杂度O(nm)。
最长的words[i]，除去第一个字符后，最多9个字符。后续字符4个。故处理每个起点的时间复杂是O(9n)
总时间复杂度 ≈ \approx ≈ 3$\times$107 超时边缘。
回溯部分一定要反复优化。

代码

核心代码

struct CVec
{
	int r;
	int c;
	CVec operator*(const int len)const
	{
		return { r * len,c * len };
	}
};

struct CPos
{	
	int r = 0, c = 0;
	CPos(int tr, int tc) {
		r = tr;
		c = tc;
	}
	int ToMask()const { return s_MaxC * r + c; };
	bool operator==(const CPos& o)const
	{
		return (r == o.r) && (c == o.c);
	}
	CPos operator+(const CVec& v)const
	{
		return { r + v.r, c + v.c };
	}
	CPos operator-(const CVec& v)const
	{
		return{ r - v.r, c - v.c };
	}
	CVec operator-(const CPos& o)const
	{
		return {r - o.r,c- o.c};
	}
	inline static  int s_MaxC = 10'000;
};

struct SPosHash {
	// 哈希函数的操作符重载，接受一个指针作为参数
	size_t operator()(const CPos& pos) const {
		// 简单的哈希函数示例，将指针地址转换为哈希值
		return (long long)100'000 * pos.r + pos.c;;
	}
};

class CRange
{
public:
	CRange(int rCount, int cCount, std::function<bool(int, int)> funVilidCur):m_r(rCount),m_c(cCount), m_funVilidCur(funVilidCur)
	{

	}
	bool Vilid(CPos pos)const
	{
		return (pos.r >= 0) && (pos.r < m_r) && (pos.c >= 0) && (pos.c < m_c) && m_funVilidCur(pos.r, pos.c);
	}
	const int m_r, m_c;
protected:
	std::function<bool(int, int)> m_funVilidCur;
};

template<class TData = char, int iTypeNum = 26, TData cBegin = 'a'>
class CTrieNode
{
public:
	CTrieNode* AddChar(TData ele, int& iMaxID)
	{
#ifdef _DEBUG
		if ((ele < cBegin) || (ele >= cBegin + iTypeNum))
		{
			return nullptr;
		}
#endif
		const int index = ele - cBegin;
		auto ptr = m_vPChilds[ele - cBegin];
		if (!ptr)
		{
			m_vPChilds[index] = new CTrieNode();
#ifdef _DEBUG
			m_vPChilds[index]->m_iID = ++iMaxID;
			m_childForDebug[ele] = m_vPChilds[index];
#endif
		}
		return m_vPChilds[index];
	}
	CTrieNode* GetChild(TData ele)const
	{
#ifdef _DEBUG
		if ((ele < cBegin) || (ele >= cBegin + iTypeNum))
		{
			return nullptr;
		}
#endif
		return m_vPChilds[ele - cBegin];
	}
protected:
#ifdef _DEBUG
	int m_iID = -1;
	std::unordered_map<TData, CTrieNode*> m_childForDebug;
#endif
public:
	int m_iLeafIndex = -1;
protected:
	CTrieNode* m_vPChilds[iTypeNum] = { nullptr };
};

template<class TData = char, int iTypeNum = 26, TData cBegin = 'a'>
class CTrie
{
public:
	int GetLeadCount()
	{
		return m_iLeafCount;
	}
	CTrieNode<TData, iTypeNum, cBegin>* AddA(CTrieNode<TData, iTypeNum, cBegin>* par,TData curValue)
	{
		auto curNode =par->AddChar(curValue, m_iMaxID);
		FreshLeafIndex(curNode);
		return curNode;
	}
	template<class IT>
	int Add(IT begin, IT end)
	{
		auto pNode = &m_root;
		for (; begin != end; ++begin)
		{
			pNode = pNode->AddChar(*begin, m_iMaxID);
		}
		FreshLeafIndex(pNode);
		return pNode->m_iLeafIndex;
	}	
	template<class IT>
	CTrieNode<TData, iTypeNum, cBegin>* Search(IT begin, IT end)
	{
		auto ptr = &m_root;
		for (; begin != end; ++begin)
		{
			ptr = ptr->GetChild(*begin);
			if (nullptr == ptr)
			{
				return nullptr;
			}
		}
		return ptr;
	}
	CTrieNode<TData, iTypeNum, cBegin> m_root;
protected:
	void FreshLeafIndex(CTrieNode<TData, iTypeNum, cBegin>* pNode)
	{
		if (-1 == pNode->m_iLeafIndex)
		{
			pNode->m_iLeafIndex = m_iLeafCount++;
		}
	}
	int m_iMaxID = 0;
	int m_iLeafCount = 0;
};

class Solution {
public:
	vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
		const int R = board.size();
		const int C = board[0].size();
		for (const auto& s : words) {
			m_trie.Add(s.begin(), s.end());
		}		
		vector<bool> vHas(m_trie.GetLeadCount());
		int hasVisit[12][12] ;
		memset(hasVisit, 0, sizeof(hasVisit));
		CRange rang(R, C, [&](int r, int c) {return !hasVisit[r][c]; });
		int move[][2] = { {0,1},{0,-1},{1,0},{-1,0}};
		std::function<void(CTrieNode<>* par, int r, int c)> BackTrack = [&](CTrieNode<>* par, int r, int c)
		{
			auto p = par->GetChild(board[r][c]);
			if (nullptr == p) { return; }
			if (-1 != p->m_iLeafIndex) {
				vHas[p->m_iLeafIndex] = true;
			}
			hasVisit[r][c]=true;
			for (int i = 0; i < 4; i++) {
				const int r1 = r + move[i][0];
				const int c1 = c + move[i][1];
				if(!rang.Vilid(CPos(r1,c1 ))){continue;}
				BackTrack(p, r1, c1);
			}
			hasVisit[r][c] = false;
		};
		for (int r = 0; r < R; r++) {
			for (int c = 0; c < C; c++) {
				BackTrack(&m_trie.m_root, r, c);
			}
		}
		vector<string> vRet;
		for (const auto& s : words) {
			auto p = m_trie.Search(s.begin(), s.end());
			if (vHas[p->m_iLeafIndex]) {
				vRet.emplace_back(s);
			}
		}
		return vRet;
	}	
	CTrie<> m_trie;
};

测试用例

template<class T>
void Assert(const vector<T>& v1, const vector<T>& v2)
{
	if (v1.size() != v2.size())
	{
		assert(false);
		return;
	}
	for (int i = 0; i < v1.size(); i++)
	{
		assert(v1[i] == v2[i]);
	}
}

template<class T>
void Assert(const T& t1, const T& t2)
{
	assert(t1 == t2);
}

int main()
{
	vector<vector<char>> board;
	vector<string> words;
	{
		Solution slu;
		board= { {'o','a','a','n'},{'e','t','a','e'},{'i','h','k','r'},{'i','f','l','v'} },words= { "oath","pea","eat","rain" };
		auto res = slu.findWords(board, words);
		Assert({ "oath","eat" }, res);
	}
	{
		Solution slu;
		board = { {'a','b'},{'c','d'} }, words = { "abcb" };
		auto res = slu.findWords(board, words);
		Assert({  }, res);
	}	
	
}

2023年4月版

class Solution {
public:
	vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
		m_r = board.size();
		m_c = board[0].size();
		m_iMaskNum = m_r*m_c;
		m_vMove.resize(m_iMaskNum);
		
		for (int r = 0; r < m_r; r++)
		{
			for (int c = 0; c < m_c; c++)
			{
				int iMask = m_c*r + c;
				if (r > 0)
				{
					m_vMove[iMask].emplace_back(iMask - m_c);
				}
				if (r + 1 < m_r)
				{
					m_vMove[iMask].emplace_back(iMask + m_c);
				}
				if ( c > 0 )
				{
					m_vMove[iMask].emplace_back(iMask - 1);
				}
				if (c + 1 < m_c)
				{
					m_vMove[iMask].emplace_back(iMask + 1);
				}
			}
		}
		for (const auto& s : words)
		{
			long long llRet = 0;
			for (const auto& ch : s)
			{
				llRet = llRet * 27 + ch - 'a'+1;
				m_setNeedSub.emplace(llRet);
			}
			m_setNeed.emplace(llRet);
		}
		int v[12] = { 0 };
		bool bPre[12 * 12] = { 0 };
		for (int i = 0; i < m_iMaskNum; i++)
		{			
			const char& ch = board[i / m_c][i%m_c];
			v[0] = i;
			bPre[i] = true;
			dfs(v, 1,bPre, ch - 'a' + 1, board);
			
			bPre[i] = false;
		}


		vector<string> vRet;
		for (const auto& s : words)
		{
			const long long llMask = Mask(s);
			if (!m_setNeed.count(llMask))
			{
				vRet.emplace_back(s);
			}
		}
		return vRet;
	}

	void dfs(int* v,int iSize, bool* vPre, const long long llMask, vector<vector<char>>& board)
	{
		if (iSize > 10)
		{
			return;
		}
		
		if (m_setNeed.count(llMask))
		{
			m_setNeed.erase(llMask);
			auto tmp = llMask;
			while (tmp > 0)
			{
				auto it = m_setNeedSub.find(tmp);
				m_setNeedSub.erase(it);
				tmp /= 27;
			}
		}

		if (!m_setNeedSub.count(llMask))
		{
			return;
		}
		//const int iSize = v.size();
		const int iCur = v[iSize-1];
		for (const auto& next : m_vMove[iCur])
		{
			if (vPre[next])
			{
				continue;
			}
			v[iSize] = next;
			vPre[next] = true;
			dfs(v, iSize+1,vPre, llMask * 27 + board[next / m_c][next%m_c] - 'a' + 1, board);
		
			vPre[next] = false;
		}

	}
	long long Mask(const std::string& str)
	{
		long long llRet = 0;
		for (const auto& ch : str)
		{
			llRet = llRet * 27 + ch - 'a'+1;
		}
		return llRet;
	}
	int m_r, m_c;
	int m_iMaskNum;
	vector<vector<int>> m_vMove;
	std::unordered_multiset<long long> m_setNeedSub;
	std::unordered_set<long long> m_setNeed;
};

扩展阅读

视频课程

有效学习：明确的目标 及时的反馈 拉伸区（难度合适），可以先学简单的课程，请移步CSDN学院，听白银讲师（也就是鄙人）的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了，为老板分忧，请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

相关下载

想高屋建瓴的学习算法，请下载《喜缺全书算法册》doc版
https://download.csdn.net/download/he_zhidan/88348653

我想对大家说的话
《喜缺全书算法册》以原理、正确性证明、总结为主。
闻缺陷则喜是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛

测试环境

操作系统：win7 开发环境： VS2019 C++17
或者 操作系统：win10 开发环境： VS2022 C++17
如无特殊说明，本算法用**C++**实现。

闻缺陷则喜何志丹
关注
  • 25
    点赞
  • 踩
  • 24
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 4
    评论
专栏目录
leetcode: 212.单词搜索II
uncle_ll的博客
09-16 162
题目 来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/word-search-ii 给定一个 m x n 二维字符网格 board 和一个单词（字符串）列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过 相邻的单元格 内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例 1： 输入：board = [[“o”,“a”,“a”,“n”],[“e
LeetCode_前缀树_困难_212.单词搜索 II
代码星辰的博客
06-11 872
给定一个 m x n 二维字符网格 board 和一个单词（字符串）列表 words， 返回所有二维网格上的单词 。单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。
LeetCode 212. 单词搜索 II 字典树 回溯 C++
BeforeEasy的博客
07-15 645
描述 给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = [“oath”,“pea”,“eat”,“rain”] and board = [ [‘o’,‘...
前缀树、字典树、Trie | 回溯 ：力扣212. 单词搜索 II
weixin_42042056的博客
09-14 245
1、题目描述： 实现Trie（前缀树） 2、题解： 3、复杂度分析：
前缀和+字典树 208. 实现 Trie (前缀树) 字典树+dfs 212. 单词搜索 II
BLUEsang的博客
05-12 206
208. 实现 Trie (前缀树) 实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。 示例: Trie trie = new Trie(); trie.insert("apple"); trie.search("apple"); // 返回 true trie.search("app"); // 返回 false trie.startsWith("app"); // 返回 true trie.insert("app"); tr
leetocde212. 单词搜索 II/前缀树
yyf_blog
09-18 165
文章目录212. 单词搜索 II前缀树 212. 单词搜索 II 给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = ["oath","pea","eat","rain"] and board = [ ['o','a','a','n'], ['e','t'
leetcode 212. 单词搜索 II
Qin_酱的博客
08-02 149
212. 单词搜索 II 给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = [“oath”,“pea”,“eat”,“rain”] and board = [ [‘o’,‘a’,‘a’,‘n’], [‘e’,‘t’,‘a’,‘e’], [‘i’,‘h’,‘k’,‘
LeetCode 212. 单词搜索 II（Word Search II）
botke的博客
11-04 329
LeetCode 212. 单词搜索 II（Word Search II） 题目的????：3082 在这里插入代码片 题目的????：130 1、问题描述 给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = ["oath","pea","eat","ra
Leetcode 212. 单词搜索 II 字典树+DFS
Evildoer_llc的博客
03-01 261
给定一个二维网格board和一个字典中的单词列表words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = ["oath","pea","eat","rain"] and board = [...
leetcode-140. 单词拆分 II (字典树/dp + 回溯法) + 字节测开字典树算法题
sxtopc的博客
05-22 515
给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，在字符串中增加空格来构建一个句子，使得句子中所有的单词都在词典中。返回所有这些可能的句子。 说明： 分隔时可以重复使用字典中的单词。 你可以假设字典中没有重复的单词。 示例 1： 输入: s = "catsanddog" wordDict = ["cat", "cats", "and", "sand", "dog"] 输出: [ "cats and dog", "cat sand dog" ] 示例 2： 输入: s =...
Trie树(前缀树、字典树)
u011764940的博客
12-16 3043
目录 什么是前缀树 前缀树的优缺点： 前缀树的应用 什么是前缀树 Trie树，又叫字典树、前缀树（Prefix Tree）、单词查找树 或 键树，是一种多叉树结构。如下图： 好比假设有b，abc，abd，bcd，abcd，efg，hii 这6个单词,那我们创建trie树就得到 上图可以归纳出 Trie 树的基本性质： 1. 根节点不包含字符，除根节点外的每一个子节点都包含一个字符。 2. 从根节点到某一个节点，路径上经过的字符连接起来，为该节点对应的字符串。 3. 每个节点的所有子节
【LeetCode】212. 单词搜索 II
HeavenDan的博客
09-22 214
一、题目 给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = ["oath","pea","eat","rain"] and board = [ ['o','a','a','n'], ['e','t','a','e'], ['i','h','k','r
leetcode 212. 单词搜索 II Java 实现
陆上溪流
09-06 304
关键词 回溯 trie(前缀树) 题目描述 给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。 单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。 示例: 输入: words = ["oath","pea","eat","rain"] a...
Leetcode42接雨水（单调栈）
AGNING的博客
08-28 261
维护一个栈，满足从栈底到栈顶元素大小递减。如果当前元素高于栈顶，且栈内有至少2个元素，则形成了一个凹槽，及一个图中横着的红色矩形，将它的面积累加即可，第一次写难在维护单调栈中计算面积和。求出前缀最大和后缀最大，用两者较小值减去当前高度，累加即可，这个思路容易想到，这里不赘述。
代码随想录算法训练营第62天|Floyd 算法精讲、A * 算法精讲 （A star算法）
Yinems的博客
08-28 156
分两种情况，一种是节点 i 到节点 j 的最短路径经过节点 k，grid[i][j][k] = grid[i][k][k-1] + grid[k][j][k-1]；另一种情况，节点 i 到节点 j 的最短路径不经过节点 k，grid[i][j][k] = grip[i][j][k-1]；刚开始初始化 k 是不确定的，如果赋值为1，那么不能知道节点 i 到节点 j 经过1个节点的最短距离，所以只能把 k 赋值为0，本题中节点0是无意义的，节点是从1到n。本题求的是最小值，所以其余值初始化为最大值。
深度学习（DL）算法分分类
最新发布
m0_46171719的博客
08-28 226
深度学习（DL）算法是机器学习的一个子集，专注于使用深度神经网络（DNN）来模拟人脑的学习过程，从而解决复杂的数据处理和模式识别问题。
冒泡排序算法
m0_74051652的博客
08-28 273
冒泡排序是一种简单的排序算法，它重复地遍历要排序的数列，一次比较两个元素，如果它们的顺序错误就把它们交换过来。遍历数列的工作是重复进行的，直到没有再需要交换的元素为止，这意味着数列已经排序完成。这个算法的名字由来是因为越小的元素会经由交换慢慢“浮”到数列的顶端（或说是排序的开头位置），就如同水中的气泡一样上升到水面。冒泡排序的写法可以简单分为两种方式，分别是：前同后异、前异后同。
【算法】深入浅出聚类算法：原理、应用与Java实现
一个不断前行的程序者
08-26 881
聚类算法是一种将数据集中的对象自动划分为多个子集的算法。每个子集称为一个簇，簇内的对象具有较高的相似度，而簇与簇之间的相似度较低。聚类算法的主要目的是发现数据中的自然结构，而不是通过已知的类别标签。本文介绍了聚类算法的原理、应用场景以及如何在Java中实现一个简单的聚类算法。通过聚类算法，我们可以发现数据集中的自然结构，有助于更好地理解数据。在实际应用中，可以根据具体需求选择合适的聚类算法。希望本文能够帮助读者更好地理解和应用聚类算法。
字典树与KMP算法解析
在字典树中，每个节点代表字符串的一个前缀，从根节点到任意节点的路径上的边标签组合起来就构成了一个完整的字符串。例如，题目中的例1展示了如何根据给定的单词列表构建字典树。在这个例子中，我们逐步插入单词，...
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

闻缺陷则喜何志丹

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值