POJ-1789 Truck History

题目链接:https://vjudge.net/problem/POJ-1789

一开始看这道题是一脸懵(英文水平有待提高),后来去搜了下题解,懂了题意就很简单了

题意大概是这样的:用一个7位的string代表一个编号,两个编号之间的距离就是这两个编号之间不同字母的个数

然后就是求n个编号的最小生成树,我用了Kruskal,一开始TLE,做了些常数优化就AC了

以后碰到数据量大的要注意尽量节约时间

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=2000+10;//点数
const int MAXM=2000000+10;//边数
int UF[MAXN];
struct Edge
{
	int u,v,w;
}edge[MAXM];
int tol; //初始化为0 
void addedge(int u,int v,int w)
{
	edge[tol].u=u;
	edge[tol].v=v;
	edge[tol++].w=w;
} 
bool cmp(Edge a,Edge b) 
{
	return a.w<b.w; 
}
int find(int x) 
{  
	if(UF[x]==-1) return x;  
	else return UF[x]=find(UF[x]);
}
int Kruskal(int n)
{
	memset(UF,-1,sizeof(UF));
	sort(edge,edge+tol,cmp);
	int ans=0;
	int cnt=0;
	for(int i=0;i<tol;i++)
	{
		int u=edge[i].u;   
		int v=edge[i].v;   
		int w=edge[i].w; 
		int t1=find(u);
		int t2=find(v);
		if(t1!=t2)
		{
			UF[t1]=t2;
			ans+=w;
			cnt++;
		}
		if(cnt==n-1) break;
	}
	if(cnt<n-1) return -1;
	return ans;
}
int f(char a[],char b[])
{
	int num=0;
	for(int i=0;i<7;i++)
	if(a[i]!=b[i]) num++;
	return num;
}
char a[MAXN][8];
int main()
{
	int n;
	while(~scanf("%d",&n)&&n)
	{
		tol=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%s",a[i]);
			for(int j=1;j<i;j++)
			addedge(i,j,f(a[i],a[j]));
		}
		printf("The highest possible quality is 1/%d.\n",Kruskal(n));
	}
	return 0;
}
本站公众号
   欢迎关注本站公众号,获取更多程序园信息
开发小院