HNUCM-OJ春季学期《算法分析与设计》练习13题解

HNUCM-OJ春季学期《算法分析与设计》练习13题解

图论=>并查集+Kruskal+Dijkstra

来源/分类	解决
A	迷路的牛牛
B	工作单位
C	隔离14天
D	最小生成树（Kruskal）
E	搭建电路
F	单源最短路径问题

A：取余+模拟

AC源码：

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n;
    while (~scanf("%d",&n))
    {
        int a=0;
        char tmp;
        while (n--)
        {
            cin>>tmp;
            if (tmp=='L')
            {
                a--;
                if(a<0)
                a=(a+4)%4;
            }
            else
            {
                a++;
                if (a>3)
                {
                    a=a%4;
                }
            }
            
        }
        if (a==0)
        {
            cout<<"N"<<endl;
        }
        if (a==1)
        {
            cout<<"E"<<endl;
        }
        if (a==2)
        {
            cout<<"S"<<endl;
        }
        if (a==3)
        {
           cout<<"W"<<endl;
        }
    }
    return 0;
}

B：用并查集数集合个数

AC源码：

#include <bits/stdc++.h>
using namespace std;
const int cmax = 1e3+5;
int pa[cmax];   //pa[x]=x表示结点x的父节点是x
int rnk[cmax];  //rnk[x]=x表示结点x的子节点的层数，代表结点x的秩
int res;

//  查找结点x的父亲结点
int find_set(int x){
    if(x!=pa[x])
    pa[x]=find_set(pa[x]);   // 带路径压缩的算法
    return pa[x];
}
// 将集合合并
void union_set(int x,int y){
    x=find_set(x);
    y=find_set(y);
    if(x==y)
    return ;
    res--;
    if(rnk[x]>rnk[y]){
        pa[y]=x;
    }
    else{
        pa[x]=y;
        if(rnk[x]==rnk[y]){
            rnk[y]++;
        }
    }
}
int main()
{
    int n,m;
    res=0;
    cin>>n>>m;
    //  初始化每一个结点，结点i的父亲结点是他本身
    for (int i = 0; i <=n ; i++)
    {
        pa[i]=i;
        rnk[i]=0;
    }
    res=n;
    int tmp1,tmp2;
    for (int i = 0; i < m; i++)
    {
        scanf("%d%d",&tmp1,&tmp2);
        union_set(tmp1,tmp2);
    }   
    cout<<res<<endl;
    return 0;
}

C:使用并查集数某一个结点所在的集合的元素的总数

AC源码

#include <bits/stdc++.h>
using namespace std;
const int cmax = 1e3+5;
int pa[cmax];   //pa[x]=x表示结点x的父节点是x
int rnk[cmax];  //rnk[x]=x表示结点x的子节点的层数，代表结点x的秩
int num[cmax];  //num[x]=k 表示结点x的数的集合元素的个数

//  查找结点x的父亲结点
int find_set(int x){
    if(x!=pa[x])
    pa[x]=find_set(pa[x]);   // 带路径压缩的算法
    return pa[x];
}
// 将集合合并
void union_set(int x,int y){
    x=find_set(x);
    y=find_set(y);
    if(x==y)
    return ;
    if(rnk[x]>rnk[y]){
        pa[y]=x;
        num[x]+=num[y];
    }
    else{
        pa[x]=y;
        num[y]+=num[x];
        if(rnk[x]==rnk[y]){
            rnk[y]++;
        }
    }
}
int main()
{
    int n,m;
    cin>>n>>m;
    for (int i = 0; i <=n ; i++)
    {
        pa[i]=i;
        rnk[i]=0;
        num[i]=1;
    }
    for(int i=0;i<m;i++){
        int tmp,k1,k2;
        cin>>tmp>>k1;
        tmp--;
        while (tmp>0)
        {
            cin>>k2;
            union_set(k1,k2);
            k1=k2;
            tmp--;
        }
    }
    cout<<num[pa[0]]<<endl;
    return 0;
}

D：Kruskal算法，求图的最小生成树(MST)的权重

AC源码：

#include <bits/stdc++.h>
using namespace std;
const int cmax = 1e3+5;
struct edge
{
    int x;int y;
    int w;
};

