Cod sursa(job #2742940)

Utilizator mateilazarescumateilazarescu mateilazarescu Data 22 aprilie 2021 12:55:28
Problema Castel Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.22 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("castel.in");
ofstream fout("castel.out");
int n,m,k,i,j,cnt,x,y,rez,xn,yn;
int mat[155][155];
vector <int> v[22505];
bool used[22505],a[22505];
queue <int> q;
int vi[4]={-1,0,1,0};
int vj[4]={0,1,0,-1};
int main()
{
    fin>>n>>m>>k;
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++) fin>>mat[i][j];
    q.push(k);
    used[k]=true;
    while(!q.empty())
    {
        cnt=q.front();
        q.pop();
        x=cnt/m;
        if(cnt%m!=0) x++;
        y=cnt%m;
        if(cnt%m==0) y=m;
        a[cnt]=true;
        rez++;
        for(i=0;i<v[cnt].size();i++)
        {
            if(!used[v[cnt][i]])
                q.push(v[cnt][i]);
            used[v[cnt][i]]=true;
        }
        v[cnt].clear();
        for(i=0;i<4;i++)
        {
            xn=x+vi[i];
            yn=y+vj[i];
            cnt=(xn-1)*m+yn;
            if(xn<1 or xn>n or yn<1 or yn>m)
                continue;
            if(!used[cnt] and a[mat[xn][yn]])
            {
                used[cnt]=true;
                q.push(cnt);
            }
            else
                v[mat[xn][yn]].push_back(cnt);
        }
    }
    fout<<rez;
    return 0;
}