动态规划DP：计算二维数组从(1,1)到(i,j)的矩形区域的值的和

最新推荐文章于 2021-11-18 15:58:42 发布

Dennis F

最新推荐文章于 2021-11-18 15:58:42 发布

阅读量308

点赞数

分类专栏： CSP 文章标签：动态规划二维数组矩形区域前缀和 C语言

本文链接：https://blog.csdn.net/WZZ_2363173126/article/details/115628677

版权

CSP 专栏收录该内容

25 篇文章 4 订阅

订阅专栏

问题描述

存在一个N*N的二维数组(1<=N<=500)，计算并输出二维数组每一个从(1,1)到(i,j)的矩形区域的值的和(1<=i<=N,1<=j<=N)。

输入与输出

输入格式

第1行输入一个整数N(1<=N<=500)。
第i（2<=i<=N+1）行输入N个数（共计N*N个），代表二维数组第i-1行的N个数字。

输出格式

       输出N行。
       第i（1<=i<=N）行输出N个数（共计N*N个）。
       第 i 行第 j 列(1<=j<=N)输出二维数组从(1,1)到(i,j)的矩形区域的值的和。数字之间以空格隔开。

测试样例

测试输入

4
0 1 2 3 
4 5 6 7 
8 9 10 11 
12 13 14 15

测试输出

0 1 3 6
4 10 18 28
12 27 45 66
24 52 84 120

问题思路

动态规划（时间复杂度为O(N²)）：

DP_Sum[ i ][ j ] = DP_Sum[ i - 1 ][ j ] + DP_Sum[ i ][ j - 1 ] - DP_Sum[ i - 1 ][ j - 1 ] + Num[ i ][ j ];

代码示例

#include <stdio.h>

#define MaxSize 510

int Num[MaxSize][MaxSize] = {0};

//DP_Sum保存二维数组从(1,1)到(i,j)的矩形区域的值的和
int DP_Sum[MaxSize][MaxSize] = {0};

int main()
{
    int N;

    scanf("%d", &N);

    for (int i = 1; i <= N; ++i)
    {
        for (int j = 1; j <= N; ++j)
        {
            scanf("%d", &Num[i][j]);
        }
    }

    for (int i = 1; i <= N; ++i)
    {
        for (int j = 1; j <= N; ++j)
        {
            DP_Sum[i][j] = DP_Sum[i - 1][j] + DP_Sum[i][j - 1] - DP_Sum[i - 1][j - 1] + Num[i][j];

            printf("%d ", DP_Sum[i][j]);
        }
        putchar('\n');
    }

    return 0;
}