ACM新手DAY 7 最小生成树

题解

A - Constructing Roads（样题）

题目：有N个村庄，编号从1到N，建造一些道路使每两个村庄可以相互连接。 我们说两个村A和B是相连的，当且仅当A和B之间有一条道路，或者存在一个村C以便在A和C之间有一条道路，并且C和B相连。
一些村庄之间已经有一些道路，你的工作就是修建一些道路，使所有村庄都连通起来，所有道路的长度都是最小的。求此时需要修多长的路。

• 最小生成树
• 第一题是模板，直接放代码。当然因为题目中存在有一些路已经修过了，算是附加条件，这个模板还不算普遍的。

#include<bits/stdc++.h>
using namespace std;
int f[200];
struct edge
{
    int s,to,cost;
} e[200*200];
bool cmp(edge a,edge b)
{
    return a.cost<b.cost;
}
int getf(int n)
{
    if(n==f[n]) return f[n];
    return f[n]=getf(f[n]);
}
bool merge1(int u,int v)
{
    int t1=getf(u),t2=getf(v);
    if(t1==t2) return false;
    f[t1]=t2;//f[t2]=t1也行
    return true;
}
int main()
{
    ios::sync_with_stdio(0);
    int n;
    while(cin>>n)
    {
        int cnt=0;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                int a;
                cin>>a;
                e[cnt++]= {i,j,a};
            }
        }
        int c;
        cin>>c;
        for(int i=0; i<c; i++)
        {
            int a,b;
            cin>>a>>b;
            e[cnt++]= {a,b,0};
        }
        //将每条边按从小到大的顺序排序
        sort(e,e+cnt,cmp);
        //并查集初始化
        for(int i=1; i<=n; i++) f[i]=i;
        int cc=0;
        int sum=0;
        for(int i=0; i<cnt; i++)
        {
            //每次选出一条边的两个顶点进行判断
            if(merge1(e[i].s,e[i].to))
            {
                sum+=e[i].cost;
                cc++;
            }
            //cc用来记录已经选了多少条边了 如果选的边数量为n-1条边
            //就选够了 可以退出 不用选了
            if(cc==n-1) break;
        }
        cout<<sum<<endl;
    }
}

B - Jungle Roads

题目：依旧是权值和最小

• 最小生成树
• 城镇编号变成了字符，转换一下，然后道路长度换成维护费用就好了。
• 有一点注意一下

 for(int i=1; i<n; i++)
        {
            char cha; cin >> cha;//因为看样例是A.B.C.D...依次排的，读入是读入，但不会用到
            int k; cin >> k;
            while(k--)
            {
                char ch; cin >> ch;
                int j = ch - 'A' + 1;
                int a; cin >> a;
                e[cnt++]= {i,j,a};
            }
        }

C - Tree

题目：权值和最小，两个点可不可以连在一起需要判断，两个点之间的权值需要计算。

• 最小生成树（变化）
• 两个能不能连在一起加一个判断，主要的问题在控制不超时。
• 比较快的素数判断

bool sushu(int n)
{
    if(n < 2)return 0;
    if(n ==2 || n == 3)return 1;
    if(n % 6 != 1 && n % 6 != 5)return 0;
    for(int i = 5; i * i <= n; i += 6)
    {
        if(n % i == 0 || n % (i + 2) == 0)
        return 0;
    }
    return 1;
}