Cod sursa(job #2124150)

Utilizator tziplea_stefanTiplea Stefan tziplea_stefan Data 6 februarie 2018 22:45:10
Problema Castel Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.84 kb
#include <fstream>
#include <vector>
#include <queue>
#define VAL 22505

using namespace std;

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

int N, M, K, i, j, nr, ANS;
int ch[VAL], l, c, nod;
bool deb[VAL], gas, dp[VAL];
vector <int> vec[VAL], S;
queue <int> Q;

int Trans(int lin, int col)
{
    return (lin-1)*M+col;
}

void BFS()
{
    while (Q.empty()==false)
    {
        nod=Q.front();
        Q.pop();
        for (j=0; j<vec[nod].size(); j++)
        {
            if (deb[ch[vec[nod][j]]]==true && dp[vec[nod][j]]==false)
            {
                deb[vec[nod][j]]=true;
                dp[vec[nod][j]]=true;
                S.push_back(vec[nod][j]);
                Q.push(vec[nod][j]);
            }
        }
    }
}

int main()
{
    fin >> N >> M >> K;
    for (i=1; i<=N; i++)
    {
        for (j=1; j<=M; j++)
        {
            fin >> ch[++nr];
            l=i-1;
            c=j;
            if (l>=1)
              vec[nr].push_back(Trans(l, c));
            l=i+1;
            if (l<=N)
              vec[nr].push_back(Trans(l, c));
            c=j-1;
            l=i;
            if (c>=1)
              vec[nr].push_back(Trans(l, c));
            c=j+1;
            if (c<=M)
              vec[nr].push_back(Trans(l, c));
        }
    }
    S.push_back(K);
    dp[K]=true;
    deb[ch[K]]=true;
    while (1)
    {
        gas=false;
        for (i=0; i<S.size(); i++)
        {
            nr=0;
            for (j=0; j<vec[S[i]].size(); j++)
            {
                if (dp[vec[S[i]][j]]==false && deb[ch[vec[S[i]][j]]]==true)
                {
                    gas=true;
                    nr++;
                }
            }
            if (nr>0)
              Q.push(S[i]);
        }
        if (gas==false)
          break;
        BFS();
    }
    fout << S.size() << '\n';
    fin.close();
    fout.close();
    return 0;
}