HDU:1559 最大子矩阵

Problem Description:
给你一个m×n的整数矩阵,在上面找一个x×y的子矩阵,使子矩阵中所有元素的和最大。

Input
输入数据的第一行为一个正整数T,表示有T组测试数据。每一组测试数据的第一行为四个正整数m,n,x,y(0<m,n<1000 AND 0<x<=m AND 0<y<=n),表示给定的矩形有m行n列。接下来这个矩阵,有m行,每行有n个不大于1000的正整数。

Output
对于每组数据,输出一个整数,表示子矩阵的最大和。

Sample Input
1
4 5 2 2
3 361 649 676 588
992 762 156 993 169
662 34 638 89 543
525 165 254 809 280

Sample Output
2474

这道题是二维树状数组的模板题,看网上的神犇们大都用的DP,所以我想发一个树状数组的题解

最后查询的时候注意这是个二维的,solve(i,j)是算(i-x,j-y)到(i,j)所有元素的和。所以要减去两个矩形加上一个矩形的面积才是最后的答案

Coding:

#include <bits/stdc++.h>
#define DEBUG freopen("_in_.txt", "r", stdin);
typedef long long ll;
using namespace std;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll maxn = 1000 + 10;

ll c[maxn][maxn];

ll t, x, y, m, n, v, ans;

ll lowbit(ll i)//二进制最低1的位置
{
    return i & (-i);
}

void update(ll x, ll y, ll v) //更新
{
    for (ll i = x; i < maxn; i += lowbit(i))//注意区间的开闭
    {
        for (ll j = y; j < maxn; j += lowbit(j))
        {
            c[i][j] += v;
        }
    }
}

ll query(ll x, ll y)//查询
{
    ll sum = 0;
    for (ll i = x; i > 0; i -= lowbit(i))//注意区间的开闭
    {
        for (ll j = y; j > 0; j -= lowbit(j))
        {
            sum += c[i][j];
        }
    }
    return sum;
}

ll solve(ll i, ll j)//求(i-x,j-y)到(i,j)所有元素的和
{
    return query(i, j) - query(i, j - y) - query(i - x, j) + query(i - x, j - y);
}

int main()
{
    // DEBUG;
    scanf("%lld", &t);
    while (t--)
    {
        memset(c, 0, sizeof(c));
        scanf("%lld%lld%lld%lld", &n, &m, &x, &y);
        for (ll i = 1; i <= n; i++)
        {
            for (ll j = 1; j <= m; j++)
            {
                scanf("%lld", &v);
                update(i, j, v);
            }
        }
        ans = 0;
        for (ll i = 1; i <= n; i++)
        {
            for (ll j = 1; j <= m; j++)
            {
                if (i >= x && j >= y)
                {
                    ans = max(ans, solve(i, j));
                }
            }
        }
        printf("%lld\n", ans);
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值