小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;
}
相关推荐
uhakadotcom1 小时前
Apache Airflow入门指南:数据管道的强大工具
算法·面试·github
跳跳糖炒酸奶1 小时前
第四章、Isaacsim在GUI中构建机器人(2):组装一个简单的机器人
人工智能·python·算法·ubuntu·机器人
绵绵细雨中的乡音2 小时前
动态规划-第六篇
算法·动态规划
程序员黄同学2 小时前
动态规划,如何应用动态规划解决实际问题?
算法·动态规划
march_birds2 小时前
FreeRTOS 与 RT-Thread 事件组对比分析
c语言·单片机·算法·系统架构
斯汤雷3 小时前
Matlab绘图案例,设置图片大小,坐标轴比例为黄金比
数据库·人工智能·算法·matlab·信息可视化
云 无 心 以 出 岫3 小时前
贪心算法QwQ
数据结构·c++·算法·贪心算法
俏布斯3 小时前
算法日常记录
java·算法·leetcode
独好紫罗兰3 小时前
洛谷题单3-P5719 【深基4.例3】分类平均-python-流程图重构
开发语言·python·算法
SheepMeMe4 小时前
蓝桥杯2024省赛PythonB组——日期问题
python·算法·蓝桥杯