Cod sursa(job #674080)

Utilizator blustudioPaul Herman blustudio Data 5 februarie 2012 15:11:05
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.48 kb
#include <fstream>
#include <vector>
#include <iostream>
using namespace std;

#define oo 0x3f3f3f

int n, m, k;
int ns[17];
vector<int> graf[2001];
vector<int> cost[2001];
int dist[17][2001];
int dinamica[17][17];
int distanta;

inline void citire()
{
	int a, b, c;
	ifstream fin("ubuntzei.in");
	fin >> n >> m >> k;
	for (int i = 1; i <= k; i++)
		fin >> ns[i];
	for (int i = 0; i < m; i++)
	{
		fin >> a >> b >> c;
		graf[a].push_back(b);
		graf[b].push_back(a);
		cost[a].push_back(c);
		cost[b].push_back(c);
	}
	fin.close();
}
inline void bellman_ford(int p, int s)
{
	bool relaxare = true;
	for (int i = 1; i <= n; i++)
		dist[p][i] = oo;
	dist[p][s] = 0;
	for (int i = 1; (i <= n) && (relaxare == true); i++)
	{
		relaxare = false;
		for (int j = 1; j <= n; j++)
		{
			for (int k = 0; k < graf[j].size(); k++)
			{
				if (dist[p][graf[j][k]] > (dist[p][j] + cost[j][k]))
				{
					dist[p][graf[j][k]] = dist[p][j] + cost[j][k];
					relaxare = true;
				}
			}
		}
	}
}
inline void rezolvare()
{
	k++;
	ns[0] = 1;
	ns[k] = n;
	for (int i = 0; i <= k; i++)
		bellman_ford(i, ns[i]);
	for (int i = 0; i <= k; i++)
		for (int j = 0; j <= k; j++)
			dinamica[i][j] = dist[i][ns[j]];
	for (int i = 1; i <= k; i++)
		for (int j = 1; j <= k; j++)
			dinamica[i][j] += min(dinamica[i - 1][j], dinamica[i][j - 1]);
	distanta = dinamica[k][k];
}
inline void scriere()
{
	ofstream fout("ubuntzei.out");
	fout << distanta << '\n';
	fout.close();
}
int main()
{
	citire();
	rezolvare();
	scriere();
	return 0;
}