hdu 2682 Tree

点击打开链接

和hdu1875一样的

#include
#include
#include
#define inf 99999999
#define N 610
int map[N][N],v[N],dis[N],n,sum,num[N],a[1000005];
int min(int a,int b,int c)
{
	if(a>b)a=b;
	if(a>c)a=c;
	return a;
}
void prm()
{
	memset(a,0,sizeof(a));
	a[0]=a[1]=1;
	for(int i=2;i<=1000000;i++)
		if(!a[i])
			for(int j=i+i;j<=1000000;j+=i)
				a[j]=1;
}
void DIJ()
{
	int i,j,tem,t;
	memset(v,0,sizeof(v));
	for(i=1;i<=n;i++)dis[i]=map[1][i];
	v[1]=1;
	dis[1]=0;
	for(i=1;idis[j])
			{
				tem=dis[j];
				t=j;
			}
		}
		if(tem==inf)
		{
			sum=-1;
			break;
		}
		v[t]=1;sum+=tem;
		for(j=1;j<=n;j++)
		{
			if(!v[j]&&map[t][j]temp)
						map[i][j]=map[j][i]=temp;
				}
		}
		DIJ();
		printf("%d\n",sum);
	}
	return 0;
}