题目
现在,我们用一些方块来堆砌一个金字塔。 每个方块用仅包含一个字母的字符串表示。
使用三元组表示金字塔的堆砌规则如下:
对于三元组(A, B, C) ,“C”为顶层方块,方块“A”、“B”分别作为方块“C”下一层的的左、右子块。当且仅当(A, B, C)是被允许的三元组,我们才可以将其堆砌上。
初始时,给定金字塔的基层 bottom,用一个字符串表示。一个允许的三元组列表 allowed,每个三元组用一个长度为 3 的字符串表示。
如果可以由基层一直堆到塔尖就返回 true ,否则返回 false 。
示例 1:
输入:bottom = “BCD”, allowed = [“BCG”, “CDE”, “GEA”, “FFF”]
输出:true
解析:
可以堆砌成这样的金字塔:
A
/
G E
/ \ /
B C D
因为符合(‘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’}。
思路
递归遍历,先检查bottom,pos==bottom.size()-1的时候说明这一次层检查好了,开始检查下一层也就是upper。如果下一个位置没有可以放置的砖块,也就是vc.empty(),就返回false,否则就尝试所有可以放置的砖块。upper.size()为1的时候,就是到顶了可以返回true。
代码
class Solution {
public:
unordered_map<string,vector<chat>> mp;
bool pyramidTransition(string bottom, vector<string>& allowed) {
for(string str:allowed){
mp[str.substr(0,2)].push_back(str.back());
}
string upper = "";
return dfs(bottom,upper,0);
}
bool dfs(string bottom,string upper,int pos){
if(pos == bottom.size()-1)
{
if(upper.size()==1)
return true;
bottom=upper;
upper=""
pos=0;
}
vector<char>& vc= mp[bottom.substr(pos,2)];
if(vc.empty())
return false;
else{
for(int i =0;i<vc.size();i++)
{
if(dfs(bottom,upper+vc[i],pos+1))
return true;
}
}
return false;
}
};