Cod sursa(job #2347083)

Utilizator anamariatoaderAna Toader anamariatoader Data 18 februarie 2019 13:28:17
Problema Castel Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.9 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);
}

void xy(int k,int &l,int &c){
    c=k%m;
    l=k/m;
    if(c==0)
        c=m;
    else
        l++;
}

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;
        int nr=(l-1)*m+c;
        for(int j=0;j<key[nr].size();j++){
                        int e=key[nr][j];
                        int a,b;
                        xy(e,a,b);
                        if(viz[a][b]!=2){
                            q[++u].l=a;
                            q[u].c=b;
                            viz[a][b]=2;
                            fr[e]=1;
                            ct++;
                        }
        }
        for(r=1;r<=4;r++){
            ln=l+dx[r];
            cn=c+dy[r];
            if(valid(ln,cn) && viz[ln][cn]!=2){
                if(fr[v[ln][cn]]){
                     ct++;
                     q[++u].l=ln;
                     q[u].c=cn;
                     viz[ln][cn]=2;
                     fr[(ln-1)*m+cn]=1;
                     }
                else
                   if (viz[ln][cn]==0)
                    {
                    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,c;
    xy(k,l,c);
    coada(l,c);
    fout<<ct;
    return 0;
}