(DFS+缓存+剪枝)—最小化目标值与所选元素的差（leetcode 5852）

最新推荐文章于 2021-12-29 17:10:16 发布

珠穆拉玛峰

最新推荐文章于 2021-12-29 17:10:16 发布

阅读量149

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/guoxinian/article/details/119857265

版权

leetcode 专栏收录该内容

89 篇文章 1 订阅

订阅专栏

该博客主要讨论了一个算法问题，即如何在一个给定的矩阵中找到一组元素，使得它们的和与目标值的绝对差最小。算法通过深度优先搜索（DFS）策略，结合剪枝优化，动态规划存储已计算过的行和，来有效地找到最优解。示例展示了不同情况下的解决方案，并分析了算法的时间复杂度。

摘要由CSDN通过智能技术生成

题目描述

给你一个大小为 m x n 的整数矩阵 mat 和一个整数 target 。

从矩阵的每一行中选择一个整数，你的目标是最小化所有选中元素之和与目标值 target 的绝对差。

返回最小的绝对差。

a 和 b 两数字的绝对差是 a - b 的绝对值。

示例 1：

输入：mat = [[1,2,3],[4,5,6],[7,8,9]], target = 13
输出：0
解释：一种可能的最优选择方案是：
- 第一行选出 1
- 第二行选出 5
- 第三行选出 7
所选元素的和是 13 ，等于目标值，所以绝对差是 0 。

示例 2：

输入：mat = [[1],[2],[3]], target = 100
输出：94
解释：唯一一种选择方案是：
- 第一行选出 1
- 第二行选出 2
- 第三行选出 3
所选元素的和是 6 ，绝对差是 94 。

示例 3：

输入：mat = [[1,2,9,8,7]], target = 6
输出：1
解释：最优的选择方案是选出第一行的 7 。
绝对差是 1 。

提示：

    m == mat.length
    n == mat[i].length
    1 <= m, n <= 70
    1 <= mat[i][j] <= 70
    1 <= target <= 800

算法分析

1、用row记录搜索到第几行，如果搜索到最后一行更新result值
2.添加缓存cache,cache[i][j]表示第i行总和为j，如果已经有访问到了，就无须再DFS，剪枝去重；

3、枚举每行的值
4.如果当前curNum - target已经大于result，name再添加也没有意义了，直接剪枝去掉即可；

5、进行递归遍历

代码

class Solution {
public:
    int result;
    vector<vector<bool>> cache;
    void dfs(vector<vector<int>>& mat, int index, int n,int sum, int target) {
        if(index > n) {
            return;
        }
        if(index == n) {
            result = min(result, abs(sum - target));
            return;
        }
        if(cache[index][sum]) {
            return;
        }
        cache[index][sum] = true;
        for(int i = 0; i < mat[index].size(); ++i) {
            int tmp_sum = sum + mat[index][i];
            if (tmp_sum - target > result) {
                continue;
            }
            dfs(mat, index+1, n, tmp_sum, target);
        }
    }
    int minimizeTheDifference(vector<vector<int>>& mat, int target) {
        int n = mat.size();
        result = INT_MAX;
        cache.resize(n, vector<bool>(4096, false));
        dfs(mat, 0, n, 0, target);
        return result;
    }
};