hihocoder #1502 : 最大子矩阵

    xiaoxiao2021-04-13  28

    #1502 : 最大子矩阵

    时间限制: 10000ms 单点时限: 1000ms 内存限制: 256MB

    描述

    给定一个NxM的矩阵A和一个整数K,小Hi希望你能求出其中最大(元素数目最多)的子矩阵,并且该子矩阵中所有元素的和不超过K。

    输入

    第一行包含三个整数N、M和K。

    以下N行每行包含M个整数,表示A。

    对于40%的数据,1 <= N, M <= 10  

    对于100%的数据,1 <= N, M <= 250 1 <= K <= 2147483647 1 <= Aij <= 10000

    输出

    满足条件最大的子矩阵所包含的元素数目。如果没有子矩阵满足条件,输出-1

    两个for 枚举 答案矩阵所在的行号,

    由于题目要求的是 元素尽可能多的子矩阵,其和显然也是随之递增的

    因此我们可以二分列的长度,然后把行号固定,所有长度为len的所有矩阵枚举一遍,复杂度 n^3*logn ,1e8还好吧

    或者two pointer(尺取法)一下就可以n^3了 

    #include<bits/stdc++.h> using namespace std; typedef long long ll; const int size = 255; int mat[size][size]; int sum[size][size]; int tok[size]; int n, m, k; int main() { int a; scanf("%d %d %d", &n, &m, &k); for (int i = 1; i <= m; i ++) sum[0][i] = 0; for (int i = 1; i <= n; i ++) for (int j = 1; j <= m; j ++) { scanf("%d", &mat[i][j]); sum[i][j] = sum[i-1][j] + mat[i][j]; } int ans = -1; for (int len = 1; len <= n; len ++) for (int i = 1; i + len - 1 <= n; i ++) { tok[0] = 0; for (int j = 1; j <= m; j ++) { tok[j] = sum[i + len - 1][j] - sum[i - 1][j]; tok[j] += tok[j-1]; } int p1 = 1, p2 = 1; while(p2<=m) { while (tok[p2]-tok[p1-1]<=k&&p2<=m) p2++; if (tok[p2-1]-tok[p1-1]<=k) ans=max(ans,(p2-p1)*len); while (tok[p2]-tok[p1-1]>k&&p1<=p2) p1++; } } if(!ans) ans=-1; assert(ans==-1||ans>0); printf("%d\n", ans); return 0; }

    转载请注明原文地址: https://ju.6miu.com/read-669485.html

    最新回复(0)