Cod sursa(job #562148)

Utilizator eudanipEugenie Daniel Posdarascu eudanip Data 22 martie 2011 14:30:57
Problema Gather Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.7 kb
#include<stdio.h>
#include<vector>
#include<set>
using namespace std;

#define INF 2147483647
#define minim(a,b) (a<b ? a : b)
#define pb push_back
#define NMAX 805
#define ll long long

int n,m,k,cell[105];
int dist[NMAX];
int d[18][18][NMAX];
int sol,bst[35006][17];
int nrbit[35006];
vector<int> v[NMAX],cost[NMAX],perm[NMAX];

class cmp
{
public:
    bool operator()(const int& a,const int& b)
    {
        return dist[a]<dist[b] ||
                (dist[a]==dist[b] && a<b);
    }
};

set<int,cmp> myset;

inline void dijkstra(int ind,int nrd)
{
    int i,j,p,lim,start=cell[ind];
    
    myset.clear();
    for(i=1;i<=n;i++)
        dist[i]=INF;
    dist[start]=0;
    for(i=1;i<=n;i++)
        myset.insert(i);
    for(i=1;i<=n;i++)
    {
        if(myset.size()==0)
            break;
        p=*myset.begin();
        myset.erase(myset.begin());
        lim=v[p].size();
        for(j=0;j<lim;j++)
            if(perm[p][j]>=nrd
            && dist[v[p][j]]>dist[p]+cost[p][j])
            {
                int vec=v[p][j];
                myset.erase(vec);
                dist[vec]=dist[p]+cost[p][j];
                myset.insert(vec);
            }
    }
    for(i=1;i<=n;i++)
        if(dist[i]!=INF)
            d[ind][nrd][i]=dist[i]*(nrd+1);
        else
            d[ind][nrd][i]=dist[i];
}

int main ()
{
    int i,j,t;
    int a,b,c,d2,cf;
    
    freopen("gather.in","r",stdin);
    freopen("gather.out","w",stdout);
    scanf("%d%d%d",&k,&n,&m);
    for(i=1;i<=k;i++)
        scanf("%d",&cell[i]);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d%d",&a,&b,&c,&d2);
        v[a].pb(b);
        cost[a].pb(c);
        perm[a].pb(d2);
        
        v[b].pb(a);
        cost[b].pb(c);
        perm[b].pb(d2);
    }
    cell[0]=1;
    dijkstra(0,0);
    
    for(i=1;i<=k;i++)
        for(j=1;j<k;j++)
            dijkstra(i,j);

    for(i=1;i<(1<<k);i++)
        for(j=1;j<=k;j++)
            bst[i][j]=INF;
            
    for(i=1;i<=k;i++)
        bst[1<<(i-1)][i]=d[0][0][cell[i]];

    for(i=1;i<(1<<k);i++)
    {
        nrbit[i]=nrbit[i/2]+(i&1);
        for(j=1;j<=k;j++)
            if(i&(1<<(j-1)))
            {
                if(bst[i][j]==INF)
                    continue;
                for(t=1;t<=k;t++)
                    if(!(i&(1<<(t-1))))
                    {
                        cf=i+(1<<(t-1));
                        bst[cf][t]=minim(bst[cf][t],
                            bst[i][j]+d[j][nrbit[i]][cell[t]]);
                    }
            }
    }
    sol=INF;
    for(i=1;i<=k;i++)
        sol=minim(sol,bst[(1<<k)-1][i]);
    printf("%d\n",sol);
    return 0;
}