Cod sursa(job #2742945)

Utilizator mateilazarescumateilazarescu mateilazarescu Data 22 aprilie 2021 12:59:13
Problema Castel Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.2 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,b,c;
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++)
        {
            b=x+vi[i];
            c=y+vj[i];
            cnt=(b-1)*m+c;
            if(b<1 or b>n or c<1 or c>m)
                continue;
            if(!used[cnt] and a[mat[b][c]])
            {
                used[cnt]=true;
                q.push(cnt);
            }
            else
                v[mat[b][c]].push_back(cnt);
        }
    }
    fout<<rez;
    return 0;
}