题目描述:
给你一个 2 行 n 列的二进制数组:
矩阵是一个二进制矩阵,这意味着矩阵中的每个元素不是 0 就是 1。
第 0 行的元素之和为 upper。
第 1 行的元素之和为 lower。
第 i 列(从 0 开始编号)的元素之和为 colsum[i],colsum 是一个长度为 n 的整数数组。
你需要利用 upper,lower 和 colsum 来重构这个矩阵,并以二维整数数组的形式返回它。
如果有多个不同的答案,那么任意一个都可以通过本题。
如果不存在符合要求的答案,就请返回一个空的二维数组。
示例 1:
输入:upper = 2, lower = 1, colsum = [1,1,1]
输出:[[1,1,0],[0,0,1]]
解释:[[1,0,1],[0,1,0]] 和 [[0,1,1],[1,0,0]] 也是正确答案。
示例 2:
输入:upper = 2, lower = 3, colsum = [2,2,1,1]
输出:[]
示例 3:
输入:upper = 5, lower = 5, colsum = [2,1,2,0,1,0,1,2,0,1]
输出:[[1,1,1,0,1,0,0,1,0,0],[1,0,1,0,0,0,1,1,0,1]]
提示:
1 <= colsum.length <= 10^5
0 <= upper, lower <= colsum.length
0 <= colsum[i] <= 2
实现代码:
class Solution {
public:
vector<vector<int>> reconstructMatrix(int upper, int lower, vector<int>& colsum) {
//该函数将要返回的是一个二位整数向量,用于重构的矩阵
int n = colsum.size();
int sum = 0,two = 0;
for(int i = 0; i < n;++i)
{
if(colsum[i]==2)//记录列和为2的个数
{
++two;
}
sum += colsum[i];
}
if(sum!=upper+lower || min(upper,lower)<two)
//如果总和与upper+lower不相等那么肯定没有结果
//此外如果colsum[i]==2,那么上下两行在第i列肯定是1,所以如果min的结果小于two,那么肯定不成立
{
return {};
}
upper -= two;//减去肯定为1的大小,剩下的就是upper和lower不同的部分了
lower -= two;
vector<vector<int>> res(2,vector<int>(n,0));
//创建一个大小为2xn的二维矩阵res,并将其所有元素初始化为0。
for(int i = 0;i < n;++i)
{
if(colsum[i]==2)
{
res[0][i] = res[1][i] = 1;//colsum[i]为二,则两行第i列均为1
}
//剩下的就是根据还有多少差距进行一个个加入
else if(colsum[i]==1)
{
if(upper>0)
{
res[0][i] = 1;
--upper;//upper优先进行补充
}
else
{
res[1][i] = 1;
}
}
}
return res;
}
};
其中:vector<vector<int>> reconstructMatrix(int upper, int lower, vector<int>& colsum)
用于定义一个名为reconstructMatrix
的函数,该函数接受三个参数:
upper
:一个整数,表示矩阵的上行总和。
lower
:一个整数,表示矩阵的下行总和。
colsum
:一个引用,表示一个整数向量,其中存储了每一列的和。
函数返回类型为vector<vector<int>>
,表示返回一个二维整数向量,用于重构的矩阵。
这句话定义了函数的名称、参数以及返回类型
PS:后来又看了一遍,这个题不算难,就是依旧得看清题目所给条件,colsum这个数组的数据是个关键,找到技巧或者说突破点。