Cod sursa(job #1005527)

Utilizator manutrutaEmanuel Truta manutruta Data 5 octombrie 2013 10:55:00
Problema Barbar Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.83 kb
#include <iostream>
#include <fstream>
#include <bitset>
#include <queue>
#include <set>
#include <bitset>
using namespace std;

#define MAXN 1005
#define INF 0x3f3f3f3f

FILE *f = fopen("barbar.in", "r");
FILE *g = fopen("barbar.out", "w");

int n, m, startx, starty, stopx, stopy, sol;
int a[MAXN][MAXN];
queue< pair<int, int> > q;
int d[4][2] = {{0, -1}, {-1, 0}, {0, 1}, {1, 0}};
priority_queue< pair<int, pair<int, int> > > mst;
//multiset< pair<int, pair<int, int> > > mst;
bitset<MAXN> viz[MAXN];

bool isvalid(int i, int j) {
    if (1 <= i && i <= n && 1 <= j && j <= m && a[i][j] != -1) {
        return true;
    }
    return false;
}

void lee1()
{
    while (!q.empty()) {
        int x = q.front().first;
        int y = q.front().second;
        q.pop();

        for (int i = 0; i < 4; i++) {
            int nx = x + d[i][0];
            int ny = y + d[i][1];

            if (isvalid(nx, ny) && a[nx][ny] > a[x][y] + 1) {
                a[nx][ny] = a[x][y] + 1;
                q.push(make_pair(nx, ny));
            }
        }
    }
}

void lee2()
{
    mst.push(make_pair(a[startx][starty], make_pair(startx, starty)));
    sol = a[startx][starty];
    while (!mst.empty()) {
        int x = mst.top().second.first;
        int y = mst.top().second.second;
        mst.pop();

        if (x == stopx && y == stopy) {
            cout << x << ' ' << y << endl;
            sol = min(sol, a[x][y]);
            break;
        }

        if (viz[x][y] == true) {
            continue;
        }
        viz[x][y] = true;

        cout << x << ' ' << y << endl;

        sol = min(sol, a[x][y]);

        for (int i = 0; i < 4; i++) {
            int nx = x + d[i][0];
            int ny = y + d[i][1];

            if (isvalid(nx, ny)) {
                mst.push(make_pair(a[nx][ny], make_pair(nx, ny)));
            }
        }
    }
}

int main()
{
    fscanf(f, "%d %d\n", &n, &m);
    char buf[MAXN];
    for (int i = 1; i <= n; i++) {
        fgets(buf + 1, sizeof(buf), f);
        for (int j = 1; j <= n; j++) {
            switch (buf[j]) {
                case '.':
                    a[i][j] = INF;
                    break;
                case 'D':
                    q.push(make_pair(i, j));
                    break;
                case 'I':
                    startx = i;
                    starty = j;
                    a[i][j] = INF;
                    break;
                case 'O':
                    stopx = i;
                    stopy = j;
                    a[i][j] = INF;
                    break;
                case '*':
                    a[i][j] = -1;
                    break;
            }
        }
    }

    lee1();
    lee2();

    fprintf(g, "%d\n", sol);

    fclose(f);
    fclose(g);
    return 0;
}