Cod sursa(job #2347062)

Utilizator anamariatoaderAna Toader anamariatoader Data 18 februarie 2019 12:59:29
Problema Castel Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.77 kb
#include <fstream>
#include <vector>

using namespace std;
ifstream fin("castel.in");
ofstream fout("castel.out");

int n,m,k,i,j,v[155][155],viz[155][155],ct;
int dx[]={0,-1,0,1,0},dy[]={0,0,1,0,-1};
bool fr[22505];
vector <int> key[22505];
struct numar{
    int l,c;
}q[22505];

bool valid(int x, int y){
    return (x>=1 && x<=n && y>=1 && y<=m);
}

int linie(int k){
    if(k%m==0)
        return k/m;
    return k/m+1;
}

int coloana(int k){
    if(k%m==0)
        return m;
    return k%m;
}

void coada(int x, int y){
    int p,u,r,l,c,ln,cn;
    ct=1;
    p=u=1;
    q[p].l=x;
    q[p].c=y;
    viz[x][y]=2;
    while(p<=u){
        l=q[p].l;
        c=q[p].c;
        for(r=1;r<=4;r++){
            ln=l+dx[r];
            cn=c+dy[r];
            if(valid(ln,cn) && viz[ln][cn]==0)
                if(fr[v[ln][cn]]){
                     ct++;
                     q[++u].l=ln;
                     q[u].c=cn;
                     viz[ln][cn]=2;
                     fr[(ln-1)*m+cn]=1;
                     int nr=key[(ln-1)*m+cn].size();
                     for(int j=0;j<nr;j++){
                        int e=key[(ln-1)*m+cn][j];
                        int a=linie((ln-1)*m+cn);
                        int b=coloana((ln-1)*m+cn);
                        viz[a][b]=2;
                        ct++;
                     }
                }
                else{
                    key[v[ln][cn]].push_back((ln-1)*m+cn);
                    viz[ln][cn]=1;
                }
        }
        p++;
    }

}

int main()
{
    fin>>n>>m>>k;
    fr[k]=1;
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++)
            fin>>v[i][j];
    int l=linie(k);
    int c=coloana(k);
    coada(l,c);
    fout<<ct;
    return 0;
}