Cod sursa(job #2838503)

Utilizator ana_valeriaAna Valeria Duguleanu ana_valeria Data 23 ianuarie 2022 20:35:16
Problema Castel Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin ("castel.in");
ofstream cout ("castel.out");
int n,m,k,ans,v[22600],a[22600];
vector <int> c[22600];
void castel(int i)
{
    int j;
    if (v[i])
        return;
    v[i]=1;
    ans++;
    if (i%m!=0 && v[a[i-1]])
        castel(i-1);
    if (i%m!=m-1 && v[a[i+1]])
        castel(i+1);
    if (i+m<n*m && v[a[i+m]])
        castel(i+m);
    if (i-m>=0 && v[a[i-m]])
        castel(i-m);
    for (j=0; j<c[i].size(); j++)
        if ((c[i][j]%m!=0 && v[c[i][j]-1]) || (c[i][j]%m!=m-1 && v[c[i][j]+1]) || (c[i][j]+m<n*m && v[c[i][j]+m]) || (c[i][j]-m>=0 && v[c[i][j]-m]))
            castel(c[i][j]);
}
int main()
{
    int i;
    cin>>n>>m>>k;
    k--;
    for (i=0; i<n*m; i++)
    {
        cin>>a[i];
        a[i]--;
        c[a[i]].push_back(i);
    }
    castel(k);
    cout<<ans;
    return 0;
}