Cod sursa(job #2093977)

Utilizator iDanyelArvat Ovidiu Daniel iDanyel Data 24 decembrie 2017 18:47:03
Problema Barbar Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.35 kb
#include <iostream>
#include <fstream>
#include <queue>
using namespace std;

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

queue < pair < int, int > > Q;

int dx[4]={-1,0,1,0};
int dy[4]={0,1,0,-1};

int n, m, k, v[1010][1010], S[1010][1010];
int x1, y1, x2, y2;

void reset()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            S[i][j]=v[i][j];
}

bool cauta_dragon(int x, int y, int P)
{
    //Pe Nord

    for(int i=x;i>=x-P+1&&i>=1;i--)
        if(S[i][y]==-2)
            return true;

    //Pe Sud

    for(int i=x;i<=x+P-1&&i<=n;i++)
        if(S[i][y]==-2)
            return true;

    //Pe Vest

    for(int j=y;j>=y-P+1&&y>=1;y--)
        if(S[x][j]==-2)
            return true;

    //Pe Est

    for(int j=y;j<=y+P-1&&y<=m;y++)
        if(S[x][j]==-2)
            return true;

    return false;
}

bool inauntru(int x, int y)
{
    if(x<1||y<1||x>n||y>m)
        return false;

    return true;
}

bool passed(int x, int y, int P)
{
    reset();

    S[x][y]=1;
    Q.push(make_pair(x,y));

    while(!Q.empty())
    {
        int ii=Q.front().first, jj=Q.front().second;
        Q.pop();

        for(int dir=0;dir<4;dir++)
            if(inauntru(ii+dx[dir],jj+dy[dir])&&!S[ii+dx[dir]][jj+dy[dir]])
            {
                if(cauta_dragon(ii+dx[dir],jj+dy[dir],P))
                    return false;
                else
                {
                    S[ii+dx[dir]][jj+dy[dir]]=S[ii][jj]+1;
                    Q.push(make_pair(ii+dx[dir],jj+dy[dir]));
                }
            }
    }

    if(S[x2][y2]>0)
        return true;

    return false;
}

int binary()
{
    int li=1, lf=n, MAX=0;

    while(li<=lf)
    {
        int mid=(li+lf)/2;

        if(passed(x1,y1,mid))
            MAX=mid, li=mid+1;
        else
            lf=mid-1;
    }

    return MAX;
}

int main()
{
    fin >> n >> m;

    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            char c;
            fin >> c;

            if(c=='D')
                v[i][j]=-2;

            else if(c=='*')
                v[i][j]=-1;

            else if(c=='I')
                x1=i, y1=j;

            else if(c=='O')
                x2=i, y2=j;
        }

    int value=binary();

    fout << value;
    return 0;
}