Cod sursa(job #2562958)

Utilizator dogaru_roxanaDogaru Roxana dogaru_roxana Data 29 februarie 2020 20:38:08
Problema Ubuntzei Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.7 kb
#include <fstream>
#include <vector>
#include <queue>
#define pinf 1000000007
#define f first
#define s second
using namespace std;

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

int n, m, k, pr[20], x, y, l, a[20][2005], d[2005], viz[2005], pd[32800][20], nr, aux, p, rez;
vector <pair <int, int> > v[2005];
priority_queue < pair <int, int>, vector < pair <int, int> >, greater < pair <int, int> > > pq;
pair <int, int> mini;

void dijkstra (int start)
{
    int i;

    for (i=1; i<=n; i++)
    {
        d[i]=pinf;
        viz[i]=0;
    }

    viz[start]=1;
    d[start]=0;

    for (auto it: v[start])
    {
        d[it.s]=it.f;
        pq.push(make_pair(d[it.s], it.s));
    }

    while (pq.empty()==0)
    {
        mini=pq.top();
        pq.pop();

        if (viz[mini.s]==0)
        {
            viz[mini.s]=1;

            for (auto it: v[mini.s])
            {
                if (viz[it.s]==0)
                {
                    if (d[mini.s]+it.f<d[it.s])
                    {
                        d[it.s]=d[mini.s]+it.f;
                        pq.push(make_pair(d[it.s], it.s));
                    }
                }
            }
        }
    }
}

int main()
{
    int i, j, r;

    fin>>n>>m>>k;
    for (i=0; i<k; i++)
    {
        fin>>pr[i];
    }

    for (i=1; i<=m; i++)
    {
        fin>>x>>y>>l;
        v[x].push_back(make_pair(l, y));
        v[y].push_back(make_pair(l, x));
    }

    if (k==0)
    {
        dijkstra(1);
        fout<<d[n];
    }
    else
    {
        for (i=0; i<k; i++)
        {
            dijkstra(pr[i]);
            for (j=1; j<=n; j++)
            {
                a[i][j]=d[j];
            }
        }

        nr=(1<<k)-1;

        for (i=0; i<=nr; i++)
        {
            for (j=0; j<k; j++)
                pd[i][j]=pinf;
        }

        for (i=0; i<k; i++)
        {
            pd[1<<i][i]=a[i][1];
        }

        for (i=1; i<=nr; i++)
        {
            for (j=0; j<k; j++)
            {
                p=1<<j;

                if ((p&i)>0)
                {
                    aux=i-p;

                    for (r=0; r<k; r++)
                    {
                        if (pd[aux][r]+a[r][pr[j]]<pd[i][j])
                        {
                            pd[i][j]=pd[aux][r]+a[r][pr[j]];
                        }
                    }
                }
            }
        }

        rez=pinf;
        for (j=0; j<k; j++)
        {
            if (pd[nr][j]+a[j][n]<rez)
                rez=pd[nr][j]+a[j][n];
        }

        fout<<rez;
    }

    fin.close();
    fout.close();
    return 0;
}