Word Search II (leetcode 212)

Word Search I 用基本的DFS就可以做出,由于僅找一個(gè)單詞躯畴,難度不高刨晴。
而Word Search II 是要找一系列的單詞。對DFS的難度增大不少茫死。

II的要點(diǎn)是:

  1. 用Trie來建立需要尋找的單詞數(shù)組
  2. 在DFS時(shí)跪但,一定要先判斷TrieNode->mp[char] 是否存在,如果存在進(jìn)而拿到下一個(gè)TrieNode峦萎。不存在則return屡久,要先拿到TrieNode,再進(jìn)入下一個(gè)DFS loop
  3. 用set<string> 記錄string爱榔,避免重復(fù)被环。而且當(dāng)拿到一個(gè)string時(shí),不能返回搓蚪!因?yàn)檫€有可能要繼續(xù)搜索.
class TrieNode{
public:
    bool isWord;
    unordered_map<char, TrieNode*> mp;
    TrieNode() : isWord(false){}
};

class Trie{
public:
    Trie(){ root = new TrieNode(); }
    TrieNode *getRoot(){
        return root;
    }
    void insertWord(string cur){
        TrieNode *p = root;
        for(int i=0; i<cur.length(); i++){
            if(!p->mp.count(cur[i])){
                p->mp[cur[i]] = new TrieNode();
            }
            p = p->mp[cur[i]];
        }
        p->isWord = true;
    }
private:
    TrieNode *root;
 
};

class Solution {
public:

    void dfs(vector<vector<char>>& board, set<string> &allcomb, string &comb, vector<vector<bool>> &visited, TrieNode *p, int i, int j){
        
        int row = board.size(), col = board[0].size();
        comb.push_back(board[i][j]);
        visited[i][j] = true;
        if(p->isWord) {allcomb.insert(comb);}
        
        for(auto it : directions){
            int x = i + it.first, y = j + it.second;
            if(x < 0 || x >= row || y < 0 || y >= col || visited[x][y]) continue;
            if(p->mp.count(board[x][y])){
                dfs(board, allcomb, comb, visited, p->mp[board[x][y]], x, y);
            }
        }
        comb.pop_back();
        visited[i][j] = false;
    }

    vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
        set<string> allcomb;
        if(board.empty() || board[0].empty()) return vector<string>();
        int row = board.size(), col = board[0].size();
        myTrie = new Trie();
        for(string s : words){
            myTrie->insertWord(s);
        }
        string comb;
        vector<vector<bool>> visited(row, vector<bool>(col, false));
        TrieNode *root = myTrie->getRoot();
        for(int i=0; i<row; i++){
            for(int j=0; j<col; j++){
                if(!root->mp.count(board[i][j])) continue;
                dfs(board, allcomb, comb, visited, root->mp[board[i][j]], i, j);
            }
        }
        return vector<string>(allcomb.begin(), allcomb.end());
    }
private:
    Trie *myTrie;
     vector<pair<int, int>> directions{{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
};
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市丁鹉,隨后出現(xiàn)的幾起案子妒潭,更是在濱河造成了極大的恐慌,老刑警劉巖揣钦,帶你破解...
    沈念sama閱讀 212,332評論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件雳灾,死亡現(xiàn)場離奇詭異,居然都是意外死亡冯凹,警方通過查閱死者的電腦和手機(jī)谎亩,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,508評論 3 385
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來宇姚,“玉大人匈庭,你說我怎么就攤上這事』肜停” “怎么了阱持?”我有些...
    開封第一講書人閱讀 157,812評論 0 348
  • 文/不壞的土叔 我叫張陵,是天一觀的道長魔熏。 經(jīng)常有香客問我衷咽,道長,這世上最難降的妖魔是什么蒜绽? 我笑而不...
    開封第一講書人閱讀 56,607評論 1 284
  • 正文 為了忘掉前任镶骗,我火速辦了婚禮,結(jié)果婚禮上躲雅,老公的妹妹穿的比我還像新娘鼎姊。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,728評論 6 386
  • 文/花漫 我一把揭開白布此蜈。 她就那樣靜靜地躺著即横,像睡著了一般。 火紅的嫁衣襯著肌膚如雪裆赵。 梳的紋絲不亂的頭發(fā)上东囚,一...
    開封第一講書人閱讀 49,919評論 1 290
  • 那天,我揣著相機(jī)與錄音战授,去河邊找鬼页藻。 笑死,一個(gè)胖子當(dāng)著我的面吹牛植兰,可吹牛的內(nèi)容都是我干的份帐。 我是一名探鬼主播,決...
    沈念sama閱讀 39,071評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼楣导,長吁一口氣:“原來是場噩夢啊……” “哼废境!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起筒繁,我...
    開封第一講書人閱讀 37,802評論 0 268
  • 序言:老撾萬榮一對情侶失蹤噩凹,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后毡咏,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體驮宴,經(jīng)...
    沈念sama閱讀 44,256評論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,576評論 2 327
  • 正文 我和宋清朗相戀三年呕缭,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了堵泽。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,712評論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡恢总,死狀恐怖迎罗,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情片仿,我是刑警寧澤佳谦,帶...
    沈念sama閱讀 34,389評論 4 332
  • 正文 年R本政府宣布,位于F島的核電站滋戳,受9級特大地震影響钻蔑,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜奸鸯,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 40,032評論 3 316
  • 文/蒙蒙 一咪笑、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧娄涩,春花似錦窗怒、人聲如沸映跟。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,798評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽努隙。三九已至,卻和暖如春辜昵,著一層夾襖步出監(jiān)牢的瞬間荸镊,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,026評論 1 266
  • 我被黑心中介騙來泰國打工堪置, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留躬存,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 46,473評論 2 360
  • 正文 我出身青樓舀锨,卻偏偏與公主長得像岭洲,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子坎匿,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,606評論 2 350

推薦閱讀更多精彩內(nèi)容