记忆化搜索代码模板

代码模板:

  1. 首先需要思考在dfs过程中需要记录的对象
  2. 确定特殊的返回边界
  3. 每一层的问题的拆分与记录
  4. 遍历记录过的过程将解决的问题记录下来
  5. 返回记录的结果

public:

    // 记忆化搜索的本质是在dfs过程中,维护一个dp数组来记录之前遇到的状态遇到过的状态用该数组记录下来不再去计算。

    //其代码步骤和回溯差不多

    unordered_map<int,vector<TreeNode *>> dp;

    vector<TreeNode*> allPossibleFBT(int n) {

        if(dp.find(n)==dp.end()){//这是没遇到过的状态//边界思考

         vector<TreeNode*> root;

         if(n==1){

             dp[1].push_back(new TreeNode(0));

             return dp[1];

         }

         else if(n%2==1){

             for(int i=1;i<n;i+=2){//因为是真二叉树所有以节点个数是两个两个加的        

                 int j=n-i-1;//右子树节点个数

                    vector<TreeNode*> le=allPossibleFBT(i);//得到左子树的所有可能

                    vector<TreeNode*> ri=allPossibleFBT(j);//得到右子树的所有可能

                    for(auto &l:le){

                        for(auto &r:ri){

                            TreeNode * key =new TreeNode(0,l,r);

                            root.emplace_back(key);

                        }

                    }

                 

             }

             dp[n]=root;

         }

        }

        return dp[n];

    }

unordered_map<string ,vector<int>> dp;

     bool juage(string s){//判断是否全为数字

         for(auto key :s){

             if(key=='+'||key=='-'||key=='*'){

                 return false;

             }

         }

         return true;

     }

     vector<int> dfs_recall(string s){

         if(dp.find(s)==dp.end()){

             if(juage(s)){

                 dp[s].emplace_back(stoi(s));

                 return dp[s];//边界的返回条件

             }

             int lf = 0;

             int rig=s.size()-1;

             for(int i=lf;i<rig;i++){

                 if(s[i]=='+'){

                     string l = s.substr(lf,i);//分界

                     string r = s.substr(i+1,s.size()-i-1);

                     dp[l]=dfs_recall(l);

                     dp[r]=dfs_recall(r);

                     for(auto < :dp[l]){//枚举所有情况

                         for(auto &rg :dp[r]){

                             dp[s].emplace_back(lt + rg);

                         }

                     }

                 }

                 else if(s[i]=='-'){

                      string l = s.substr(lf,i);

                     string r = s.substr(i+1,s.size()-i-1);

                     dp[l]=dfs_recall(l);

                     dp[r]=dfs_recall(r);

                     for(auto < :dp[l]){

                         for(auto &rg :dp[r]){

                             dp[s].emplace_back(lt - rg);//记录所有情况

                         }

                     }

                 }

                  else if(s[i]=='*'){

                      string l = s.substr(lf,i);

                     string r = s.substr(i+1,s.size()-i-1);

                     dp[l]=dfs_recall(l);

                     dp[r]=dfs_recall(r);

                     for(auto < :dp[l]){

                         for(auto &rg :dp[r]){

                             dp[s].emplace_back(lt * rg);

                         }

                     }

                 }

             }    

         }

         return dp[s];

     }

    vector<int> diffWaysToCompute(string expression) {

              dfs_recall(expression);

              return dp[expression];

    }

  • 15
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值