prim算法(普里姆算法)

Prim算法:
http://blog.chinaunix.net/uid-25324849-id-2182922.html

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=105;
const int INF=1<<29;
int nearvex[N],lowcost[N];
int edge[N][N];
int n,m;
void init()
{
    fill(nearvex,nearvex+N,1);
    fill(lowcost,lowcost+N,INF);
    lowcost[1]=0;
    for(int i=1;i<=n;i++)
    {
        if(edge[1][i])
        {
            lowcost[i]=edge[1][i];
        }
    }
    nearvex[1]=-1;
}
void Prim()
{
    int v;
    int lg=INF;
    for(int i=1;i<=n;i++)
    {
        if(nearvex[i]==-1) continue;
        if(lg>=edge[nearvex[i]][i])
        {
            v=i;
            lg=edge[nearvex[i]][i];
        }
    }
    cout<<nearvex[v]<<" "<<v<<" "<<edge[nearvex[v]][v]<<endl;
    nearvex[v]=-1;
    for(int i=1;i<=n;i++)
    {
        lowcost[i]=min(lowcost[i],edge[v][i]);
    }
    for(int i=1;i<=n;i++)
    {
        if(nearvex[i]==-1) continue;
        if(edge[v][i]<edge[nearvex[i]][i])
            nearvex[i]=v;
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    int u,v;
    for(int i=0;i<N;i++)
        fill(edge[i],edge[i]+N,INF);
    for(int i=0;i<m;i++)
    {
        cin>>u>>v;
        scanf("%d",&edge[u][v]);
        edge[v][u]=edge[u][v];
    }
    init();
    for(int i=1;i<=n-1;i++)
        Prim();
    return 0;
}

zoj 1586(简单题)
http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=586
将lowcost和nearvex数组合并成一个lowcost数组。若lowcost【i】=-1;则表示该顶点在构造的最小生成树中(最小生成树上的顶点集合设为T),否则lowcost的值表示顶点i到T中各顶点权值的最小边权值。
思路:每条边的权值=该边两端适配器的价格+网线的价格。然后转为模板题。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1005;
const int INF=1<<29;
int lowcost[N];
int edge[N][N];
int n;
int cost[N];
int weight; //费用
void init()
{
    weight=0;
    fill(lowcost,lowcost+N,INF);
    lowcost[1]=-1;
}
void prim(int num)
{
    for(int i=1;i<=n;i++)
    {
        if(lowcost[i]==-1) continue;
        lowcost[i]=edge[num][i]+cost[num]+cost[i];
    }
    for(int i=1;i<n;i++)
    {
        int lg=INF;
        int v=-1;
        for(int j=1;j<=n;j++)
        {
            if(lowcost[j]==-1) continue;
            if(lg>=lowcost[j])
            {
                v=j;
                lg=lowcost[j];
            }
        }
        if(v!=-1)
        {
            weight+=lg;
            //cout<<weight<<endl;
            lowcost[v]=-1;
            for(int j=1;j<=n;j++)
            {
                if(lowcost[j]==-1) continue;
                lowcost[j]=min(edge[v][j]+cost[v]+cost[j],lowcost[j]);
            }
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    int Cas;
    cin>>Cas;
    while(Cas--)
    {
        cin>>n;
        for(int i=1;i<=n;i++)
        {
            cin>>cost[i];
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                cin>>edge[i][j];
            }
        }
        init();
        prim(1);
        cout<<weight<<endl;
    }
    return 0;
}

zoj2158(水题):
http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=1158
计算出所有卡车类型相互之间的距离,存储于edge数组,然后就变成模板题了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=7;
const int maxn=2005;
const int INF=1<<29;
char ch[maxn][N];
int lowcost[maxn];
int edge[maxn][maxn];
int n;
int weight;
void init()
{
    weight=0;
    memset(edge,0,sizeof(edge));
}
int compare(char s[],char c[])
{
    int ans=0;
    for(int i=0;i<7;i++)
    {
        if(s[i]!=c[i])
            ans++;
    }
    return ans;
}
void prim()
{
    for(int i=1;i<=n;i++)
    {
        lowcost[i]=edge[1][i];
    }
    lowcost[1]=-1;
    for(int j=1;j<n;j++)
    {
        int v=-1;
        int lg=INF;
        for(int i=1;i<=n;i++)
        {
            if(lowcost[i]==-1) continue;
            if(lg>lowcost[i])
            {
                lg=lowcost[i];
                v=i;
            }
        }
        weight+=lg;
        if(v!=-1)
        {
            lowcost[v]=-1;
            for(int i=1;i<=n;i++)
            {
                if(lowcost[i]==-1) continue;
                if(edge[v][i]<lowcost[i])
                {
                    lowcost[i]=edge[v][i];
                }
            }
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(cin>>n)
    {
        if(!n) break;
        for(int i=1;i<=n;i++)
        {
            getchar();
            cin>>ch[i];
        }
        init();
        for(int i=1;i<n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                int d=compare(ch[i],ch[j]);
                //cout<<d<<endl;
                edge[i][j]=edge[j][i]=d;
            }
        }
        prim();
        printf("The highest possible quality is 1/%d.\n",weight);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值