Cod sursa(job #447007)

Utilizator loginLogin Iustin Anca login Data 27 aprilie 2010 12:58:48
Problema Barbar Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.85 kb
# include <fstream>
# include <queue>
# include <algorithm>
# include <iostream>
using namespace std;
struct nod{
	int i, j;
	nod(){}
	nod(int I, int J){
		i=I;j=J;}
};
int n, m, a[1003][1003], d[1003][1003], sol, c[1003][1003];
int ist, jst, ifin, jfin, di[4]={0, 0, 1, -1}, dj[4]={-1, 1, 0, 0};
queue<nod>Q;

int in_m(int i, int j)
{
	if (i && j && i<=n && j<=m)return 1;
	return 0;
}

void lee ()
{
	int i, j, ii, jj;
	nod kk;
	while(Q.size())
	{
		kk=Q.front();
		Q.pop();
		i=kk.i;j=kk.j;
		for(int k=0;k<4;++k)
		{
			ii=i+di[k];
			jj=j+dj[k];
			if (a[ii][jj]==0 && in_m(ii, jj) && d[i][j]+1<d[ii][jj])
			{
				d[ii][jj]=d[i][j]+1;
				Q.push(nod(ii, jj));
			}
		}
	}
}

void solve ()
{
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			c[i][j]=10000000;
	Q.push(nod(ist, jst));
	sol=d[ist][jst];
	c[ist][jst]=sol;
	nod kk;
	int i, j, ii, jj;
	while (Q.size())
	{
		kk=Q.front();
		Q.pop();
		i=kk.i;j=kk.j;
		for(int k=0;k<4;++k)
		{
			ii=i+di[k];
			jj=j+dj[k];
			if (in_m(ii, jj) && a[ii][jj]==0 && c[i][j]+1<c[ii][jj])
			{
				if (d[ii][jj]>c[i][j]+1)
					c[ii][jj]=d[ii][jj];
				else
					c[ii][jj]=c[i][j]+1;
				if (ii!=ifin || jj!=jfin)
					Q.push(nod(ii, jj));
			}
		}
	}
}	

int main ()
{
	ifstream fin ("barbar.in");
	ofstream fout ("barbar.out");
	fin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			d[i][j]=1000000000;
	char cc;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			fin>>cc;
			if (cc=='*')a[i][j]=-1;
			if (cc=='D')
			{
				a[i][j]=1;
				d[i][j]=0;
				Q.push(nod(i, j));
			}
			else
			{
				if (cc=='I'){ist=i;jst=j;}
				else if (cc=='O'){ifin=i;jfin=j;}
			}
		}
	lee();
	solve ();
	fout<<c[ifin][jfin];
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
			cout<<c[i][j]<<" ";
		cout<<endl;
	}
	return 0;
}