Cod sursa(job #2706214)

Utilizator alessiamtr12Mitrica Alessia alessiamtr12 Data 14 februarie 2021 12:01:57
Problema Barbar Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.25 kb
#include <fstream>
#include<queue>
#include<cstring>
using namespace std;
ifstream fin("barbar.in");
ofstream fout("barbar.out");
int d[1001][1001],n,m,st,dr,mij,i1,j1,i2,j2,sol;
int dx[]={0,1,0,-1},dy[]={1,0,-1,0};
bool viz[1001][1001];
struct elem
{
    int l,c;
}poz,poz1;
bool interior(int i,int j)
{
    if(i<1||i>n)
        return 0;
    if(j<1||j>m)
        return 0;
    return 1;
}
queue<elem>q;
char c;
void lee1()
{
    while(!q.empty())
    {
        poz=q.front();
        q.pop();
        for(int k=0;k<4;k++)
        {
            poz1.l=poz.l+dx[k];
            poz1.c=poz.c+dy[k];
            if(interior(poz1.l,poz1.c)&&(d[poz1.l][poz1.c]==0||d[poz1.l][poz1.c]>d[poz.l][poz.c]+1)&&d[poz1.l][poz1.c]!=-1)
               {
                   d[poz1.l][poz1.c]=d[poz.l][poz.c]+1;
                   q.push(poz1);
               }
        }
    }
}
int lee2(int x)
{
    if(d[i1][j1]<x)
        return 0;
    memset(viz,0,sizeof(viz));
    while(!q.empty())
        q.pop();
    viz[i1][j1]=1;
    q.push({i1,j1});
    while(!q.empty())
    {
        poz=q.front();
        q.pop();
        for(int k=0;k<4;k++)
        {
            poz1.l=poz.l+dx[k];
            poz1.c=poz.c+dy[k];
            if(interior(poz1.l,poz1.c)&&viz[poz1.l][poz1.c]==0&&d[poz1.l][poz1.c]>=x)
            {

                viz[poz1.l][poz1.c]=1;
                if(poz1.l==i2&&poz1.c==j2)
                    return 1;
                q.push(poz1);
            }
        }
    }
    return 0;
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
    {
        fin>>c;
        if(c=='*')
            d[i][j]=-1;
        else
            if(c=='I')
        {
            i1=i;
            j1=j;
        }
        else
            if(c=='D')
        {
            d[i][j]=1;
            q.push({i,j});
        }
        else
            if(c=='O')
        {
            i2=i;
            j2=j;
        }
    }
    lee1();
    st=1;
    dr=n;
    sol=-1;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        if(lee2(mij)==1)
        {
            sol=mij-1;
            st=mij+1;
        }
        else
            dr=mij-1;
    }
    fout<<sol;
    return 0;
}