[数据结构与算法] 蓝桥杯2022统计子矩阵

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 数据结构与算法 -> 蓝桥杯2022统计子矩阵_二维前缀和 -> 正文阅读

[数据结构与算法]蓝桥杯2022统计子矩阵_二维前缀和

P2036 - [蓝桥杯2022初赛] 统计子矩阵 - New Online Judgehttp://oj.ecustacm.cn/problem.php?id=2036暴力能过80%的数据。。。

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
ll a[512][512] = { 0 };
ll sum[512][512] = { 0 }; // sum[i][j]表示从a[1][1]到a[i][j]的和
int main() {
    // system("chcp 65001");
    ios::sync_with_stdio(false);
    cin.tie(0);
    // freopen("C:/Users/zhaochen/Desktop/input.txt", "r", stdin);
    int n, m, k;
    cin >> n >> m >> k;
    ll ans = 0;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            cin >> a[i][j];
        }
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + a[i][j];
        }
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) { // 左上角
            for (int p = i; p <= n; p++) {
                for (int q = j; q <= m; q++) { // 右下角
                    ll t = sum[p][q] - sum[p][j - 1] - sum[i - 1][q] + sum[i - 1][j - 1];
                    if (t <= k) {
                        ans++;
                    } else {
                        break;
                    }
                }
            }
        }
    }
    cout << ans;
    return 0;
}