小R的蛋糕分享

小R的蛋糕分享

问题描述

小R手里有一个大小为 n 行 m 列的矩形蛋糕,每个小正方形区域都有一个代表美味度的整数。小R打算切割出一个正方形的小蛋糕给自己,而剩下的部分将给小S。她希望两人吃的部分的美味度之和尽量接近。

我们定义小R吃到的部分的美味度之和为 s_1,而小S吃到的部分的美味度之和为 s_2,请你帮助小R找到一个切割方案,使得 |s_1 - s_2| 的值最小。

测试样例

样例1:

输入:n = 3, m = 3, a = [[1, 2, 3], [2, 3, 4], [3, 2, 1]]

输出:1
样例2:

输入:n = 4, m = 4, a = [[1, 2, 3, 4], [4, 3, 2, 1], [1, 2, 3, 4], [4, 3, 2, 1]]

输出:2
样例3:

输入:n = 2, m = 2, a = [[5, 5], [5, 5]]

输出:10

题解

核心思想,枚举每一个点作为正方形右下角的那个点。使用前缀和数组进行重复求和运算。

cpp 复制代码
#include <iostream>
#include <vector>
#include <string>

using namespace std;

int solution(int n, int m, vector<vector<int>>& a) {
    // PLEASE DO NOT MODIFY THE FUNCTION SIGNATURE
    // write code here
    int sum = 0;
    // 二维前缀和,记录0,0到i,j的美味度和
    //int num[n][m+1];
    vector<vector<int>> num(n + 1, vector<int>(m + 1, 0));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            num[i + 1][j + 1] = a[i][j] + num[i][j+1] + num[i+1][j] - num[i][j];
            sum += a[i][j];
        }
    }
    int ans = INT32_MAX;
    int maxNum = min(n, m);
    for (int i = 1; i <= maxNum; i++) {
        for (int r = 0; r < n; r++) {
            for (int c = 0; c < m; c++) {
                if (c + 1 < i  || r + 1 < i) {
                    continue;
                }
                int tmp = num[r+1][c+1] - num[r+1][c+1-i] - num[r+1-i][c+1] + num[r+1-i][c+1-i];
                ans = min(ans, abs(sum - 2 * tmp));

            }
        }
    }
    return ans;
}
相关推荐
独好紫罗兰1 分钟前
洛谷题单3-P1980 [NOIP 2013 普及组] 计数问题-python-流程图重构
开发语言·python·算法
独好紫罗兰6 分钟前
洛谷题单3-P1009 [NOIP 1998 普及组] 阶乘之和-python-流程图重构
开发语言·python·算法
曦月逸霜17 分钟前
蓝桥杯高频考点——高精度(含C++源码)
c++·算法·蓝桥杯
ゞ 正在缓冲99%…27 分钟前
leetcode152.乘积最大子数组
数据结构·算法·leetcode
闯闯爱编程1 小时前
数组与特殊压缩矩阵
数据结构·算法·矩阵
秋风战士1 小时前
通信算法之255:无人机频谱探测设备技术详解
算法·无人机
laimaxgg2 小时前
数据结构B树的实现
开发语言·数据结构·c++·b树·算法
mit6.8242 小时前
[Lc6_记忆化搜索] 最长递增子序列 | 矩阵中的最长递增路径
c++·算法·leetcode
ylfhpy4 小时前
Java面试黄金宝典30
java·数据库·算法·面试·职场和发展
明.2444 小时前
DFS 洛谷P1123 取数游戏
算法·深度优先