Cod sursa(job #1855906)

Utilizator miruna999Morarasu Miruna miruna999 Data 24 ianuarie 2017 12:47:38
Problema Ubuntzei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.28 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");
struct nod{
    int nr,val;
    nod *urm;
};
nod *a[2001],*p,*prim,*ultim;
int n,m,k,d[2001],prieten[2001],L[2001][2001],mini=1000000000,x[2001],viz[2001];

void citire()
{
    f>>n>>m>>k;
    for(int i=1;i<=k;i++)
        f>>prieten[k];
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        f>>x>>y>>z;
        p=new nod;
        p->nr=y;
        p->val=z;
        p->urm=0;
        p->urm=a[x];
        a[x]=p;
    }
}

void dijkstra(int v1)
{
    for(int i=2;i<=n;i++)
        d[i]=99999999;
    d[v1]=0;
    prim=new nod;
    prim->nr=v1;
    prim->urm=NULL;
    ultim=prim;
    while(prim)
    {
        int nod1=prim->nr;
        nod *q=a[nod1];
        while(q)
        {
            if(d[q->nr]>d[nod1]+q->val)
            {
                d[q->nr]=d[nod1]+q->val;
                p=new nod;
                p->nr=q->nr;
                p->urm=NULL;
                ultim->urm=p;
                ultim=p;
            }
            q=q->urm;
        }
        p=prim;
        prim=prim->urm;
        delete p;
    }
}

int sum()
{
    int s=0;
    for(int i=1;i<k;i++)
        s+=L[x[i]][x[i+1]];
    return s;
}

void afis()
{
    for(int i=1;i<=k;i++)
        cout<<x[i]<<" ";
    cout<<endl;
}

void backt(int j)
{
    for(int i=2;i<k;i++)
        if(!viz[i])
        {
            viz[i]=1;
            x[j]=prieten[i-1];

            if(j==k-1)
            {
                int s=sum();
                if(s<mini)
                    mini=s;
                //afis();
            }
            else
                backt(k+1);
            viz[i]=0;
        }
}

int main()
{
    citire();
    dijkstra(1);
    for(int j=1;j<=n;j++)
        if(d[j]!=99999999)
            L[1][j]=d[j];
    for(int i=1;i<=k;i++)
    {
        dijkstra(prieten[i]);
        for(int j=1;j<=n;j++)
            if(d[j]!=99999999)
                L[prieten[i]][j]=d[j];
    }
    /*
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
            cout<<L[i][j]<<" ";
        cout<<endl;
    }*/
    x[1]=1;viz[1]=1;
    k+=2;
    x[k]=n;viz[n]=1;
    backt(2);
    g<<mini;
    return 0;
}