poj1789 Truck History(最小生成树)

我是通过prim来做的,代码写得很水,一直超时。。。最后1300ms过的,看到别人用相同的方法的只要300ms,一直搞不懂为啥。我的思路是,把题目转化为求最小生成树。两个truck的不同可以看为两点之间的权值。那么,就可以构造出一幅无相完全图,稠密图一般使用prim算法,求出其生成树的权值和即可。
#include<iostream>
#include<string>
using namespace std;
const int inf=10; 
const int max_type = 2001;
string type[max_type];
int dis[max_type][max_type];
bool vis[max_type];
int s_dis[max_type];
int cmp(string s1,string s2)					/*比较字符串,返回权值*/
{
	int count(0);
	for(string::size_type i=0;i!=7;i++)
	{
		if(s1[i] != s2[i])
			count++;
	}
	return count;
}
int main()
{
	int N;
	while(cin>>N&&N!=0)
	{
		for(int i=0;i!=N;i++)
			cin>>type[i];
		for(int i=0;i!=N-1;i++)              /*这个循环可以减少一些重复比较(一开始是两重1->n的循环,超时了!!!)*/
			for(int j=i+1;j!=N;j++)
				dis[i][j] = dis[j][i] = cmp(type[i],type[j]);
		memset(vis,false,sizeof(vis));		 /*记录这个点是否在生成树中已经有了*/
		memset(s_dis,inf,sizeof(s_dis));	 /*s_dis[s]数组记录s到达其邻接点的最短距离*/
		int Q(0);int n(1),s(0);
		vis[0] = true;						 /*现将0点放入树中*/
		while(N!=n)							 /*prim算法更新树,当所有节点进入树中,结束更新*/
		{
			int min_dis = 10,point(-1);
			for(int i=1;i!=N;i++)
			{
				if(!vis[i] && dis[s][i]<s_dis[i])
					s_dis[i] = dis[s][i];
				if(!vis[i] && min_dis>s_dis[i])
				{
					min_dis=s_dis[i];
					point = i;
				}
			}
			s = point;
			vis[s] = true;
			Q+=min_dis;
			n++;
		}
		cout<<"The highest possible quality is 1/"<<Q<<"."<<endl;
	}
	return 0;
}

发布了50 篇原创文章 · 获赞 5 · 访问量 8万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章