#1502 : 最大子矩阵

本文介绍了一种解决最大子矩阵求和问题的算法,通过将矩阵转化为一维数组并运用尺取法,实现了在时间复杂度O(n^3)内找到满足条件的最大子矩阵,适用于特定的数据规模。

最大子矩阵

时间限制: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。

样例输入

3 3 9
1 2 3  
2 3 4  
3 4 5

样例输出

4 

思路:

  • 将以rows为起点的矩阵存入一维sum数组中
  • 之后使用尺取法遍历sum数组来找出满足题目条件的元素数目
  • 时间复杂度O(n^3)满足题目条件

AcCode:

import java.util.Arrays;
import java.util.Scanner;

public class Main{
	public static int getMax(int[][] nums,int k) {
		int maxNum = -1;
		int[] sum = new int[nums[0].length];
		for (int rows = 0; rows < nums.length; rows++) {
			Arrays.fill(sum, 0);
			for (int i = rows; i < nums.length; i++) {
				for (int j = 0; j < nums[i].length; j++) {
					sum[j]+=nums[i][j];
				}
				//尺取法
				int begin = 0;
				int end = begin+1;
				int p = 0;
				int thisSum = sum[begin];
				if(thisSum<k) {
					if((begin+1)*(i+1)>maxNum) {
						maxNum = (begin+1)*(i+1);
					}
				}
				while(begin<sum.length && end<sum.length) {
					if(end!=p) {
						thisSum+=sum[end];
					}
					
					if(thisSum<k) {
						if((end-begin+1)*(i-rows+1)>maxNum) {
							maxNum = (end-begin+1)*(i-rows+1);
							//System.out.println(i+" "+begin+" "+end);
						}
						end++;
					}else {
						p = end;
						thisSum-=sum[begin];
						if(begin==end) {
							p=-1;
							end = begin+1;
							continue;
						}
						begin++;
						
					}
				}
				
			}			
		}
		
		
		
		return maxNum;
	}
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		int N = in.nextInt();
		int M = in.nextInt();
		int k = in.nextInt();
		int[][] nums = new int[N][M];
		for (int i = 0; i < nums.length; i++) {
			for (int j = 0; j < nums[i].length; j++) {
				nums[i][j] = in.nextInt();
			}
		}
		System.out.println(getMax(nums, k));
	}
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值