Pagini recente » Cod sursa (job #1716442) | Cod sursa (job #3184580) | Cod sursa (job #1360544) | Cod sursa (job #3242589) | Cod sursa (job #2563343)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("castel.in");
ofstream fout("castel.out");
int n,m,v[201][201],k,ans;
set<int> sol[151][151];
set<int> reuniune;
queue< pair<int,int> > coada;
bool use[201][201];
int dirx[5]={-1,1,0,0};
int diry[5]={0,0,-1,1};
int main()
{
fin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
fin>>v[i][j];
sol[i][j].insert(m*(i-1)+j);
}
int x=k/m+1;
int y=k%m;
if(y==0)
{
y=m;
x--;
}
coada.push({x,y});
reuniune.insert(k);
use[x][y]=1;
while(!coada.empty())
{
set<int> s=sol[coada.front().first][coada.front().second];
x=coada.front().first;
y=coada.front().second;
use[x][y]=1;
bool nw[5];
memset(nw,0,sizeof(nw));
for(int i=0;i<4;i++)
{
x+=dirx[i];
y+=diry[i];
if(x<=n&&x>0&&y<=m&&y>0)
if(reuniune.find(v[x][y])!=reuniune.end())
{
set<int> ss=sol[x][y];
for(set<int>::iterator it=ss.begin(); it!=ss.end(); it++)
{
reuniune.insert(*it);
}
}
x-=dirx[i];
y-=diry[i];
}
for(int i=0;i<4;i++)
{
x+=dirx[i];
y+=diry[i];
if(x<=n&&x>0&&y<=m&&y>0&&s.find(v[x][y])!=s.end())
{
if(sol[x][y]!=reuniune)
{
sol[x][y]=reuniune;
coada.push({x,y});
}
}
x-=dirx[i];
y-=diry[i];
}
coada.pop();
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
ans+=use[i][j];
fout<<ans;
/*for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(use[i][j])
{
set<int> sss=sol[i][j];
for(auto it=sss.begin();it!=sss.end();it++)
fout<<*it<<" ";
fout<<'\n';
}*/
return 0;
}