重构 2 行二进制矩阵

/*
 * 思路：
 * 我们使用 Java Stream 构建解法。
 * 使用 List<List<Integer>> 来表示矩阵，并使用流来遍历 colsum 数组。
 */

import java.util.*;
import java.util.stream.*;

public class SolutionStream {
    public List<List<Integer>> reconstructMatrix(int upper, int lower, int[] colsum) {
        int n = colsum.length;
        int[] upperRow = new int[n];
        int[] lowerRow = new int[n];

        for (int i = 0; i < n; i++) {
            if (colsum[i] == 2) {
                upperRow[i] = 1;
                lowerRow[i] = 1;
                upper--;
                lower--;
            } else if (colsum[i] == 1) {
                if (upper > lower) {
                    upperRow[i] = 1;
                    upper--;
                } else {
                    lowerRow[i] = 1;
                    lower--;
                }
            }
        }

        if (upper != 0 || lower != 0) {
            return new ArrayList<>(); // 返回空数组表示无法构建
        }

        List<List<Integer>> result = Arrays.asList(
            Arrays.stream(upperRow).boxed().collect(Collectors.toList()),
            Arrays.stream(lowerRow).boxed().collect(Collectors.toList())
        );

        return result;
    }
}