Leetcode 212. Word Search II

Given a 2D board and a list of words from the dictionary, find all words in the board.
Each word must be constructed from letters of sequentially adjacent cell, where "adjacent" cells are those horizontally or vertically neighboring. The same letter cell may not be used more than once in a word.

For example,
Given words = ["oath","pea","eat","rain"] and board =
[
['o','a','a','n'],
['e','t','a','e'],
['i','h','k','r'],
['i','f','l','v']
]
Return ["eat","oath"].

思路:
首先想到的是如何在一個board中搜索一個單詞失晴。以board中每個字符嘗試作為單詞首部涂屁,進行深度搜索灰伟。由于board每個字符拆又,在一個單詞中最多允許用一次帖族,因此需要一個boolean數(shù)組作為標記數(shù)組挡爵。
知道了如果搜索一個單詞,那么遍歷words進行搜索就能得到答案捻激。
題目有一個test case是["a", "a"] 和 [['a']]前计,返回結(jié)果只能是["a"],因此為了滿足此case男杈,需要事先對words進行去重。

public List<String> findWords(char[][] board, String[] words) {
    Set<String> set = new HashSet<>();
    for (String word : words) {
        set.add(word);
    }

    List<String> res = new ArrayList<>();
    for (String word : set) {
        if (findWord(board, word)) {
            res.add(word);
        }
    }

    return res;
}

private boolean findWord(char[][] board, String word) {
    int m = board.length, n = board[0].length;
    boolean[][] used = new boolean[m][n];

    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (findHelper(board, used, word, 0, i, j)) {
                return true;
            }
        }
    }

    return false;
}

private boolean findHelper(char[][] board, boolean[][] used, String word, int pos, int x, int y) {
    if (pos == word.length()) {
        return true;
    }

    int m = board.length, n = board[0].length;
    if (x < 0 || x >= m || y < 0 || y >= n) {
        return false;
    }

    if (board[x][y] != word.charAt(pos) || used[x][y]) {
        return false;
    }

    used[x][y] = true;
    boolean res = findHelper(board, used, word, pos + 1, x + 1, y) ||
            findHelper(board, used, word, pos + 1, x - 1, y) ||
            findHelper(board, used, word, pos + 1, x, y + 1) ||
            findHelper(board, used, word, pos + 1, x, y - 1);
    used[x][y] = false;
    return res;
}

Trie樹優(yōu)化版本

public List<String> findWords(char[][] board, String[] words) {
List<String> res = new ArrayList<>();
TrieNode root = buildTrie(words);
for (int i = 0; i < board.length; i++) {
    for (int j = 0; j < board[0].length; j++) {
        dfs (board, i, j, root, res);
    }
}
return res;
}

public void dfs(char[][] board, int i, int j, TrieNode p, List<String> res) {
char c = board[i][j];
if (c == '#' || p.next[c - 'a'] == null) return;
p = p.next[c - 'a'];
if (p.word != null) {   // found one
    res.add(p.word);
    p.word = null;     // de-duplicate
}

board[i][j] = '#';
if (i > 0) dfs(board, i - 1, j ,p, res); 
if (j > 0) dfs(board, i, j - 1, p, res);
if (i < board.length - 1) dfs(board, i + 1, j, p, res); 
if (j < board[0].length - 1) dfs(board, i, j + 1, p, res); 
board[i][j] = c;
}

public TrieNode buildTrie(String[] words) {
TrieNode root = new TrieNode();
for (String w : words) {
    TrieNode p = root;
    for (char c : w.toCharArray()) {
        int i = c - 'a';
        if (p.next[i] == null) p.next[i] = new TrieNode();
        p = p.next[i];
   }
   p.word = w;
}
return root;
}

class TrieNode {
TrieNode[] next = new TrieNode[26];
String word;
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末先蒋,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子眯搭,更是在濱河造成了極大的恐慌业岁,老刑警劉巖鳞仙,帶你破解...
    沈念sama閱讀 217,185評論 6 503
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件棍好,死亡現(xiàn)場離奇詭異允耿,居然都是意外死亡右犹,警方通過查閱死者的電腦和手機提澎,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,652評論 3 393
  • 文/潘曉璐 我一進店門盼忌,熙熙樓的掌柜王于貴愁眉苦臉地迎上來谦纱,“玉大人君编,你說我怎么就攤上這事§裟耍” “怎么了亮瓷?”我有些...
    開封第一講書人閱讀 163,524評論 0 353
  • 文/不壞的土叔 我叫張陵降瞳,是天一觀的道長蚓胸。 經(jīng)常有香客問我沛膳,道長汛聚,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,339評論 1 293
  • 正文 為了忘掉前任八毯,我火速辦了婚禮,結(jié)果婚禮上芯侥,老公的妹妹穿的比我還像新娘柱查。我一直安慰自己云石,他們只是感情好淋硝,可當我...
    茶點故事閱讀 67,387評論 6 391
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著铅乡,像睡著了一般。 火紅的嫁衣襯著肌膚如雪阵幸。 梳的紋絲不亂的頭發(fā)上花履,一...
    開封第一講書人閱讀 51,287評論 1 301
  • 那天,我揣著相機與錄音挚赊,去河邊找鬼诡壁。 笑死,一個胖子當著我的面吹牛咬腕,可吹牛的內(nèi)容都是我干的欢峰。 我是一名探鬼主播,決...
    沈念sama閱讀 40,130評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼纽帖!你這毒婦竟也來了宠漩?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,985評論 0 275
  • 序言:老撾萬榮一對情侶失蹤懊直,失蹤者是張志新(化名)和其女友劉穎扒吁,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體室囊,經(jīng)...
    沈念sama閱讀 45,420評論 1 313
  • 正文 獨居荒郊野嶺守林人離奇死亡雕崩,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,617評論 3 334
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了融撞。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片盼铁。...
    茶點故事閱讀 39,779評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖尝偎,靈堂內(nèi)的尸體忽然破棺而出饶火,到底是詐尸還是另有隱情,我是刑警寧澤致扯,帶...
    沈念sama閱讀 35,477評論 5 345
  • 正文 年R本政府宣布肤寝,位于F島的核電站,受9級特大地震影響抖僵,放射性物質(zhì)發(fā)生泄漏鲤看。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,088評論 3 328
  • 文/蒙蒙 一耍群、第九天 我趴在偏房一處隱蔽的房頂上張望义桂。 院中可真熱鬧,春花似錦世吨、人聲如沸澡刹。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,716評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽罢浇。三九已至,卻和暖如春沐祷,著一層夾襖步出監(jiān)牢的瞬間嚷闭,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,857評論 1 269
  • 我被黑心中介騙來泰國打工赖临, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留胞锰,地道東北人。 一個月前我還...
    沈念sama閱讀 47,876評論 2 370
  • 正文 我出身青樓兢榨,卻偏偏與公主長得像嗅榕,于是被迫代替她去往敵國和親顺饮。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,700評論 2 354

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