ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LC-212. 单词搜索 II

2021-09-18 10:33:14  阅读:180  来源: 互联网

标签:单词 TreeNode LC int word II board 212 root


  1. 单词搜索 II

给定一个 m x n 二维字符网格 board 和一个单词(字符串)列表 words,找出所有同时在二维网格和字典中出现的单词。

单词必须按照字母顺序,通过 相邻的单元格 内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。

示例 1:
image

输入:board = [["o","a","a","n"],["e","t","a","e"],["i","h","k","r"],["i","f","l","v"]], words = ["oath","pea","eat","rain"]
输出:["eat","oath"]

类似于前缀树,在建树的时候只在叶子结点存储string,每个节点都是map存储26个字符所对应的指针

class Solution {
private:
    set<string> res;
    int m, n;
    struct TreeNode {
        map<char, TreeNode*> child;
        string word;
        TreeNode () {
            word = "";
        }
    };

public:
    void insertTree (TreeNode* root, string word) {
        TreeNode *node = root;
        for (char &k : word) {
	    //这里表示新建节点
            if (!node -> child.count(k)) {
                node -> child[k] = new TreeNode();
            }
            node = node ->child[k];
        }
	//在叶子节点出存储string
        node -> word = word;
    }

    int dir[4][2] = {{1, 0}, {-1, 0}, {0, -1}, {0, 1}};
    void dfs (vector<vector<char>>& board, int i, int j, TreeNode* root) {
        char ch = board[i][j];
        if (!root -> child.count(ch)) {
            return;
        }
	//到达这里就表明存在这个单词的这个字符,而且存储的字符串长度大于0,表示可以生成这个单词
        root = root -> child[ch];
        if (root -> word.size() > 0) {
            res.insert(root -> word);
        }
        board[i][j] = '#';
        for (int k = 0; k < 4; k++) {
            int xx = i + dir[k][0], yy = j + dir[k][1];
            if (xx < 0 || xx >= m || yy < 0 || yy >= n || board[xx][yy] == '#') {

            }else {
                dfs (board, xx, yy, root);
            }
        }
        board[i][j] = ch;
    }
    vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
        TreeNode *root = new TreeNode();
        for (string &k : words) {
            insertTree(root, k);
        }
        m = board.size(), n = board[0].size();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                dfs(board, i, j, root);
            }
        }
        vector<string> ans(res.begin(), res.end());
        return ans;
    }
};

标签:单词,TreeNode,LC,int,word,II,board,212,root
来源: https://www.cnblogs.com/smilerain/p/15307446.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有