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;
}