Submatrix Sum equals to K

63 篇文章 0 订阅

给一个matrix, all elements are positive,问有没有个sub rectangle加起来和等于target。return true/false。


package matrix;

import java.util.*;

public class SubMatrixSum {
	// http://www.1point3acres.com/bbs/forum.php?mod=viewthread&tid=164888&extra=page%3D6%26filter%3Dsortid%26sortid%3D311%26searchoption%255B3046%255D%255Bvalue%255D%3D2%26searchoption%255B3046%255D%255Btype%255D%3Dradio&page=1
	// http://www.1point3acres.com/bbs/forum.php?mod=viewthread&tid=167234&pid=2505700&page=1&extra=page%3D5%26filter%3Dsortid%26sortid%3D311%26searchoption%255B3046%255D%255Bvalue%255D%3D2%26searchoption%255B3046%255D%255Btype%255D%3Dradio#pid2505700
	public static void main(String[] args) {
		int[][] matrix = { { 3, 5, 6, 7 }, { 4, -2, 14, 3 }, { 5, -10, -2, 7 }, { 6, 15, -21, 4 } };
		submatrixSum(matrix, 23);
	}

	public static int[][] submatrixSum(int[][] matrix, int target) {
		int[][] result = new int[2][2];
		int M = matrix.length;
		if (M == 0)
			return result;
		int N = matrix[0].length;
		if (N == 0)
			return result;
		// pre-compute: sum[i][j] = sum of submatrix [(0, 0), (i, j)]
		int[][] sum = new int[M + 1][N + 1];
		for (int j = 0; j <= N; ++j)
			sum[0][j] = 0;
		for (int i = 1; i <= M; ++i)
			sum[i][0] = 0;
		for (int i = 0; i < M; ++i) {
			for (int j = 0; j < N; ++j)
				sum[i + 1][j + 1] = matrix[i][j] + sum[i + 1][j] + sum[i][j + 1] - sum[i][j];
		}

		for (int l = 0; l < M; ++l) {
			for (int h = l + 1; h <= M; ++h) {
				Map<Integer, Integer> map = new HashMap<Integer, Integer>();
				for (int j = 0; j <= N; ++j) {
					// int diff = sum[h][j] - sum[l][j];
					// if (map.containsKey(diff - target)) {
					// int k = map.get(diff - target);
					int diff = sum[h][j] - sum[l][j];
					if (map.containsKey(diff - target)) {
						int k = map.get(diff - target);
						result[0][0] = l;
						result[0][1] = k;
						result[1][0] = h - 1;
						result[1][1] = j - 1;
						return result;
					} else {
						map.put(diff, j);
					}
				}
			}
		}
		return result;
	}
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值