Cod sursa(job #3242568)

Utilizator Turcanu_DavidTurcanu David Turcanu_David Data 12 septembrie 2024 16:57:19
Problema Barbar Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.3 kb
#include <bits/stdc++.h>

using namespace std;

using pii = pair<int, int>;

const int MAXN = 1e3+5;
const int INF = 1e9;
// N S E V
int dx[4]={-1, 1, 0, 0};
int dy[4]={0, 0, -1, 1};

int n, m, q;
char mat[MAXN][MAXN];
int vis[MAXN][MAXN];
int d[MAXN][MAXN];

void bfs(queue<pii> q, int maxim, int val)
{
    queue<pii> q2=q;
    while(!q2.empty())
    {
        auto [x, y]=q2.front();
        if(vis[x][y])
            return;
    }
    while(!q.empty())
    {
        auto [x, y]=q.front();
        q.pop();
        for(int i=0; i<4; i++)
        {
            int x1=x+dx[i], y1=y+dy[i];
            if(!vis[x1][y1] && d[x][y]+1 <= maxim)
                q.push({x1, y1}), vis[x1][y1]=val, d[x1][y1]=d[x][y]+1;
        }
    }
}
int x, y, z, t;

bool check(int maxim)
{
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            vis[i][j]=0;
            d[i][j]=0;
        }
    }
    queue<pii> q;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            if(mat[i][j] == 'D')
                q.push({i, j});
        }
    }
    bfs(q, maxim, 1);
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            if(mat[i][j] == '*')
                vis[i][j]=1;
        }
    }
    queue<pii> q2;
    q2.push({x, y});
    bfs(q2, INF, 2);
//    for(int i=0; i<=n+1; i++)
//    {
//        for(int j=0; j<=m+1; j++)
//        {
//            cout<<vis[i][j]<<' ';
//        }
//        cout<<'\n';
//    }
    return (vis[z][t] == 2);
}

int main()
{
    ifstream cin("barbar.in");
    ofstream cout("barbar.out");
    cin>>n>>m;
    char ch;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            cin>>mat[i][j];
            if(mat[i][j] == 'I')
                x=i, y=j;
            if(mat[i][j] == 'O')
                z=i, t=j;
        }
    }
    for(int i=0; i<=n+1; i++)
    {
        vis[i][0]=1;
        vis[i][m+1]=1;
    }
    for(int i=0; i<=m+1; i++)
    {
        vis[0][i]=1;
        vis[n+1][i]=1;
    }
    int st=-1, dr=10*(n+m);
    while(st+1 < dr)
    {
        int mid=(st+dr)/2;
        if(check(mid))
            st=mid;
        else
            dr=mid;
    }
    cout<<st+1;
    return 0;
}