Cod sursa(job #1157341)

Utilizator andrettiAndretti Naiden andretti Data 28 martie 2014 13:53:07
Problema Plantatie Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.3 kb
#include<fstream>
#include<algorithm>
#define maxn 505
#define maxlg 25
using namespace std;

ifstream fin("plantatie.in");
ofstream fout("plantatie.out");

int n,m;
int a[maxn][maxn];
int rmq[maxn][maxn][maxlg],lg[maxn];

void read()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
       fin>>a[i][j];
}

void RMQ()
{
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      rmq[i][j][0]=a[i][j];

    for(int i=2;i<=n;i++) lg[i]=lg[i/2]+1;
    int cnt;
    for(int k=1;(1<<k)<=n;k++)
     for(int i=1;i+(1<<k)-1<=n;i++)
      for(int j=1;j+(1<<k)-1<=n;j++)
      {
          cnt=(1<<(k-1));
          rmq[i][j][k]=max(rmq[i][j][k-1],rmq[i][j+cnt][k-1]);
          rmq[i][j][k]=max(rmq[i][j][k],rmq[i+cnt][j][k-1]);
          rmq[i][j][k]=max(rmq[i][j][k],rmq[i+cnt][j+cnt][k-1]);
      }
}

int det(int x,int y,int k)
{
    int sol,cnt;
    int p=lg[k]; cnt=(1<<p);
    sol=max(rmq[x][y][p],rmq[x][y+k-cnt][p]);
    sol=max(sol,rmq[x+k-cnt][y][p]);
    sol=max(sol,rmq[x+k-cnt][y+k-cnt][p]);
    return sol;
}

void solve()
{
    int x,y,z;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>z;
        fout<<det(x,y,z)<<'\n';
    }
}

int main()
{
    read();
    RMQ();
    solve();

    fin.close();
    fout.close();
    return 0;
}