1253. 重构 2 行二进制矩阵-贪心算法

1253. 重构 2 行二进制矩阵

给你一个 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,这题解题代码如下:

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int** reconstructMatrix(int upper, int lower, int* colsum, int colsumSize, int* returnSize, int** returnColumnSizes){
    int sum=0;
    int **re=(int **)malloc(sizeof(int *)*2);
    for(int i=0;i<colsumSize;i++){
        sum=sum+colsum[i];
    }
    if(sum!=upper+lower){
        *returnSize=0;
        return re;
    }
     *returnColumnSizes=(int *)malloc(sizeof(int )*2);
    
    re[0]=(int *)malloc(sizeof(int)*colsumSize);
    (*returnColumnSizes)[0]=colsumSize;
    (*returnColumnSizes)[1]=colsumSize;
     re[1]=(int *)malloc(sizeof(int)*colsumSize);
      for(int i=0;i<colsumSize;i++){
        //  printf("%d ",colsum[i]);
       if(colsum[i]==2){
           upper--;
           lower--;
           re[0][i]=1;
           re[1][i]=1;
       }
       else if(colsum[i]==0){
           re[0][i]=0;
           re[1][i]=0;

       }
       else{
           if(upper>lower){
                re[0][i]=1;
                re[1][i]=0;
                upper--;

           }
           else{
                 re[0][i]=0;
                re[1][i]=1;
               lower--;
           }
         
           


       }
       if(lower<0||upper<0){
           *returnSize=0;
        return re;

       }
     //  printf("-- %d %d |",upper,lower);
    }
    *returnSize=2;
    return re;

     

}

你可能感兴趣的:(c++与c语言,力扣刷题,贪心算法,重构,矩阵)