从一道题目中思考dfs函数需不需要返回值(void or int···)

有返回值int的代码:

#include<bits/stdc++.h>
using namespace std;
int main(){
    int a,b;
    cin>>a>>b;
    vector<vector<int>> map(a,vector<int>(b,0));
    vector<vector<int>> f(a,vector<int>(5005,-1));
    for(int i=0;i<a;i++){
        for(int j=0;j<b;j++){
            cin>>map[i][j];
        }
    }
    int target;
    cin>>target;
    function<int(int,int)> dfs=[&](int row,int sum){
        int res=INT_MAX;
       if(row==a) return abs(sum-target);
       else if(f[row][sum]!=-1) return f[row][sum];
       else{
           for(int i=0;i<b;i++){
               res=min(dfs(row+1,sum+map[row][i]),res);
           }
       }
        f[row][sum]=res;
        return res;
    };
    cout<<dfs(0,0)<<endl;
    return 0;
}

无返回值的代码:

#include<bits/stdc++.h>
using namespace std;
int main(){
    int a,b;
    int ans=INT_MAX;
    cin>>a>>b;
    vector<vector<int>> map(a,vector<int>(b,0));
    vector<vector<int>> f(a,vector<int>(5005,-1));
    for(int i=0;i<a;i++){
        for(int j=0;j<b;j++){
            cin>>map[i][j];
        }
    }
    int target;
    cin>>target;
    function<void(int,int)> dfs=[&](int row,int sum){
        if(row==a-1){
            for(int i=0;i<b;i++){
                ans=min(ans,abs(target-(sum+map[row][i])));
            }
        }
        else{
            for(int i=0;i<b;i++){
                dfs(row+1,sum+map[row][i]);
            }            
        }
    };
    dfs(0,0);
    cout<<ans<<endl;
    return 0;
}

上面两份代码第一份是可以通过的(因为多了记忆化搜索),第二份超时了。在这题中,由于我们想加入记忆化搜索,如果没有返回值,会导致运行和记录的不同步(确切值的记录),所以选择有返回值的会容易写一点。

另附上一种dp写法:

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n, m;
    cin >> n >> m;
    
    vector<vector<int>> map(n, vector<int>(m));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> map[i][j];
        }
    }
    
    int target;
    cin >> target;

    // 最大可能的和
    int max_sum = n * 50;
    vector<bool> dp(max_sum + 1, false);
    dp[0] = true;

    for (int i = 0; i < n; i++) {
        vector<bool> new_dp(max_sum + 1, false);
        for (int j = 0; j <= max_sum; j++) {
            if (dp[j]) {
                for (int k = 0; k < m; k++) {
                    if (j + map[i][k] <= max_sum) {
                        new_dp[j + map[i][k]] = true;
                    }
                }
            }
        }
        dp = new_dp;
    }

    int ans = INT_MAX;
    for (int i = 0; i <= max_sum; i++) {
        if (dp[i]) {
            ans = min(ans, abs(target - i));
        }
    }
    
    cout << ans << endl;
    return 0;
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值