Mai intai trebuie sa te autentifici.

Cod sursa(job #1799375)

Utilizator vasi461Vasiliu Dragos vasi461 Data 6 noiembrie 2016 11:28:17
Problema Barbar Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.94 kb
#include <fstream>
#include <queue>
using namespace std;

#define INF 1000002
#define NM 1002

ifstream cin("barbar.in");
ofstream cout("barbar.out");

int n, m, x1, y1, x2, y2, d[NM][NM], dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1}, dr[NM][NM];
queue<int> q;
char s[NM][NM];

bool coord(int cx, int cy)
{
    return (cx > 0 and cx <= n and cy > 0 and cy <= m);
}

void mat(int val)
{
    for(int i = 1; i <= n; ++i)
    {
        for(int j = 1; j <= m; ++j)
        {
            if(s[i][j] == '.' or s[i][j] == 'I' or s[i][j] == 'O')
            {
                d[i][j] = val;
            }
            if(s[i][j] == 'D')
            {
                d[i][j] = 0;
            }
            if(s[i][j] == '*')
            {
                d[i][j] = -1;
            }
            if(s[i][j] == 'I')
            {
                x1 = i;
                y1 = j;
            }
            if(s[i][j] == 'O')
            {
                x2 = i;
                y2 = j;
            }
        }
    }
}

void dragon()
{
    mat(INF);
    for(int i = 1; i <= n; ++i)
    {
        for(int j = 1; j <= m; ++j)
        {
            if(s[i][j] == 'D')
            {
                q.push(i);
                q.push(j);
                d[i][j] = 0;
            }
        }
    }
    while(!q.empty())
    {
        int x = q.front();
        q.pop();
        int y = q.front();
        q.pop();
        for(int i = 0; i < 4; ++i)
        {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if(coord(nx, ny))
            {
                if(d[nx][ny] == INF)
                {
                    d[nx][ny] = d[x][y] + 1;
                    q.push(nx);
                    q.push(ny);
                }
            }
        }
    }
    for(int i = 1; i <= n; ++i)
    {
        for(int j = 1; j <= m; ++j)
        {
            dr[i][j] = d[i][j];
        }
    }
}

void calc_dist()
{
    dragon();
    mat(-INF);
    q.push(x1);
    q.push(y1);
    d[x1][y1] = dr[x1][y1];
    while(!q.empty())
    {
        int x = q.front();
        q.pop();
        int y = q.front();
        q.pop();
        for(int i = 0; i < 4; ++i)
        {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if(coord(nx, ny) and s[nx][ny] != '*')
            {
                int dv = min(dr[nx][ny], d[x][y]);
                if(d[nx][ny] < dv)
                {
                    d[nx][ny] = dv;
                    q.push(nx);
                    q.push(ny);
                }
            }
        }
    }
}

int main()
{
    cin >> n >> m;
    cin.getline(s[0], 1002);
    for(int i = 1; i <= n; ++i)
    {
        cin.getline(s[i] + 1, m + 2);
    }
    calc_dist();
    if(d[x2][y2] == -INF)
    {
        cout << -1 << '\n';
    }
        else
        {
            cout << d[x2][y2] << '\n';
        }
    return 0;
}