Cod sursa(job #887206)

Utilizator anca1243Popescu Anca anca1243 Data 23 februarie 2013 16:49:27
Problema Ubuntzei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.47 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream in("ubuntzei.in");
ofstream out ("ubuntzei.out");
struct cost
{
    int vf,c;
};
const int N=2002,INF=200000000;
vector <cost> a[N];
queue <int> q;
int n,m,k,o[N],d[N][N];
bool inq[N];
void bfs(int p)
{
    int y,x;
    q.push(p);
    while(!q.empty())
    {
        x=q.front();
        q.pop();
        inq[x]=false;
        for(size_t i=0;i<a[x].size();i++)
        {
            y=a[x][i].vf;
            if(d[p][x]+a[x][i].c<d[p][y])
            {
                d[p][y]=d[p][x]+a[x][i].c;
                if(!inq[y])
                {
                    q.push(y);
                    inq[y]=true;
                }
            }
        }
    }
}
void citire()
{
    int x,y,c;
    cost aux;
    in>>n>>m>>k;
    for(int i=1;i<=n;i++)
        in>>o[i];
    for(int i=1;i<=m;i++)
    {
        in>>x>>y>>c;
        aux.vf=y;
        aux.c=c;
        a[x].push_back(aux);
        aux.vf=x;
        a[y].push_back(aux);
    }
}
void init()
{
    for(int i=0;i<=N;i++)
        for(int j=0;j<=N;j++)
            d[i][j]=INF;
}
void init1()
{
    for(int i=0;i<=N;i++)
        inq[i]=0;
}
int main()
{
    citire();
    init();
    o[0]=1;
    for(int i=0;i<=k;i++)
    {
        init1();
        bfs(o[i]);
    }
    int s=0;
    for(int i=0;i<k;i++)
        s+=d[i][i+1];
    s+=d[k][n];
    out<<s;
    return 0;
}