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;
}
};
二维优化成一维:利用滚动数组