1. 題目

給定一份單詞的清單，設計一個算法，創建由字母組成的面積最大的矩形，其中每一行組成一個單詞(自左向右)，每一列也組成一個單詞(自上而下)。
不要求這些單詞在清單里連續出現，但要求所有行等長，所有列等高。

如果有多個面積最大的矩形，輸出任意一個均可。一個單詞可以重復使用。

示例 1: 輸入: ["this", "real", "hard", "trh", "hea", "iar", "sld"] 輸出: ["this","real","hard" ]示例 2: 輸入: ["aa"] 輸出: ["aa","aa"]說明： words.length <= 1000 words[i].length <= 100 數據保證單詞足夠隨機

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/word-rectangle-lcci
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

類似題目：LeetCode 425. 單詞方塊（Trie樹+DFS）

• 將所有單詞插入Trie樹
• 將單詞按長度分組，哈希map
• 從單詞長度最長組的開始遍歷，對每組單詞進行DFS搜索
• 利用Trie樹檢查是否合法，不合法回溯
• 有幾處優化見注釋，容易超時

class trie { public:bool isEnd = false;trie* next[26] = {NULL};void insert(string& s){trie *cur = this;for(int i = 0; i < s.size(); ++i){if(!cur->next[s[i]-'a'])cur->next[s[i]-'a'] = new trie();cur = cur->next[s[i]-'a'];}cur->isEnd = true;} }; class Solution {trie* t;vector<string> ans;vector<string> temp; public:vector<string> maxRectangle(vector<string>& words) {t = new trie();map<int,vector<string>> m;int maxlen = 0, maxarea = 0, area;for(auto& w : words){t->insert(w);//單詞插入triem[w.size()].push_back(w);//單詞按長度分組maxlen = max(maxlen, int(w.size()));//最大單詞長度}for(auto it = m.rbegin(); it != m.rend(); ++it){ //反向遍歷，從長度大的開始if(maxarea/(it->first) >= maxlen)break;//最長的單詞*寬度 都不夠大，這組不用找了temp.clear();area = 0;dfs(it->second,maxarea,maxlen,area);}return ans;}void dfs(vector<string>& wd, int& maxarea, int maxlen, int area){if(wd[0].size()*maxlen <= maxarea)//找到的面積到極限了，退出吧return;//這個優化必須有，沒有會超時vector<bool> res;for(int i = 0; i < wd.size(); ++i){temp.push_back(wd[i]);res = isgood(temp);//是否合法if(res[0])//可以往下加單詞{area = temp.size()*temp[0].size();if(res[1] && area > maxarea)//都是結束單詞{ //更新最大值maxarea = area;ans = temp;}dfs(wd, maxarea, maxlen, area);}// else//不能有else，有的話 if內的dfs出來，沒有回溯temp.pop_back();}}vector<bool> isgood(vector<string>& tp){trie *cur;bool allend = true;int i, j, n = tp[0].size();for(j = 0; j < n; ++j)//按列在trie中檢查{cur = t;for(i = 0; i < tp.size(); ++i){if(!cur->next[tp[i][j]-'a'])return {false, false};cur = cur->next[tp[i][j]-'a'];}allend &= cur->isEnd;}return {true, allend};//可以繼續插入、每個列向單詞都在字典中} };

總結

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 17.25. 单词矩阵（Trie树+DFS回溯，hard）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。