Cod sursa(job #2563279)

Utilizator hhhhhhhAndrei Boaca hhhhhhh Data 1 martie 2020 10:13:59
Problema Castel Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.09 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("castel.in");
ofstream fout("castel.out");
int n,m,v[201][201],k,ans;
set<int> sol[151][151];
set<int> reuniune;
queue< pair<int,int> > coada;
bool use[201][201];
int dirx[5]={-1,1,0,0};
int diry[5]={0,0,-1,1};
int main()
{
    fin>>n>>m>>k;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            fin>>v[i][j];
            sol[i][j].insert(m*(i-1)+j);
        }
    int x=k/n+1;
    int y=k%n;
    if(y==0)
    {
        y=m;
        x--;
    }
    coada.push({x,y});
    reuniune.insert(k);
    use[x][y]=1;
    while(!coada.empty())
    {
        set<int> s=sol[coada.front().first][coada.front().second];
        x=coada.front().first;
        y=coada.front().second;
        use[x][y]=1;
        bool nw[5];
        memset(nw,0,sizeof(nw));
        for(int i=0;i<4;i++)
        {
            x+=dirx[i];
            y+=diry[i];
            if(x<=n&&x>0&&y<=m&&y>0)
                if(reuniune.find(v[x][y])!=reuniune.end())
            {
                set<int> ss=sol[x][y];
                for(set<int>::iterator it=ss.begin(); it!=ss.end(); it++)
                {
                    reuniune.insert(*it);
                }
            }
            x-=dirx[i];
            y-=diry[i];
        }
        for(int i=0;i<4;i++)
        {
            x+=dirx[i];
            y+=diry[i];
            if(x<=n&&x>0&&y<=m&&y>0&&s.find(v[x][y])!=s.end())
            {
                if(sol[x][y]!=reuniune)
                {
                    sol[x][y]=reuniune;
                    coada.push({x,y});
                }
            }
            x-=dirx[i];
            y-=diry[i];
        }
        coada.pop();
    }

    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
           ans+=use[i][j];
    fout<<ans;
    /*for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            if(use[i][j])
    {
        set<int> sss=sol[i][j];
        for(auto it=sss.begin();it!=sss.end();it++)
            fout<<*it<<" ";
        fout<<'\n';
    }*/
    return 0;
}