Cod sursa(job #1452395)

Utilizator alexxxx_iulianalex iulian alexxxx_iulian Data 20 iunie 2015 18:11:06
Problema Ubuntzei Scor 25
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.44 kb
#include<stdio.h>
#include<stdlib.h>



typedef struct nod{
	int val;
	int cost;
	nod *next;
}nod;


#define INT_MAX 100000

int * dijkstra(nod **a,int n,int k)
{
	int *tata=(int *)malloc((n+2)*sizeof(int)),
		*distante=(int *)malloc((n+2)*sizeof(int)),
		*viz=(int *)malloc((n+2)*sizeof(int));

	for(int i=1;i<=n;i++)
	{
		distante[i]=INT_MAX;
		viz[i]=0;
		tata[i]=0;
	}
	nod*p=a[k];
	while (p)
	{
		distante[p->val]=p->cost;
		tata[p->val]=k;
		p=p->next;
	}
	viz[k]=1;
	int z,min;
	for(int i=1;i<n;i++)
	{
		min=INT_MAX;
		for(int j=1;j<=n;j++)
		{
			if(distante[j]<min && viz[j]==0)
			{
				min=distante[j];
				z=j;
			}
		}
		viz[z]=1;
		for(p=a[z];p;p=p->next)
		{
			if(viz[p->val]==0 && distante[z]+p->cost<distante[p->val])
			{
				distante[p->val]=distante[z]+p->cost;
				tata[p->val]=z;
			}
		}
	}
	return distante;
}

bool bun(int *sol,int k)
{
	if(k==0)
		return true;
	for(int i=0;i<k;i++)
		if(sol[k]==sol[i])
			return false;
	return true;
}

void back(int **mat,int *v,int k,int n,int s,int &min,int d,int *sol,int c)
{
	if(c==k)
	{
		d+=mat[sol[k-1]][n];
		if(d<min)
			min=d;
		d-=mat[sol[k-1]][n];
		return;
	}
	for(int i=0;i<k;i++)
	{
		d+=mat[s][v[i]];
		sol[c]=v[i];
		if(bun(sol,c))
		back(mat,v,k,n,v[i],min,d,sol,c+1);
		d-=mat[s][v[i]];
	}
}


int main()
{


	FILE*f=fopen("ubuntzei.in","r"),*fout=fopen("ubuntzei.out","w");
	int n,m,k,*v;
	fscanf(f,"%d",&n);
	fscanf(f,"%d",&m);
	fscanf(f,"%d",&k);
	v=(int *)malloc(k*sizeof(int));
	for(int i=0;i<k;i++)
		fscanf(f,"%d",&v[i]);
	nod **a=(nod**)malloc((n+2)*sizeof(nod*));
	for(int i=1;i<=n;i++)
		a[i]=NULL;
	int x,y,z;
	for(int i=1;i<=m;i++)
	{
		fscanf(f,"%d",&x);
		fscanf(f,"%d",&y);
		fscanf(f,"%d",&z);
		nod*p=(nod*)malloc(sizeof(nod)),*r=(nod*)malloc(sizeof(nod));
		p->val=y;
		p->next=r->next=NULL;
		p->cost=r->cost=z;
		r->val=x;
		if(a[x]==NULL)
		{
			a[x]=p;
		}
		else
		{
			nod *q=a[x];
			while (q->next)
			{
				q=q->next;
			}
			q->next=p;
		}
		if(a[y]==NULL)
		{
			a[y]=r;
		}
		else
		{
			nod *q=a[y];
			while (q->next)
			{
				q=q->next;
			}
			q->next=r;
		}
	}
	int **mat=(int **)malloc((n+2)*sizeof(int*));
	mat[1]=dijkstra(a,n,1);
	for(int i=0;i<k;i++)
	{
		mat[v[i]]=dijkstra(a,n,v[i]);
	}
	int min=INT_MAX,sol[200];
	back(mat,v,k,n,1,min,0,sol,0);
	
	fprintf(fout,"%d",min);
	fclose(f);
	fclose(fout);
	return 0;
}