算法4(金字塔转换矩阵)

金字塔转换矩阵

题目描述

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

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

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

解题思路

1. 利用深度优先搜索算法
2. 在dfs之前先对所给的数据进行整理，利用哈希表处理金字塔的三元组特性。然后，为了求解下一行，我们生成下一行所有的可能组合并求解它们。如果这些组合中有任一一行是可解的，则返回 True，反之返回 False。
3. 由下一行的两个可以推出上一行，可将下两个作为key值，上一个作为value值
4. 递归函数dfs接受三个参数，第一个last为目前检查的层，now为已经形成的部分下一层，dir为已生成的映射关系。
5. .再根据当前层获取下一层可能的组成的字符串，然后逐层递归处理，直到该层长度为1，则找到了结果返回true，否则获取下一个可能组成的字符串，继续上述处理，遍历完还没有结果则返回false

代码实现

class Solution {
    public boolean pyramidTransition(String bottom, List<String> allowed) {
        HashMap<String, List<String>> dir = new HashMap<>(16);
        for (String str : allowed) { //将allowed转为键值对形式
            String head = str.substring(0, 2);
            String ch = str.substring(2, 3);
            if (dir.containsKey(head)) {
                dir.get(head).add(ch);
            } else {
                List<String> p = new ArrayList<>();
                p.add(ch);
                dir.put(head, p);
            }
        }
        return (dfs(bottom, "", dir));
    }

    static boolean dfs(String last, String now, HashMap<String, List<String>> dir){
         //到达第一行结束判断
        if (last.length() == 2 && now.length() == 1) {
            return true;
        }
        //中间行填充完成判断
        if (now.length() == last.length() - 1) {
            return dfs(now, "", dir);
        }
        int start = now.length();
        int end = now.length() + 2; //指示器，两个两个指
        //通过保存的键值对，获取上一行对应的值
        List<String> p = dir.get(last.substring(start, end));
        if (p == null) return false; //没有匹配的值，当前串不满足金字塔
         for (int i = 0; i < (p).size(); i++) {
            if (dfs(last, now + p.get(i), dir)) { //获取bottom上一层一个之后继续下一个
                return true; //上一行有值才返回true
            }
        }
        return false;
    }
}