Cod sursa(job #1678342)

Utilizator alex99Chelba Alexandru alex99 Data 7 aprilie 2016 11:02:28
Problema Castel Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.6 kb
#include <fstream>
using namespace std;
ifstream f("castel.in");
ofstream g("castel.out");
int a[160][160],b[160][160],c[160][160],nr1,n,m,nr,st,sti,stj;
int dl[]={0,1,0,-1},dc[]={1,0,-1,0}, pozi[160],pozj[160];
struct pct
{
    int x,y;
};
pct c1[160*160],w,viz[160][160];
void acces(int x,int y);
void backt(int sti, int stj);
void lee(int i, int j);
int main()
{
    f>>n>>m>>st; nr=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            f>>a[i][j];
            nr++; c[i][j]=nr;
            if(nr==st) {sti=i; stj=j;}
        }
    }
    acces(a[sti][stj],c[sti][stj]); ///viz[sti][stj]=1;
    backt(sti,stj);
    b[sti][stj]=-1;
    nr1=1;
    lee(sti,stj);
    int nr2=1;
    for(int i=1;i<=n;i++)
    {
        ///for(int j=1;j<=m;j++)
        ///    g<<b[i][j]<<" ";
        ///g<<'\n';
        for(int j=1;j<=m;j++)
            if(b[i][j]>1)
                {
                    ///g<<viz[i][j].x<<" "<<viz[i][j].y<<'\n';
                    if(b[viz[i][j].x][viz[i][j].y]>1||b[viz[i][j].x][viz[i][j].y]==-1)
                    {
                        nr2++;
                    }
                }
    }
    g<<nr2;
    /**for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            g<<b[i][j]<<" ";
        g<<'\n';
    }**/
    return 0;
}
void acces(int x,int y)
{
    int z=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        if(b[i][j]==0)
        if(a[i][j]==x||c[i][j]==x||a[i][j]==y||c[i][j]==y) {b[i][j]=1; if(!viz[i][j].x) {z++; pozi[z]=i; pozj[z]=j;}}
    for(int i=1;i<=z;i++)
    {
        backt(pozi[i],pozj[i]);
    }
}
void backt(int sti,int stj)
{
    ///g<<sti<<" "<<stj<<'\n';
    /**for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            g<<b[i][j]<<" ";
        g<<'\n';
    }
    g<<'\n';**/
    for(int k=0;k<4;k++)
    {
        if(b[sti+dl[k]][stj+dc[k]]==1)
        {
            viz[sti+dl[k]][stj+dc[k]].x=sti; viz[sti+dl[k]][stj+dc[k]].y=stj;
            acces(a[sti+dl[k]][stj+dc[k]],c[sti+dl[k]][stj+dc[k]]);
            //backt(sti+dl[k],stj+dc[k]);
        }
    }
}

void lee(int i,int j)
{
    int prim=1; int ultim=1;
    c1[prim].x=i; c1[prim].y=j;
    while(prim<=ultim)
    {
        w=c1[prim];
        for(int k=0;k<4;k++)
            if(b[w.x+dl[k]][w.y+dc[k]]==1)
            {
                ultim++;
                nr1++;
                b[w.x+dl[k]][w.y+dc[k]]=nr1;
                c1[ultim].x=w.x+dl[k];
                c1[ultim].y=w.y+dc[k];
            }
        ///nr1=b[c1[prim].x][c1[prim].y];
        prim++;
    }
}