题目描述:
现在,我们用一些方块来堆砌一个金字塔。 每个方块用仅包含一个字母的字符串表示。
使用三元组表示金字塔的堆砌规则如下:
对于三元组(A, B, C) ,“C”为顶层方块,方块“A”、“B”分别作为方块“C”下一层的的左、右子块。当且仅当(A, B, C)是被允许的三元组,我们才可以将其堆砌上。
初始时,给定金字塔的基层 bottom,用一个字符串表示。一个允许的三元组列表 allowed,每个三元组用一个长度为 3 的字符串表示。
如果可以由基层一直堆到塔尖就返回 true ,否则返回 false 。
示例 1:
输入:bottom = “BCD”, allowed = [“BCG”, “CDE”, “GEA”, “FFF”]
输出:true
解析:
可以堆砌成这样的金字塔:
因为符合(‘B’, ‘C’, ‘G’), (‘C’, ‘D’, ‘E’) 和 (‘G’, ‘E’, ‘A’) 三种规则。
示例 2:
输入:bottom = “AABA”, allowed = [“AAA”, “AAB”, “ABA”, “ABB”, “BAC”]
输出:false
解析:
无法一直堆到塔尖。
注意, 允许存在像 (A, B, C) 和 (A, B, D) 这样的三元组,其中 C != D。
提示:
bottom 的长度范围在 [2, 8]。
allowed 的长度范围在[0, 200]。
方块的标记字母范围为{‘A’, ‘B’, ‘C’, ‘D’, ‘E’, ‘F’, ‘G’}。
方法1:
主要思路:解题汇总链接
(1)深度优先搜索;
(2)先将给定的字符串数组使用字符串的前两个字符组成的子串进行分类,使用unordered_map<string,vector>进行分类统计;
(3)使用深度优先搜索,每次根据当前层的元素,生成上一层的元素,当当前元素遍历结束,使用生成的上一层的元素,作为当前层,重新进行深度搜索,生成更上一层的元素;
class Solution {
public:
bool dfs(string& bottom,string& cur_top,unordered_map<string,vector<string>>&mp,int pos){
if(bottom.size()==1){//说明已经遍历到最顶层
return true;
}
if(pos==bottom.size()-1){//说明当前层遍历结束,可以使用生成的层,来进行更高层的判断生成
string new_top;
if(dfs(cur_top,new_top,mp,0)){//更高的层的判断
return true;
}
return false;
}
string cur_sub=bottom.substr(pos,2);//当前层的判断的位置
if(mp.count(cur_sub)==0){//说明不存在此种字符串可以添加
return false;
}
for(string& str:mp[cur_sub]){//添加可能的字符串
cur_top+=str[2];//生成更高的一层
if(dfs(bottom,cur_top,mp,pos+1)){//判断当期层的下一个位置
return true;
}
cur_top.pop_back();
}
return false;
}
bool pyramidTransition(string bottom, vector<string>& allowed) {
unordered_map<string,vector<string>> mp;
for(string& str:allowed){//对字符串数组中的字符串进行分类
mp[str.substr(0,2)].push_back(str);
}
string cur_top;//存储生成的更上一侧的字符串
return dfs(bottom,cur_top,mp,0);
}
};