bool cmp (edge m ,edge n){
    return m.w<n.w;
}
int pa[cmax];   //pa[x]=x表示结点x的父节点是x
int rnk[cmax];  //rnk[x]=x表示结点x的子节点的层数，代表结点x的秩
edge a[cmax];
int sum;
//  查找结点x的父亲结点
int find_set(int x){
    if(x!=pa[x])
    pa[x]=find_set(pa[x]);   // 带路径压缩的算法
    return pa[x];
}
// 将集合合并
void union_set(int x,int y,int w){
    x=find_set(x);
    y=find_set(y);
    if(x==y)
    return ;
    if(rnk[x]>rnk[y]){
        pa[y]=x;
    }
    else{
        pa[x]=y;
        if(rnk[x]==rnk[y]){
            rnk[y]++;
        }
    }
    sum+=w;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m;
    cin>>n>>m;
    for (int i = 0; i <= n; i++)
    {
        pa[i]=i;
        rnk[i]=0;
    }
    for(int i=0;i<m;i++){
        cin>>a[i].x>>a[i].y>>a[i].w;
    }
    sum=0;
    sort(a,a+m,cmp);
    for (int i = 0; i < m; i++)
    {
        union_set(a[i].x,a[i].y,a[i].w);
    }
    cout<<sum<<endl;
    return 0;
}

E：Kruskal算法，求图的最大生成树(MST)的权重

AC源码

#include <bits/stdc++.h>
using namespace std;
const int cmax = 1e3+5;
typedef long long ll;
struct edge
{
    int x;int y;
    int w;
};
bool cmp (edge m ,edge n){
    return m.w>n.w;
}
int pa[cmax];   //pa[x]=x表示结点x的父节点是x
int rnk[cmax];  //rnk[x]=x表示结点x的子节点的层数，代表结点x的秩
edge a[cmax];
ll sum;
//  查找结点x的父亲结点
int find_set(int x){
    if(x!=pa[x])
    pa[x]=find_set(pa[x]);   // 带路径压缩的算法
    return pa[x];
}
// 将集合合并
int union_set(int x,int y,int w){
    x=find_set(x);
    y=find_set(y);
    if(x==y)
    return 0;
    if(rnk[x]>rnk[y]){
        pa[y]=x;
    }
    else{
        pa[x]=y;
        if(rnk[x]==rnk[y]){
            rnk[y]++;
        }
    }
    sum+=w;
    return 1;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m;
    cin>>n>>m;
    for (int i = 0; i <= n; i++)
    {
        pa[i]=i;
        rnk[i]=0;
    }
    for(int i=0;i<m;i++){
        cin>>a[i].x>>a[i].y>>a[i].w;
    }
    sum=0;
    sort(a,a+m,cmp);
    int flag = 0;
    for (int i = 0; i < m; i++)
    {
        int res=union_set(a[i].x,a[i].y,a[i].w);
        if(res){
            flag++;
        }
    }
    if(flag==n-1)
    cout<<sum<<endl;
    else
    cout<<"No solution."<<endl;
    return 0;
}

F：Dijkstra算法，求一个有向加权图中，从源点出发到其他各个顶点的最短路径。

单源最短路径问题

AC源码

#include <bits/stdc++.h>
using namespace std;
const int cmax = 1e3+5;
typedef long long ll ;
const int INF = 0x3f3f3f3f;
int a[cmax][cmax];
bool vis[cmax];
int len[cmax];
void prim(int n){
    vis[0]=true;
    for (int i = 0; i < n; ++i) {
        len[i]=a[0][i];
    }
    len[0]=0;
    int flag=0;
    for (int i = 0; i < n; ++i) {
        int j=INF;
        for (int k = 0; k < n; ++k) {
            if (!vis[k]&&len[k]<j){
                flag=k;
                j=len[k];
            }
        }
        vis[flag]=true;
        for (int i = 1; i < n; ++i) {
            if (!vis[i]&&(len[flag]+a[flag][i])<len[i]){
                len[i]=len[flag]+a[flag][i];
            }
        }
    }
    for (int i = 0; i < n; i++)
    {
        cout<<len[i]<<" ";
    }
    cout<<endl;
}
int main(){
    int n,m;
    while (~scanf("%d%d",&n,&m)){
        int start,end,weight;
        memset(a,INF,sizeof(a));
        memset(vis, false,sizeof(vis));
        for (int i = 0; i < m; ++i) {
            scanf("%d%d%d",&start,&end,&weight);
            a[start][end]=weight;
            a[end][start]=weight;
        }
        prim(n);
    }
    return 0;
}