YBT G. 3.生物进化

题目分析:

这道题的关键是找出隐含条件

很容易看出,原图是一颗以节点1为根的有根树。求得的祖孙关系会是一条包括所有点的树。于是就可以猜到使用最小生成树解决问题。
由于答案是输出每一个生物的直系祖先,而根据题意,点a的直系祖先一定是所有点中距离点a最近的点。(可以反证法证明)
而这个prim的过程就是不断寻找最近的点,因此选择prim作为解法。

Code

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e2+20,INF=1e9+7;
int n,w[maxn][maxn],dis[maxn],pre[maxn];
bool vis[maxn];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j) 
			scanf("%d",&w[i][j]);
	memset(dis,0x7f,sizeof(dis));
	dis[1]=0;
	for(int i=1;i<=n;++i)
	{
		int k=0,minn=INF;
		for(int j=1;j<=n;++j)
		{
			if(!vis[j] && dis[j]<minn)
			{
				k=j;
				minn=dis[j];	
			}
		}
		vis[k]=true;
		for(int j=1;j<=n;++j)
		{
			if(!vis[j] && dis[j] > w[k][j])
			{
				pre[j]=k;
				dis[j]=w[k][j];
			}
		}
	}
	for(int i=2;i<=n;++i) printf("%d\n",pre[i]);
	return 0;	
}
posted @ 2021-09-09 11:17  Mint-hexagram  阅读(74)  评论(0编辑  收藏  举报