Cod sursa(job #3131419)

Utilizator NiffSniffCojocaru Calin Marcu NiffSniff Data 20 mai 2023 00:31:26
Problema Plantatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.93 kb
#include <fstream>
#include <vector>
using namespace std;
string file = "plantatie";
ifstream cin (file + ".in");
ofstream cout (file + ".out");
const int N = 500;
int dp[9][N+1][N+1];

vector <int> p;
void putere(int n)
{
    p.push_back(1);
    for (int i=2; i<=n; i=i<<1)
    {
        p.push_back(i);
    }
}

int cb (int x)
{
    int st = 1, dr = p.size()-1;
    while (st <= dr)
    {
        int mid = (st+dr)/2;
        if (p[mid] <= x)
        {
            st = mid+1;
        }
        else
        {
            dr = mid-1;
        }
    }
    return dr;
}
void rmq(int n)
{
    for (int k=1; k<p.size(); k++)
    {
        for (int i=p[k]; i<=n; i++)
        {
            for (int j=p[k]; j<=n; j++)
            {
                int x = dp[k-1][i-p[k-1]][j], y = dp[k-1][i][j-p[k-1]], z = dp[k-1][i-p[k-1]][j-p[k-1]];
                dp[k][i][j] = max(max(dp[k-1][i][j],x),max(y,z));
            }
        }
    }
}

int interogare(int i, int j, int k)
{
    int e = cb(k), x = p[e];
    //cout << e << ' ' << x << ' ' << i+x-1 << ' ' << i+k-1 << ' ' << j+x-1 << ' ' << j+k-1 << ' ';
    int poz1 = dp[e][i+x-1][j+x-1]; // stanga sus
    int poz2 = dp[e][i+k-1][j+x-1]; // stanga jos
    int poz3 = dp[e][i+x-1][j+k-1]; // dreapta sus
    int poz4 = dp[e][i+k-1][j+k-1]; // dreapta jos
    return max(poz1,max(poz2,max(poz3,poz4)));
}

void afisare(int k, int n)
{
    for (int i=1; i<=n; i++)
    {
        for (int j=1; j<=n; j++)
        {
            cout << dp[k][i][j] << ' ';
        }
        cout << '\n';
    }
    cout << endl;
}
int main ()
{
    int n, m, x, y, z;
    cin >> n >> m;
    putere(n);
    for (int i=1; i<=n; i++)
    {
        for (int j=1; j<=n; j++)
        {
            cin >> dp[0][i][j];
        }
    }
    rmq(n);
    for (int i=1; i<=m; i++)
    {
        cin >> x >> y >> z;
        cout << interogare(x,y,z) << '\n';
    }
}