HDU3630----最大子矩阵+二维RMQ

来源:互联网 发布:初中生网络安全教育 编辑:程序博客网 时间:2024/04/28 05:26

地址:http://acm.hdu.edu.cn/showproblem.php?pid=3630

题目意思:

给你一个矩阵N*M

有Q次查询,然后问你一个子矩阵里面的最大收益

收益的算法是:这个子矩阵里面找一个不含-1的矩阵,然后求他的和,其中可以选一个点,是其值变为S倍,显然,选最大的最好

解题思路:

这个题分两步:

先求最大子矩阵,然后用二维RMQ求出最大值就OK,其中求和可以预处理

我之前一直TLE,过了快一个星期才查出来是在RMQ查询的时候被LOG/LOG把时间拖死了

对于这个,打表解决就OK

下面上代码:

#include<iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<cmath>using namespace std;const int maxn = 200;int a[maxn][maxn];int l[maxn][maxn],r[maxn][maxn];int h[maxn][maxn];int _sum[maxn][maxn];int n,m,q,s;int dp[maxn][maxn][9][9];int power[maxn];void init_rmq(){    for(int i=1;i<=n;i++)    {        for(int j=1;j<=m;j++)        {            dp[i][j][0][0]=a[i][j];        }    }    int n1=log(double(n))/log(2.0);    int m1=log(double(m))/log(2.0);    for(int i=1;i<maxn;i++) power[i] = log(i*1.0)/log(2.0);    for(int i=0;i<=n1;i++)    {        for(int j=0;j<=m1;j++)        {            if(i==0 && j==0)                continue;            for(int row=1;row+(1<<i)-1<=n;row++)            {                for(int col=1;col+(1<<j)-1<=m;col++)                {                    if(i==0)                    {                        dp[row][col][i][j] =                            max(dp[row][col][i][j-1],dp[row][col+(1<<(j-1))][i][j-1]);                    }                    else                    {                        dp[row][col][i][j] =                            max(dp[row][col][i-1][j],dp[row+(1<<(i-1))][col][i-1][j]);                    }                }            }        }    }}int query(int x1,int y1,int x2,int y2){    int kx=power[x2-x1+1];    int ky=power[y2-y1+1];    int m1=dp[x1][y1][kx][ky];    int m2 = dp[x2-(1<<kx)+1][y1][kx][ky];    int m3 = dp[x1][y2-(1<<ky)+1][kx][ky];    int m4 = dp[x2-(1<<kx)+1][y2-(1<<ky)+1][kx][ky];    return max(max(m1,m2), max(m3,m4) );}void pre(){    memset(h,0,sizeof(h));    for(int i=1;i<=n;i++)    {        for(int j=1;j<=m;j++)        {            if(a[i][j]>=0)            {                h[i][j]=h[i-1][j]+1;            }        }    }    for(int i=1;i<=n;i++)    {        h[i][0] = -1;        h[i][m+1] = -1;        l[i][1]=1;        r[i][m]=m;        for(int j=2;j<=m;j++)        {            l[i][j]=j;            if(h[i][j]==0)                continue;            while(h[i][l[i][j]-1] >= h[i][j])            {                l[i][j]=l[i][l[i][j]-1];            }        }        for(int j=m-1;j>=1;j--)        {            r[i][j]=j;            if(h[i][j]==0)                continue;            while(h[i][r[i][j]+1] >= h[i][j])                r[i][j] = r[i][r[i][j]+1];        }    }}int query_ans(int x1,int y1,int x2,int y2){    int ans=0;    for(int i=x1;i<=x2;i++)    {        for(int j=y1;j<=y2;j++)        {            if(h[i][j]==0)                continue;            int a1=i-h[i][j]+1>=x1?i-h[i][j]+1:x1;            int b1=l[i][j]>=y1?l[i][j]:y1;            int a2=i;            int b2=r[i][j]<=y2?r[i][j]:y2;            int tmp=_sum[a2][b2]-_sum[a2][b1-1]-_sum[a1-1][b2]+_sum[a1-1][b1-1];            int _max=query(a1,b1,a2,b2);            if(_max>0)                tmp=tmp+(s-1)*_max;            if(ans<tmp)                ans=tmp;        }    }    return ans;}int main(){    int t;    scanf("%d",&t);    int cnt=1;    while(t--)    {        if(cnt>1)            printf("\n");        printf("Case %d:\n",cnt++);        scanf("%d%d%d%d",&n,&m,&q,&s);        memset(_sum,0,sizeof(_sum));        for(int i=1;i<=n;i++)        {            for(int j=1;j<=m;j++)            {                scanf("%d",&a[i][j]);                _sum[i][j] = _sum[i][j-1]+_sum[i-1][j]-_sum[i-1][j-1]+a[i][j];            }        }        pre();        init_rmq();        int x1,y1,x2,y2;        for(int i=1;i<=q;i++)        {            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);            printf("%d\n",query_ans(x1,y1,x2,y2));        }    }    return 0;}