Cod sursa(job #353200)

Utilizator bent_larsenSturzu Antonio-Gabriel bent_larsen Data 4 octombrie 2009 14:04:47
Problema Castel Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.44 kb
#include<stdio.h>
#include<vector>
#include<queue>
#include<string.h>

using namespace std;

int a[151][151],n,m,sx,sy;

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

int bfs()
{
	queue< pair<int,int> > chei[25000];
	queue< pair<int,int> > coada;
	pair<int,int> nod,nod1;
	int key,vizitat[25000],total=0,x,y;

	memset(vizitat,0,sizeof(vizitat));
	coada.push(make_pair(sx,sy));
	vizitat[(sx-1)*m+sy]=1;
	vizitat[a[sx][sy]]=1;

	while(coada.size()>0)
	{
		nod=coada.front();
		coada.pop();
		key=(nod.first-1)*m+nod.second;
		total++;
	
		while(chei[key].size()>0)
		{
			nod1=chei[key].front();
			chei[key].pop();
			if(vizitat[(nod1.first-1)*m+nod1.second]==0)
			{
				vizitat[(nod1.first-1)*m+nod1.second]=1;
				coada.push(nod1);
			}
		}

		for(int i=0;i<4;i++)
		{
			x=nod.first+dx[i];
			y=nod.second+dy[i];
			if(x>=1 && x<=n && y>=1 && y<=m && vizitat[(x-1)*m+y]==0)
			{
				if(vizitat[a[x][y]]==1)
				{
					vizitat[(x-1)*m+y]=1;
					coada.push(make_pair(x,y));
				}
				else
					chei[a[x][y]].push(make_pair(x,y));	
			}
		}
	}
	return total;
}
			
int main()
{
	int i,j,k;
	FILE *f=fopen("castel.in","r");
	fscanf(f,"%i%i%i",&n,&m,&k);

	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			fscanf(f,"%i",&a[i][j]);
	fclose(f);

	if(k%m!=0)
	{
		sx=k/m+1;
		sy=k%m;
	}
	else
	{		
		sx=k/m;
		sy=m;
	}


	f=fopen("castel.out","w");
	fprintf(f,"%i\n",bfs());
	fclose(f);
	return 0;
}