HDU2888 Check Corners

来源:互联网 发布:虹口js漂亮的油压店 编辑:程序博客网 时间:2024/05/22 06:17

二维RMQ问题

题意是输入一个矩阵,然后再输入Q条指令,每条指令是一个子矩阵的两个对角顶点,让求这个子矩阵元素的最大值。

然后看这个最大值和子矩阵的4个顶点有没有相同的,如果有,再输出“yes”,没有输出“no”。

这道题也算是一个模板题了,构造出一个四维数组f 【i】【j】【kr】【kc】,表示以矩阵元素m【i】【j】为左上角,以m【i+(1<<kr)】【j+(1<<kc)】为右下角的一个子矩阵中的最大元素值。

另外注意本题内存限制比较紧,四维数组最大好像就是开到f【302】【302】【9】【9】了,开到f【302】【302】【10】【10】就已经MLE了。。。

#include <cstdio>#include <cmath>#include <cstring>#include <iostream>using namespace std;int f[302][302][9][9];int num[302][302];int n,m;void init_RMQ(){    int i,j,r,c;    for(i=1;i<=n;i++)      for(j=1;j<=m;j++)        f[i][j][0][0]=num[i][j];    int kr=(int)(log((double)n)/log(2.0));    int kc=(int)(log((double)m)/log(2.0));    for(i=0;i<=kr;i++)      for(j=0;j<=kc;j++)      {          if(i==0&&j==0) continue;          for(r=1;r+(1<<i)-1<=n;r++)            for(c=1;c+(1<<j)-1<=m;c++)              if(i==0)                f[r][c][i][j]=max(f[r][c][i][j-1],f[r][c+(1<<(j-1))][i][j-1]);              else                f[r][c][i][j]=max(f[r][c][i-1][j],f[r+(1<<(i-1))][c][i-1][j]);      }}int RMQ(int r1,int c1,int r2,int c2){    int kr=(int)(log(double(r2-r1+1))/log(2.0));    int kc=(int)(log(double(c2-c1+1))/log(2.0));    return max(max(f[r1][c1][kr][kc],f[r2-(1<<kr)+1][c1][kr][kc]),max(f[r1][c2-(1<<kc)+1][kr][kc],f[r2-(1<<kr)+1][c2-(1<<kc)+1][kr][kc]));}int main(){    while(scanf("%d%d",&n,&m)!=-1)    {        for(int i=1;i<=n;i++)          for(int j=1;j<=m;j++)            scanf("%d",&num[i][j]);        init_RMQ();        int t,r1,c1,r2,c2;        cin>>t;        while(t--)        {            scanf("%d%d%d%d",&r1,&c1,&r2,&c2);            int ans=RMQ(r1,c1,r2,c2);            printf("%d ",ans);            if(ans==num[r1][c1]||ans==num[r1][c2]||ans==num[r2][c1]||ans==num[r2][c2])              puts("yes");            else  puts("no");        }    }    return 0;}

0 0