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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章