力扣 1981. 最小化目标值与所选元素的差

题目来源：https://leetcode-cn.com/problems/minimize-the-difference-between-target-and-chosen-elements/

大致题意：
给一个 n*m 大小的矩阵，和一个目标值target。
如果从矩阵中每行选取一个数，那么所有数的和与target的差的绝对值，最小可以为多少。

思路

DP + BitSet

• 使用一个DP数组存下当前行和之前行每行选一个数可能构成的和，在本题中，可以使用BitSet（简介）来存储之前行可以组成的和（由于所有数的最大值为70，而行数最大也为70，故BitSet最大的位数即为4900）。
• 对于当前行，遍历BitSet已经set过的位（即代表之前行可能组成的和），然后加上当前数，set新的和
• 最后遍历BitSet，求出当前位与target的最小值

代码：

class Solution {
    public int minimizeTheDifference(int[][] mat, int target) {
        // 遍历每一行，存下当前行和之前和可能组成的和
        BitSet sum = new BitSet();
        int n = mat.length;
        int m = mat[0].length;
        
        // 初始时存下第一行
        for (int i = 0; i < m; i++) {
            sum.set(mat[0][i]);
        }
        for (int i = 1; i < n; i++) {
        	// 用来存新的和
            BitSet newSum = new BitSet();
            for (int j = 0; j < m; j++) {
            	// 遍历之前能存下的和
                for (int bit = sum.length(); (bit = sum.previousSetBit(bit-1)) >= 0;) {
                    newSum.set(bit+mat[i][j]);
                }
            }
            // 更新BitSet
            sum = newSum;
        }
        int ans = 4900;
        // 遍历最后得到的BitSet，求出最小差
        for (int bit = sum.length(); (bit = sum.previousSetBit(bit-1)) >= 0;) {
            int diff = Math.abs(bit - target);
            ans = Math.min(ans, diff);
            
        }
        return ans;
    }
}