Cod sursa(job #2480751)

Utilizator TudolHulubei Tudor Tudol Data 26 octombrie 2019 09:26:39
Problema Barbar Scor 70
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.18 kb
#include <bits/stdc++.h>
using namespace std;

pair<int, int> v[1000005];
char a[1005][1005];
int b[1005][1005], n, m, k;
int xi, yi, xo, yo;
/// b[i][j] = distanta minima posibila de la poz.
///           (i, j) la un dragon

bitset<1003> c[1003];

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

void Citire()
{
    fin >> n >> m;
    for (int i = 1; i <= n; i++)
        fin >> (a[i] + 1);
}

void Bordare()
{
    int i;
    /// bordez coloanele 0 si m + 1
    for (i = 0; i <= n + 1; i++)
        a[i][0] = a[i][m + 1] = '*';
    /// bordez liniile 0 si n + 1
    for (i = 0; i <= m + 1; i++)
        a[0][i] = a[n + 1][i] = '*';
}

void Initial()
{
    int i, j;
    for (i = 1; i <= n; i++)
        for (j = 1; j <= m; j++)
        {
            if (a[i][j] == 'I') {xi = i; yi = j;}
            if (a[i][j] == 'O') {xo = i; yo = j;}
            if (a[i][j] == 'D')
            {
                b[i][j] = 0;
                k++;
                v[k] = {i, j}; /// am retinut poziile lui D
            }
            else b[i][j] = 2000000;
        }
}

void Distante()
{
    int i, j, x, y;
    while (k > 0)
    {
        i = v[k].first;
        j = v[k].second;
        k--;
        /// nord:
        x = i - 1;
        y = j;
        if (a[x][y] != '*' && b[x][y] > 1 + b[i][j])
        {
            b[x][y] = 1 + b[i][j];
            k++;
            v[k] = {x, y};
        }
        /// sud:
        x = i + 1;
        y = j;
        if (a[x][y] != '*' && b[x][y] > 1 + b[i][j])
        {
            b[x][y] = 1 + b[i][j];
            k++;
            v[k] = {x, y};
        }
        /// est:
        x = i;
        y = j + 1;
        if (a[x][y] != '*' && b[x][y] > 1 + b[i][j])
        {
            b[x][y] = 1 + b[i][j];
            k++;
            v[k] = {x, y};
        }
        /// vest:
        x = i;
        y = j - 1;
        if (a[x][y] != '*' && b[x][y] > 1 + b[i][j])
        {
            b[x][y] = 1 + b[i][j];
            k++;
            v[k] = {x, y};
        }
    }
}

void Afisare()
{
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
            if (b[i][j] != 2000000)fout << b[i][j] << " ";
            else fout << "X ";
        fout << "\n";
    }
}

/// Verific daca pot ajunge de la I la O
/// mergand doar pe valori >= Val
void Fill(short i, short j, int Val)
{
    if (a[i][j] != '*' && b[i][j] >= Val && c[i][j] == 0)
    {
        c[i][j] = 1;
        Fill(i - 1, j, Val);
        Fill(i + 1, j, Val);
        Fill(i, j - 1, Val);
        Fill(i, j + 1, Val);
    }
}

void Rezolva()
{
    int Val, i;
    for (Val = b[xi][yi]; Val >= 1; Val--)
    {
        for (i = 1; i <= n; i++)
            c[i].reset();
        /// incerc sa vad daca pot ajunge in (xi,yi)
        /// in (xo, yo) mergand doar pe valori >= Val
        Fill(xi, yi, Val);
        if (c[xo][yo] == 1)
        {
            fout << Val << "\n";
            fout.close();
            return;
        }
    }
    fout << "-1\n";
    fout.close();
}

int main()
{
    Citire();
    Bordare();
    Initial();
    Distante();
    Rezolva();
    return 0;
}