LeetCode 756. 金字塔转换矩阵（回溯）

文章目录

- 1. 题目
- 2. 解题

1. 题目

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

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

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

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

2. 解题

暴力回溯穷举即可

class Solution {bool ans = false;vector<char> ch = {'A', 'B', 'C', 'D', 'E', 'F', 'G'};
public:bool pyramidTransition(string bottom, vector<string>& allowed) {int n = bottom.size();unordered_set<string> s(allowed.begin(), allowed.end());vector<vector<char>> gold(n, vector<char>(n, '*'));for(int j = 0; j < n; j++) {gold[n-1][j] = bottom[j];//填充最底层的字母}dfs(gold, s, n-2, 0);//从倒数第二行，第0列，开始搜索return ans;}void dfs(vector<vector<char>>& gold, unordered_set<string>& s, int i, int j){if(ans == true)return;char origin = gold[i][j];//记录原始字母for(int k = 0; k < 7; k++)//填入7种可能的字母{string t = string(1, gold[i+1][j])+gold[i+1][j+1]+char('A'+k);// 三元组字符串if(s.find(t) != s.end())//字符串在字典中{if(i == j)//这一行填完了{if(i == 0)ans = true;gold[i][j] = char('A'+k);//填入可能的字母dfs(gold, s, i-1, 0);//下一行，0列gold[i][j] = origin;//恢复现场}else{gold[i][j] = char('A'+k);dfs(gold, s, i, j+1);//当前行，下一列gold[i][j] = origin;}}}}
};