Online Judge——1002. 二哥种花生(c++)

1002. 二哥种花生

Description

二哥在自己的后花园里种了一些花生，也快到了收获的时候了。这片花生地是一个长度为L、宽度为W的矩形，每个单位面积上花生产量都是独立的。他想知道，对于某个指定的区域大小，在这么大的矩形区域内，花生的产量最大会是多少。

Input Format

第1行有2个整数，长度L和宽度W。

第2行至第L+1行，每行有W个整数，分别表示对应的单位面积上的花生产量A（ 0≤A<100≤A<10 ）。

第L+2行有2个整数，分别是指定的区域大小的长度a和宽度b。

Output Format

输出一个整数m，表示在指定大小的区域内，花生最大产量为m。

Sample Input

4 5
1 2 3 4 5
6 7 8 0 0
0 9 2 2 3
3 0 0 0 1
3 3

Sample Output

38

样例解释

左上角：38 = (1+2+3) + (6+7+8) + (0+9+2)

数据范围

对于30%的数据： 1≤L,W≤1001≤L,W≤100；

对于100%的数据： 1≤L,W≤10001≤L,W≤1000。

全部区域大小满足：1≤a≤L，1≤b≤W1≤a≤L，1≤b≤W 。

 

最先想着用循环遍历解决问题：

#include <iostream>

using namespace std;

int main() {
    int L, W;
    while(cin >> L >> W) {
        // 动态创建二维数组
        int **p = new int*[L];
        for(int i = 0; i < L; i++) {
            p[i] = new int[W];
        }

        // 输入每个单位面积上的花生产量
        for(int i = 0; i < L; i++) {
            for(int j = 0; j < W; j++) {
                cin >> p[i][j];
            }
            cin.clear();
        }

        // 输入规定区域大小
        int a, b;
        cin >> a >> b;

        // 根据规定区域大小确定包含区域个数
        int aa = L - a + 1;
        int bb = W - b + 1;
        int **q = new int*[aa];
        for(int i =0; i < a; i++) {
            q[i] = new int[bb];
        }

        // 区域总产量初始化
        for(int i = 0; i < aa; i++) {
            for(int j = 0; j < bb; j++) {
                q[i][j] = 0;
            }
        }

        // 根据规定区域大小创建二维数据记录该区域总产量
        for(int i = 0; i < aa; i++) {
            for(int j = 0; j < bb; j++) {
                for(int k = i; k < a; k++) {
                    for(int l = j; l < b; l++) {
                        q[i][j] += p[k][l];
                    }
                }
            }
        }

        int maxValue = 0;
        for(int i = 0; i < aa; i++) {
            for(int j = 0; j < bb; j++) {
                if(q[i][j] > maxValue)
                    maxValue = q[i][j];
            }
        }

        cout << maxValue << endl;
    }
}

 

四重循环提交超时，上网查找了一些优化方法，都是从减少产量的计算量出发（每个区域的产量实际上是重复计算了）：

1.空间换时间，利用数据存储每列的产量，矩形区域的产量=上一相邻区域产量+新增列产量-删除列产量。

2.计算前缀和，即a[i][j] = a[i][j] + a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1]，处理好位于边缘的点的计算（左、上、左上位置的点不存在则去除对应项的计算），前缀和大幅度减少计算量。

 

利用前缀和解题代码：

#include <iostream>

using namespace std;

int main() {
    int L, W;
    while(cin >> L >> W) {
        // 动态创建二维数组
        int **p = new int*[L];
        for(int i = 0; i < L; i++) {
            p[i] = new int[W];
        }

        // 输入每个单位面积上的花生产量
        for(int i = 0; i < L; i++) {
            for(int j = 0; j < W; j++) {
                cin >> p[i][j];
                // 【前缀和】
                if(i > 0 && j >0)
                    p[i][j] = p[i - 1][j] + p[i][j - 1] - p[i - 1][j - 1] + p[i][j];
                else if(i == 0 && j > 0)
                    p[i][j] = p[i][j - 1] + p[i][j];
                else if(j == 0 && i > 0)
                    p[i][j] = p[i - 1][j] + p[i][j];
                else if(i == 0 && j == 0)
                    p[i][j] = p[i][j];
            }
            cin.clear();
        }

        // 输入规定区域大小
        int a, b;
        cin >> a >> b;


        int maxValue = 0;

        // 计算并找到最大产量
        for(int i = a - 1; i < L; i++) {
            for(int j = b - 1; j < W; j++) {
                int temp = 0;
                if(i == a - 1 && j != b - 1)
                    temp = p[i][j] - p[i][j - b];
                else if(i != a - 1 && j == b - 1)
                    temp = p[i][j] - p[i - a][j];
                else if(i == a - 1 && j ==b - 1)
                    temp = p[i][j];
                else
                    temp = p[i][j] - p[i][j - b] - p[i - a][j] + p[i - a][j - b];
                maxValue = temp > maxValue ? temp : maxValue;
            }
        }
        cout << maxValue << endl;
    }
}

yeah，一道容易超时的题。