Cod sursa(job #488783)

Utilizator ProtomanAndrei Purice Protoman Data 29 septembrie 2010 21:27:11
Problema Gather Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.72 kb
#include <algorithm>
#include <stdio.h>
#include <vector>
#include <queue>

#define MAX 1500
#define INF 0x3f3f3f3f
#define pb push_back
#define mp make_pair
#define f first
#define s second

using namespace std;

int k, n, m, dimHeap;
int vctDist[MAX], det[MAX];
vector <pair <int, int> > vctDrum[MAX];
vector <pair <pair <int, int>, int> > vctMuchie[16];
int sol[65000][17];
int dist[32][32][32];
priority_queue <pair <int, int> > minHeap;

inline void dijkstra(int source)
{
	dimHeap = n;
	for (int i = 1; i <= n; i++)
		vctDist[i] = INF;
	vctDist[source] = 0;
	minHeap.push(mp(0, source));

	for (; minHeap.size(); )
		if (minHeap.top().f == vctDist[minHeap.top().s])
		{
			int nod = minHeap.top().s;
			minHeap.pop();

			if (vctDist[nod] == INF)
				return;

			for (int i = 0; i < vctDrum[nod].size(); i++)
				if (vctDist[nod] + vctDrum[nod][i].s < vctDist[vctDrum[nod][i].f])
				{
					vctDist[vctDrum[nod][i].f] = vctDist[nod] + vctDrum[nod][i].s;

					minHeap.push(mp(-vctDist[vctDrum[nod][i].f], vctDrum[nod][i].f));
				}
		}
		else minHeap.pop();
}

inline void addMuchii(int nrDet)
{
	for (int i = 0; i < vctMuchie[nrDet].size(); i++)
	{
		int n1 = vctMuchie[nrDet][i].f.f, n2 = vctMuchie[nrDet][i].f.s, c = vctMuchie[nrDet][i].s;
		
		vctDrum[n1].pb(mp(n2, c));
		vctDrum[n2].pb(mp(n1, c));
	}
}	

int main()
{
	freopen("gather.in", "r", stdin);
	freopen("gather.out", "w", stdout);

	scanf("%d %d %d", &k, &n, &m);

	for (int i = 0; i < k; i++)
		scanf("%d\n", &det[i]);

	for (int i = 1; i <= m; i++)
	{
		int n1, n2, c, d;
		scanf("%d %d %d %d", &n1, &n2, &c, &d);

		vctMuchie[d].pb(mp(mp(n1, n2), c));
	}

	for (int nrDet = k; nrDet; nrDet--)
	{
		addMuchii(nrDet);

		for (int i = 0; i < k; i++)
		{
			dijkstra(det[i]);

			for (int j = 0; j < k; j++)
				dist[nrDet][i][j] = vctDist[det[j]];
		}
	}

	addMuchii(0);
	dijkstra(1);
	for (int i = 1; i < (1 << k); i++)
		for (int j = 0; j < k; j++)
			sol[i][j] = INF;

	for (int i = 0; i < k; i++)
		sol[1 << i][i] = vctDist[det[i]];

	for (int conf = 1; conf < (1 << k); conf++)
	{
		for (int ult = 0; ult < k; ult++)
			if (conf & (1 << ult) && sol[conf][ult] != INF)
			{
				int nrDet = 0;
				for (int j = conf; j; j &= (j - 1))
					nrDet++;

				for (int nou = 0; nou < k; nou++)
					if (!(conf & (1 << nou)) && dist[nrDet][ult][nou] != INF)
						sol[conf | (1 << nou)][nou] = min(sol[conf | (1 << nou)][nou], sol[conf][ult] + dist[nrDet][ult][nou] * (nrDet + 1));
			}
	}

	int minGs = INF;
	for (int i = 0; i < k; i++)
		minGs = min(minGs, sol[(1 << k) - 1][i]);

	if (minGs == INF)
		for (; ; );

	printf("%d\n", minGs);

	fclose(stdin);
	fclose(stdout);
	return 0;
}