hihoCoder1502蓝桥算法基础第四章测验题目3最大子矩阵Java代码

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

解析
/**

• 假定只有一行，那就和求”和不超过K的最长子数组的长度“一样
• 如果限定两行，可以把两行按列求和，同上
• ……
• 所有我们从把第一行当做起点，依次累加后面的每一行后，都求一个”和不超过K的最长子数组的长度“*累加行数
• 以第二行作为起点，依次累加后面的每一行后，都求一个”和不超过K的最长子数组的长度“*累加行数
• 每次求出来的和与历史最大值比较，如果更大，则更新
  */

import java.util.Scanner;
import java.util.Arrays;
public class Main {
	static int k;
	static int n;
	static int m;
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n=sc.nextInt();
		m=sc.nextInt();
		k=sc.nextInt();
		int [][]mt=new int[n][m];
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++) {
				mt[i][j]=sc.nextInt();
			}
		}
		int res=findMax(mt);
		System.out.println(res);
	}
	private static int findMax(int[][] mt) {
		int stR=0;//以它为起始行
		int max=0;//历史最大的”和不超过K的最长子数组“*累加行数
		int []sums=new int[m];//按列求和
		while(stR<n) {//起始行
			for(int i=stR;i<n;i++) {//从起始行到第i行
				//按列累加
				for(int j=0;j<m;j++) {
					sums[j]+=mt[i][j];
				}
				//  累加完成
		        //  求出sums的”和不超过K的最长子数组的长度“*累加行数
				int t=findByDp(sums)*(i-stR+1);
				if(t>max)
					max=t;
			}
			//另起一行作为起始行.把sums清零
			Arrays.fill(sums, 0);//快速地将sums的每个元素都设定为0
			stR++;
		}
		return max;
	}
	private static int findByDp(int[] sums) {
		int sum=sums[0],max,l=0;
		if(sums[0]<=k) {
			max=1;
		}else {
			max=0;
		}
		for(int i=1;i<sums.length;i++) {
			if(sum+sums[i]<=k) {// 左子数组加上当前元素的和小于等于K，继续累加
				sum+=sums[i];
			}else {//左子数组加上当前元素的和大于K
				l++;
				i=l;//丢弃第一个元素，以下一个元素为第一个元素继续计算
				sum=sums[i];//丢弃前部分和，和初始为下一个元素
				continue;
			}
			if(i-l+1>max) {//更新最大长度
				max=i-l+1;
			}
		}
		return max;
	}
}