Pagini recente » Cod sursa (job #687552) | Cod sursa (job #2051590) | Cod sursa (job #2087953) | Cod sursa (job #1495488) | Cod sursa (job #2563359)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("castel.in");
ofstream fout("castel.out");
int n,m,v[201][201],k,ans;
vector<int> nums;
set<int> sol[151][151];
set<int> reuniune,toreach;
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});
use[x][y]=1;
while(!coada.empty())
{
x=coada.front().first;
y=coada.front().second;
reuniune.insert((x-1)*m+y);
use[x][y]=1;
for(int i=0;i<4;i++)
{
x+=dirx[i];
y+=diry[i];
if(x>0&&x<=n&&y>0&&y<=m)
toreach.insert((x-1)*m+y);
x-=dirx[i];
y-=diry[i];
}
nums.clear();
for(auto it=toreach.begin();it!=toreach.end();it++)
{
int nr=*it;
x=nr/m+1;
y=nr%m;
if(y==0)
{
y=m;
x--;
}
if(reuniune.find(v[x][y])!=reuniune.end()&&use[x][y]==0)
{
coada.push({x,y});
nums.push_back(nr);
}
}
for(int i=0;i<nums.size();i++)
toreach.erase(nums[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;
}