1253. 重构 2 行二进制矩阵(力扣)
- 题目
- 第一种方式
-
- 分析
- 测试代码
- 运行结果
- 第二种方式
-
- 测试代码
- 运行结果
题目
给你一个 2 行 n 列的二进制数组:
矩阵是一个二进制矩阵,这意味着矩阵中的每个元素不是 0 就是 1。
第 0 行的元素之和为 upper。
第 1 行的元素之和为 lower。
第 i 列(从 0 开始编号)的元素之和为 colsum[i],colsum 是一个长度为 n 的整数数组。
你需要利用 upper,lower 和 colsum 来重构这个矩阵,并以二维整数数组的形式返回它。
如果有多个不同的答案,那么任意一个都可以通过本题。
如果不存在符合要求的答案,就请返回一个空的二维数组。
输入: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
第一种方式
分析
遍历到的元素colsum[i],我们有以下几种情况:
如果colsum[j]=2,那么我们将ans[0][j]和ans[1][j]都置为1。
如果colsum[j]=0,那么我们将ans[0][i]和ans[1][j]都置为0。
如果colsumli]= 1,先满足第一行为1,满足完后剩下的都是另一行的。
之后需要判断, 计算每一行的总和,看是否满足条件与upper和lower相等。
测试代码
class Solution {
public ListListInteger>> reconstructMatrix(int upper, int lower, int[] colsum) {
int length=colsum.length;
//声明数组,来表示两行length列
int ans[][]=new int [2][length];
int sum=0;
//colsum为2那么当前列数字都为1,colsum为0当前列数字都为为0
for (int i = 0; i length ; i++) {
if (colsum[i]==2){
ans[0][i]=1;
ans[1][i]=1;
sum++;
}else if (colsum[i]==0){
ans[0][i]=0;
ans[1][i]=0;
}
}
//colsum为1的列先满足使第一行为1
int count=upper-sum;
for (int i = 0; i length ; i++) {
if (colsum[i]==1&&count>0){
ans[0][i]=1;
ans[1][i]=0;
--count;
}
}
//剩下colsum为1的列再使第二行为1
for (int i = 0; i length ; i++) {
if (colsum[i]==1&&ans[0][i]!=1){
ans[1][i]=1;
ans[0][i]=0;
}
}
ListInteger> list;
ListListInteger>> lists=new ArrayList>();
//计算每一行的总和,看是否满足条件
int num1=0,num2=0;
for (int i = 0; i length ; i++) {
num1+=ans[0][i];
num2+=ans[1][i];
}
//不满足条件
if (num2!=lower||num1!=upper){
return lists;
}
//满足条件
for (int i = 0; i 2 ; i++) {
list=new ArrayList>();
for (int j = 0; j length; j++) {
list.add(ans[i][j]);
}
lists.add(list);
}
return lists;
}
}
运行结果
第二种方式
大致思想是一样的
这里是挨个,遍历元素colsum[i],我们有以下几种情况:
如果colsum[j]=2,那么我们将ans[0][j]和ans[1][j]都置为1,此时upper都lower 减去1。
如果colsum[j]=0,那么我们将ans[0][i]和ans[1][j]都置为0。
如果colsum[i]= 1,那么我们将ans[0][i]或ans[1][j]置为1。判断upper和lower谁大,谁就为1,此时upper或lower 减去1。
如果upper
然后判断是否满足条件。
测试代码
class Solution {
public ListListInteger>> reconstructMatrix(int upper, int lower, int[] colsum) {
int length = colsum.length;
ListInteger> ans1 = new ArrayList>();
ListInteger> ans2 = new ArrayList>();
for (int j = 0; j length; ++j) {
int a = 0, b = 0;
if (colsum[j] == 2) {
a = b = 1;
upper--;
lower--;
} else if (colsum[j] == 1) {
if (upper > lower) {
upper--;
a = 1;
} else {
lower--;
b = 1;
}
}
if (upper 0 || lower 0) {
break;
}
ans1.add(a);
ans2.add(b);
}
ListListInteger>> lists = new ArrayList>();
if (upper !=0 || lower !=0) {
return lists;
} else {
lists.add(ans1);
lists.add(ans2);
return lists;
}
}
}
运行结果
服务器托管,北京服务器托管,服务器租用 http://www.fwqtg.net
相关推荐: VulnStack – ATT&CK红队评估实战(四) Writeup
VulnStack – ATT&CK红队评估实战(四) Writeup VulnStack(四)环境搭建 1、项目地址 http://vulnstack.qiyuanxuetang.net/vuln/detail/6/ 2、网络拓扑 机器名称 网卡VM…