首页 > 技术文章 > b_lc_ 最小化目标值与所选元素的差(dp / bitset 优化内存)

wdt1 2021-08-22 15:54 原文

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

思路:朴素 dp

const int N = 70;
const int MAX_S = N * N;
class Solution {
public:
    bool f[N][MAX_S + 1]; //第i行,和为j是否可能
    int minimizeTheDifference(vector<vector<int>>& g, int tar) {
        int n = g.size(), m = g[0].size();
        for (int j = 0; j < m; j++) {
            f[0][g[0][j]] = true;
        }
        for (int i = 1; i < n; i++) {
            for (int s = 0; s <= MAX_S; s++) {
                if (!f[i-1][s]) {
                    continue;
                }
                for (int j = 0; j < m; j++) {
                    if (s + g[i][j] > MAX_S) {
                        continue;
                    }
                    f[i][s + g[i][j]] = true; 
                }
            }
        }
        int ans = INT_MAX;
        for (int s = 0; s <= MAX_S; s++) {
            if (f[n-1][s]) {
                ans = min(ans, abs(s-tar));
            }
        }
        return ans;
    }
};

MAX_S 大一点就超时了。
优化:可以用 bitset 来压缩内存

class Solution {
public:
    int minimizeTheDifference(vector<vector<int>>& A, int target) {
        int n=A.size();
        int m=A[0].size();
        bitset<5000> F[n];
        F[0]=0;
        for (int i=0;i<m;++i) F[0][A[0][i]]=1;
        for (int i=1;i<n;++i)
        {
            F[i]=0;
            for (int j=0;j<m;++j)
                F[i]|=F[i-1]<<A[i][j];
        }
        int ans=4900;
        for (int i=1;i<=4900;++i)
            if (F[n-1][i])
                ans=min(ans,abs(target-i));
        return ans;
    }
};

二维优化成一维:利用滚动数组


推荐阅读