Cod sursa(job #2356696)

Utilizator TudorFinaruTudor Cristian Finaru TudorFinaru Data 26 februarie 2019 20:58:50
Problema Ubuntzei Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.81 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");

int n,m,k,p[20],dist[2003],s[2003],must[2003];
int d[2002][2002];
const int INF=0x3f3f3f3f;
int ans=999999999;

set<pair<int, int> >h;
vector<pair<int, int> >v[2003];

void blabla()
{ //g<<p[s[1]]<<' ';
    //g<<d[1][2]<<' '<<d[2][n]<<' ';
    int D,i;
    D=d[1][s[1]];
    for(i=2;i<=k;i++) D+=d[s[i-1]][s[i]];
    D+=d[s[k]][n];
    ans=min(ans,D);
}

bool valid(int pas)
{
    if(s[pas]==s[pas-1])return false;
    return true;
}

void backtr(int pas)
{
    int i;
    for(i=1;i<=k;i++)
    {
        s[pas]=must[i];
        if(valid(pas))
        {
            if(pas==k) blabla();
            else backtr(pas+1);
        }
    }
}

int main()
{
    int i,x,y,c,j;
    f>>n>>m;
    f>>k;
    for(i=1;i<=k;i++) f>>must[i];
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>c;
        v[x].push_back(make_pair(y,c));
        v[y].push_back(make_pair(x,c));
    }
    //backtr(1);
    if(k==0)
    {
        vector<pair<int, int> >::iterator it;
        int nod,to,cost;
        ///dijkstra
        fill(dist+1,dist+n+1,INF);
        dist[1]=0;
        h.insert(make_pair(0,1));
        while(!h.empty())
        {
            nod=h.begin()->second;
            h.erase(h.begin());
            for(it=v[nod].begin();it!=v[nod].end();++it)
            {
                to=it->first; cost=it->second;
                if(dist[to]>dist[nod]+cost)
                {
                    if(dist[to]!=INF) h.erase(h.find(make_pair(dist[to],to)));
                    dist[to]=dist[nod]+cost;
                    h.insert(make_pair(dist[to],to));
                }
            }
        }
        g<<dist[n]<<'\n';
    }
    else
    {
        vector<pair<int, int> >::iterator it;
        int nod,to,cost;
        ///dijkstra
        for(i=1;i<=n;i++)
        {
            fill(dist+1,dist+n+1,INF);
            dist[i]=0;
            h.insert(make_pair(0,i));
            while(!h.empty())
            {
                nod=h.begin()->second;
                h.erase(h.begin());
                for(it=v[nod].begin();it!=v[nod].end();++it)
                {
                    to=it->first; cost=it->second;
                    if(dist[to]>dist[nod]+cost)
                    {
                        if(dist[to]!=INF) h.erase(h.find(make_pair(dist[to],to)));
                        dist[to]=dist[nod]+cost;
                        h.insert(make_pair(dist[to],to));
                    }
                }
            }
            for(j=1;j<=n;j++)
                if(j!=i)
                {
                    d[i][j]=d[j][i]=dist[j];
                }
        }
        backtr(1);
        g<<ans<<'\n';
    }
    f.close();
    g.close();
    return 0;
}