Leetcode 756. 金字塔转换矩阵 C++

Leetcode 756. 金字塔转换矩阵

题目

现在，我们用一些方块来堆砌一个金字塔。 每个方块用仅包含一个字母的字符串表示。

使用三元组表示金字塔的堆砌规则如下：

对于三元组(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。

提示：

1. bottom 的长度范围在 [2, 8]。
2. allowed 的长度范围在[0, 200]。
3. 方块的标记字母范围为{‘A’, ‘B’, ‘C’, ‘D’, ‘E’, ‘F’, ‘G’}。

题解

dfs深搜回溯
在搭建某层时，我们需要用到下一层的信息。另外，我们知道第level层有level个砖块，在构建第i个砖块时，它需要和下一层的第i个和第i+1个砖块进行合体进行检验。详细过程见代码

代码

	unordered_map<string,int> allow;		//哈希表记录可以存在的状态
    bool dfs(string& now,string newStr,int i,int level){
        if(level == 0)  return true;
        for(char c='A'; c<='G'; c++){
            string x;
            x = now[i];		//当前选的方块应该和下一层第i块和第i+1块进行结合后检验，按照左右上的结构构造
            x += now[i+1];
            x += c;
            if(allow.count(x) != 0){		//允许这么搭建
                newStr.push_back(c);
                if(i == level-1){			//该层搭建完成，开始上一层的搭建
                    if(dfs(newStr,"",0,level-1)) return true;
                }else if(dfs(now,newStr,i+1,level)) return true;		//下一个方块的搭建
                newStr.pop_back();
            }
        }
        return false;
    }
    bool pyramidTransition(string bottom, vector<string>& allowed) {
        int n = allowed.size();
        for(int i=0; i<n; i++){
            allow[allowed[i]] = 1;
        } 
        return dfs(bottom,"",0,bottom.length()-1);
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/pyramid-transition-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。