Cod sursa(job #3131422)

Utilizator NiffSniffCojocaru Calin Marcu NiffSniff Data 20 mai 2023 00:37:41
Problema Plantatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 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], np;

vector <int> p;

inline int cb (int x)
{
    int st = 1, dr = np-1;
    while (st <= dr)
    {
        int mid = (st+dr)/2;
        if (p[mid] <= x)
        {
            st = mid+1;
        }
        else
        {
            dr = mid-1;
        }
    }
    return dr;
}
int main ()
{
    int n, m,x,k,i,j,e;
    cin >> n >> m;
    p.push_back(1);
    for (i=2; i<=n; i=i<<1)
    {
        p.push_back(i);
    }
    np = p.size();
    for (i=1; i<=n; i++)
    {
        for (j=1; j<=n; j++)
        {
            cin >> dp[0][i][j];
        }
    }
    for (k=1; k<np; k++)
    {
        for (i=p[k]; i<=n; i++)
        {
            for (j=p[k]; j<=n; j++)
            {
                dp[k][i][j] = max(max(dp[k-1][i][j],dp[k-1][i-p[k-1]][j]),max(dp[k-1][i][j-p[k-1]],dp[k-1][i-p[k-1]][j-p[k-1]]));
            }
        }
    }
    while (m--)
    {
        cin >> i >> j >> k;
        e = cb(k), x = p[e];
        cout << max(max(dp[e][i+x-1][j+x-1],dp[e][i+k-1][j+x-1]),max(dp[e][i+x-1][j+k-1],dp[e][i+k-1][j+k-1])) << '\n';
    }
}