Cod sursa(job #594757)

Utilizator unknownliviuMaria Liviu Valentin unknownliviu Data 9 iunie 2011 14:51:21
Problema Ubuntzei Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.21 kb
#include<queue>
#include<vector>
#include<fstream>
using namespace std;
ifstream in("ubuntzei.in");
ofstream out("ubuntzei.out");
const int N=2010;
const int INF=0x3f3f3f;

int n,m,k,v[N],d[N],t[N],cost[18][18];
vector<int>a[N];
vector<int>c[N];


bool viz[N];
queue<int>coada;
int cc[131072][18];


inline int min(int eee,int www)
{
	return eee<www?eee:www;
}


void read()
{
	int x,y,z;
	
	in>>n>>m>>k;
	for(int i=1;i<=k;i++)
		in>>v[i];
	
	for(int i=1;i<=m;i++)
	{
		in>>x>>y>>z;
		a[x].push_back(y);
		a[y].push_back(x);
		c[x].push_back(z);
		c[y].push_back(z);
	}
}

void init()
{
	for(int i=0;i<=n;i++)
		d[i]=INF,viz[i]=0,t[i]=0;
}

void bford(int param)
{
	init();
	int r,x,costt;
	
	coada.push(param);
	d[param]=0;
	viz[param]=1;
	bool ok=1;
		
	while(ok && !coada.empty())
	{
		r=coada.front();
		coada.pop();
		viz[r]=0;
		
		for(size_t i=0;i<a[r].size();i++)
		{
			x=a[r][i];
			costt=c[r][i];
			if(d[r]+costt<d[x])
			{
				d[x]=d[r]+costt;
				if(!viz[x])
				{
					viz[x]=true;
					t[x]++;
					if(t[x]>n-1)
						ok=0;
					coada.push(x);
				}
			}
		}
	}
}

void solve()
{
	int o=k+2,x;
	for(int i=0;i<=o;i++)
		for(int j=0;j<=o;j++)
			if(i!=j)
				cost[i][j]=INF;
		
	if(k==0)
	{
		bford(1);
		out<<d[n];
		exit(0);
	}
	for(int i=1;i<=k;i++)
	{
		x=v[i];
		bford(x);
		
		cost[i][0]=cost[0][i]=d[1];
		cost[i][o-1]=cost[o-1][i]=d[n];
		
		for(int j=1;j<=k;j++)
			cost[i][j]=cost[j][i]=d[ v[j] ];
	}
	
	
	cost[0][0]=cost[o][o]=0;
	
	/*
	for(int i=0;i<o;i++)
	{
		for(int j=0;j<o;j++)
			out<<cost[i][j]<<"\t";
		out<<"\n";
	}
	*/
	for(int i=0;i<(1<<o);i++)
		for(int j=0;j<o;j++)
			cc[i][j]=INF;
	
	cc[1][0]=0;
	
	for(int i=1;i<(1<<o);i++)
		for(int j=0;j<o;j++)
				if(i &(1<<j) )
				{
					for(int k=0;k<=o;k++)
					{
						if(!cost[k][j] || cost[k][j]==INF)
							continue;
						if(i & (1<<k) )
							cc[i][j]=min(cc[i][j], cc[i ^ (1<<j) ][k] + cost[k][j] );
					}
				}
	/*
	for(int i=0; i < (1<<o) ; i++)
	{
		out<<i<<":\t";
		for(int j=0;j<=o;j++)
			out<<cc[i][j]<<"\t";
		out<<"\n";
	}
	*/				
	out<<cc[(1<<o)-1][o-1];
}

int main()
{
	read();
	solve();
	return 0;
}