记忆化搜索代码模板

文章讲述了如何使用深度优先搜索(DFS)和记忆化搜索策略来处理全数字字符串表达式的计算问题,通过维护dp数组记录已计算状态并避免重复计算,最终返回所有可能的计算结果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

代码模板:

  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];

    }

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值