566. 重塑矩阵 AND 561. 数组拆分 I

150 阅读1分钟

这是我参与8月更文挑战的第31天,活动详情查看:8月更文挑战

566. 重塑矩阵

在MATLAB中,有一个非常有用的函数 reshape,它可以将一个矩阵重塑为另一个大小不同的新矩阵,但保留其原始数据。

给出一个由二维数组表示的矩阵,以及两个正整数r和c,分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

如果具有给定参数的reshape操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。

示例 1:

输入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

解题思路

将原数组从左到右 从上到下进行编号,按编号依次将元素放入新数组,只要将编号取模和进行除法运算,就能得出原数组元素的行号跟列号

代码

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {

        int[][] temp=new int[r][c];
        int k=0,z=nums.length*nums[0].length;
        if(z!=r*c) return nums;
        for (int i = 0; i < temp.length; i++) {
               for (int i1 = 0; i1 <temp[0].length; i1++) {
                temp[i][i1]=nums[k/nums[0].length][k%nums[0].length];
                k++;
            }
        }
        return temp;
    }
}

561. 数组拆分 I

给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得从 1 到 n 的 min(ai, bi) 总和最大。

返回该 最大总和 。

示例 1:

输入:nums = [1,4,3,2] 输出:4 解释:所有可能的分法(忽略元素顺序)为:

  1. (1, 4), (2, 3) -> min(1, 4) + min(2, 3) = 1 + 2 = 3
  2. (1, 3), (2, 4) -> min(1, 3) + min(2, 4) = 1 + 2 = 3
  3. (1, 2), (3, 4) -> min(1, 2) + min(3, 4) = 1 + 3 = 4 所以最大总和为 4

解题思路

每个数字对的两个元素越接近,那么min(ai, bi) 总和最大

代码

class Solution {
    public int arrayPairSum(int[] nums) {

        int res=0;
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i+=2) {
            res+= Math.min(nums[i],nums[i+1]);
        }
        return res;
    }
}