Cod sursa(job #1070915)

Utilizator andreiiiiPopa Andrei andreiiii Data 2 ianuarie 2014 12:25:10
Problema Barbar Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.51 kb
#include <fstream>
#include <cstring>
#include <bitset>
#include <queue>
#define INF 0x3f3f3f3f
#define PII pair<int, int>
#define x first
#define y second

using namespace std;

const int N=1005;

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

bitset <N> v[N];
char a[N][N];
int d[N][N], dx[]={-1, 0, 1, 0}, dy[]={0, 1, 0, -1};
int n, m;
PII start, finish;

bool verif(int dist)
{
    if(d[start.x][start.y]<dist) return 0;
    int i;
    PII p;
    queue <PII> Q;
    for(i=1;i<=n;i++) v[i].reset();
    v[start.x][start.y]=1;
    for(Q.push(start);!Q.empty();Q.pop())
    {
        p=Q.front();
        for(i=0;i<4;i++)
        {
            if(a[p.x+dx[i]][p.y+dy[i]]=='.'&&d[p.x+dx[i]][p.y+dy[i]]>=dist&&!v[p.x+dx[i]][p.y+dy[i]])
            {
                if(p.x+dx[i]==finish.x&&p.y+dy[i]==finish.y) return 1;
                v[p.x+dx[i]][p.y+dy[i]]=1;
                Q.push(make_pair(p.x+dx[i], p.y+dy[i]));
            }
        }
    }
    return 0;
}

void lee()
{
    queue <PII> Q;
    memset(d, INF, sizeof(d));
    int i, j;
    PII p;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=m;j++)
        {
            if(a[i][j]=='D')
            {
                d[i][j]=0;
                Q.push(make_pair(i, j));
            }
        }
    }
    for(;!Q.empty();Q.pop())
    {
        p=Q.front();
        for(i=0;i<4;i++)
        {
            if(a[p.x+dx[i]][p.y+dy[i]]=='.'&&d[p.x+dx[i]][p.y+dy[i]]>d[p.x][p.y]+1)
            {
                d[p.x+dx[i]][p.y+dy[i]]=d[p.x][p.y]+1;
                Q.push(make_pair(p.x+dx[i], p.y+dy[i]));
            }
        }
    }
    /*for(i=1;i<=n;i++)
    {
        for(j=1;j<=m;j++)
        {
            if(d[i][j]!=INF) fout<<d[i][j]<<" ";
            else fout<<"-1 ";
        }
        fout<<"\n";
    }*/
}

int main()
{
    int i, j, sol=-1, st=0, dr, mij;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>(a[i]+1);
        for(j=1;j<=m;j++)
        {
            if(a[i][j]=='I')
            {
                a[i][j]='.';
                start=make_pair(i, j);
            }
            else if(a[i][j]=='O')
            {
                finish=make_pair(i, j);
                a[i][j]='.';
            }
        }
    }
    lee();
    dr=n*m;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        if(verif(mij))
        {
            sol=mij;
            st=mij+1;
        }
        else dr=mij-1;
    }
    fout<<sol;
    fin.close();
    fout.close();
}