Cod sursa(job #908260)

Utilizator razvan.popaPopa Razvan razvan.popa Data 8 martie 2013 23:16:56
Problema Barbar Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.02 kb
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<string>
#include<vector>
#include<queue>
#include<deque>
#include<set>
#include<list>
#include<stack>
#include<bitset>
#include<algorithm>
#define INF (1 << 30)
#define pb push_back
#define mkp make_pair
#define pii pair<int, int>
#define ll long long
#define nxt (*it)
#define x first
#define y second
#define type int
#define FOR(i,a,b)\
   for(int i=a; i<=b; ++i)
#define FORR(i,a,b)\
   for(int i=a; i>=b; --i)
#define ALLR(g) \
   for(typeof(g.rbegin()) it=g.rbegin(); it!=g.rend(); ++it)
#define ALL(g)\
   for(typeof(g.begin()) it=g.begin(); it!=g.end(); ++it)
#define infile "barbar.in"
#define outfile "barbar.out"
#define nMax 1005
using namespace std;

 queue < pii > Q;

 priority_queue<pair < int, pii > > PQ;

 pii iPos, fPos;

 int A[nMax][nMax], Path[nMax][nMax];

 int dx[] = {0,1,0,-1};
 int dy[] = {1,0,-1,0};

int N, M;

 void init(){
     FOR(i,1,N)
        FOR(j,1,M)
             Path[i][j] = A[i][j] = INF;
 }

void read(){
    freopen(infile, "r", stdin);

    scanf("%d %d", &N, &M);

    init();

    char c;
    FOR(i,1,N){
        getchar();
        FOR(j,1,M){
            scanf("%c", &c);

            switch(c){
                case 'I':{iPos = mkp(i,j);
                }break;
               case 'D':{Q.push(mkp(i,j));
               A[i][j] = 0;
               } break;
               case '*':{A[i][j] = -1;
               }break;
               case 'O':{fPos = mkp(i,j);
               }break;
            }
        }
    }

    fclose(stdin);
}

 inline bool inMatrix(const pii &p){
     if(p.x >= 1 && p.x <= N && p.y >= 1 && p.y <= M)
          return true;
    return false;
 }

 void BFS(){
     while(!Q.empty()){
          pii nod = Q.front();
          Q.pop();

          FOR(i,0,3){
              pii next = mkp(nod.x + dx[i], nod.y + dy[i]);
              if(!inMatrix(next)|| A[next.x][next.y]  <= A[nod.x][nod.y] + 1)
                   continue;
             A[next.x][next.y] = A[nod.x][nod.y] + 1;
             Q.push(next);
          }
     }
 }

 void Dijkstra(){
     PQ.push(mkp(Path[iPos.x][iPos.y] = A[iPos.x][iPos.y], iPos));

     while(!PQ.empty()){
         pii nod = PQ.top().second;
         PQ.pop();

         FOR(i,0,3){
              pii next = mkp(nod.x + dx[i], nod.y + dy[i]);
              if(!inMatrix(next)|| A[next.x][next.y]  == -1 || Path[next.x][next.y] == INF ||  Path[next.x][next.y]  >= Path[nod.x][nod.y])
                   continue;
             Path[next.x][next.y] = min(A[next.x][next.y], Path[nod.x][nod.y]);
             PQ.push(mkp(Path[next.x][next.y], next));
          }
     }
 }


void solve(){
    BFS();
    Dijkstra();
}

void print(){
    freopen(outfile, "w", stdout);

    if(Path[fPos.x][fPos.y] == INF)
    	printf("-1\n");
    else
    	printf("%d\n",Path[fPos.x][fPos.y]);

    fclose(stdout);
}

int main(){
    read();
    solve();
    print();

    return 0;
}