Cod sursa(job #356628)

Utilizator stocarulCosmin-Mihai Tutunaru stocarul Data 15 octombrie 2009 17:35:47
Problema Castel Scor 30
Compilator c Status done
Runda Arhiva de probleme Marime 1.91 kb
#include<stdio.h>
#define infile "castel.in"
#define outfile "castel.out"
#define nmax 151
#define nnmax (nmax*nmax)
struct coada
{
	int c[nnmax]; //coada
	int n; //numarul de elemente din coada
} c[2];
char v[nnmax]; //v[i]=1 daca camera i a fost vizitata
int h[nnmax]; //harta
int m,n; //numarul de linii, respectiv coloane
int x; //camera in care se afla printesa
int nr; //numarul de camere pe care le poate deschide printesa

inline void add_vecini_coada(int c[nnmax], int *nr, int x)
{
	if(x-n>0 && !v[x-n]) c[++(*nr)]=x-n; //deasupra
	if(x+n<=n*m && !v[x+n]) c[++(*nr)]=x+n; //dedesubt
	if(x-1>0 && ((x-1)%n) && !v[x-1]) c[++(*nr)]=x-1; //stanga
	if(x+1<=n*m && (x%n) && !v[x+1]) c[++(*nr)]=x+1; //dreapta
}

void read()
{
	int i,j;
	scanf("%d %d %d\n",&m,&n,&x);
	for(i=1;i<=m;i++)
		for(j=1;j<=n;j++)
			scanf("%d",&h[(i-1)*n+j]);
}

void init()
{
	v[x]=1; //marcam ca vizitata camera initiala
	nr++; //crestem camerele vizitate
	add_vecini_coada(c[0].c,&c[0].n,x); //adaugam vecinii in coada
}

void solve()
{
	int i,j,nr2=0;
	for(i=0;nr2<nr;i++) //cat timp numarul total anterior este mai mic decat numarul total curent
	{ //i-este folosit pentru a sti cu caare coada lucram ;))
		nr2=nr; //numarul curent devine anterior
		c[(i+1)%2].n=0; //coada cealalta devine nulla
		for(j=1;j<=c[i%2].n;j++) //parcurgem coada
			if(v[h[c[i%2].c[j]]] && !v[c[i%2].c[j]]) //daca pot intra in acesta camera, si nu am intrat inca
			{
				v[c[i%2].c[j]]=1; //marcam ca am intrat
				nr++; //crestem numarul de camere vizitate
				add_vecini_coada(c[(i+1)%2].c,&c[(i+1)%2].n,c[i%2].c[j]); //adaugam vecinii in coada cealalta
			}
			else if(!v[c[i%2].c[j]]) c[(i+1)%2].c[++c[(i+1)%2].n]=c[i%2].c[j]; //daca nu a fost rezolvat, il punem in coada cealalta
	}
}

void write()
{
	printf("%d\n",nr);
}

int main()
{
	freopen(infile,"r",stdin);
	freopen(outfile,"w",stdout);
	
	read();
	init();
	solve();
	write();
	
	fclose(stdin);
	fclose(stdout);
	return 0;
}