Cod sursa(job #1089453)

Utilizator cat_red20Vasile Ioana cat_red20 Data 21 ianuarie 2014 18:32:07
Problema Team Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.45 kb
#include<fstream>
#include<queue>
#include<cstring>
#define INF 1<<30
using namespace std;
int p,n,a[51][51][51],m,dest[51],dr[51][501],viz[501],dist[501],q[5000];
vector<pair<int,int> > v[501];

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

void citire()
{
    int x,y,c;
    fin>>p>>n>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        v[x].push_back(make_pair(y,c));
        v[y].push_back(make_pair(x,c));
    }
    for(int i=1;i<=p;i++)
    {
        fin>>dest[i];
    }
}

void init()
{
    for(int i=1;i<=n;i++)
    dist[i]=INF;
}

void bellmanFord(int nod,int l)
{
    int x,p,u;
    p=u=1;
    init();
    dist[nod]=0;
    q[p]=nod;
    viz[nod]=1;
    while(p<=u)
    {
        x=q[p];
        p++;

        for(int i=0;i<v[x].size();i++)
        {
            if(dist[x]+v[x][i].second<dist[v[x][i].first])
            {
                dist[v[x][i].first]=dist[x]+v[x][i].second;
                if(viz[v[x][i].first]==0)
                {
                    q[++u]=v[x][i].first;
                    viz[v[x][i].first]=1;
                }
            }
        }
        viz[x]=0;
    }
    memcpy(dr[l],dist,sizeof(dist));
}

void din()
{
    int j,min1,min2;
    for(int i=1;i<=p;i++)
        for(int j=1;j<=p;j++)
            for(int k=1;k<=p;k++)
                a[i][j][k]=INF;

    for(int i=1;i<=p;i++)
        a[i][i][i]=0;

    for(int l=1;l<=p-1;l++)
    {
        for(int i=1;i<=p-l;i++)
        {
            j=i+l;
            for(int k=i;k<=j;k++)
            {
                min1=INF;
                min2=INF;
                for(int t=i;t<k;t++)
                {
                    if(a[i][k-1][t]+dr[t][dest[k]]<min1)
                    min1=a[i][k-1][t]+dr[t][dest[k]];
                }
                for(int t=k+1;t<=j;t++)
                {
                    if(a[k+1][j][t]+dr[t][dest[k]]<min2)
                    min2=a[k+1][j][t]+dr[t][dest[k]];
                }
                a[i][j][k]=0;
                if(min1!=INF)
                a[i][j][k]=min1;
                if(min2!=INF)
                a[i][j][k]+=min2;
            }
        }
    }
}

int main()
{
    citire();
    init();
    for(int i=1;i<=p;i++)
        bellmanFord(dest[i],i);
    din();
    int mini=INF;
    for(int i=1;i<=p;i++)
    {
        if(a[1][p][i]+dr[i][1]<mini)
        mini=a[1][p][i]+dr[i][1];
    }
    fout<<mini;
    return 0;
}