金字塔转换矩阵

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

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

对于三元组(A, B, C) ，“C”为顶层方块，方块“A”、“B”分别作为方块“C”下一层的的左、右子块。当且仅当(A, B, C)是被允许的三元组，我们才可以将其堆砌上。

初始时，给定金字塔的基层 bottom，用一个字符串表示。一个允许的三元组列表 allowed，每个三元组用一个长度为 3 的字符串表示。

如果可以由基层一直堆到塔尖就返回 true ，否则返回 false 。

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Set;

class Solution {
    public boolean pyramidTransition(String bottom, List<String> allowed) {
        if(allowed==null||allowed.size()==0) return false;
        HashMap<String,List<String>> dir=new HashMap<>(16);
        for (String str:allowed){
            String head = str.substring(0,2);//返回(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));
    }

    private boolean dfs(String bottom, String s, HashMap<String, List<String>> dir) {
        if(bottom.length()==2&&s.length()==1){
            return true;
        }
        if (s.length()==bottom.length()-1){
            return dfs(s,"",dir);
        }
        int start = s.length();
        int end=s.length()+2;
        List<String> p=dir.get(bottom.substring(start, end));
        if(p==null) return false;
        for (int i = 0; i < (p).size(); i++) {
            if(dfs(bottom,s+p.get(i),dir)) {
                return true;
            }
        }
        return false;
    }
}