模板题——图论相关（2）

最新推荐文章于 2024-05-02 14:51:33 发布

兴安仓鼠

最新推荐文章于 2024-05-02 14:51:33 发布

阅读量148

点赞数

分类专栏：算法基础模板

本文链接：https://blog.csdn.net/LovelyZhou/article/details/113176002

版权

算法基础模板专栏收录该内容

20 篇文章 0 订阅

订阅专栏

1.Floyd求最短路

#include <bits/stdc++.h>

using namespace std;
const int N=210,INF=1e9;
int n,m,q;
int d[N][N];//邻接矩阵
void floyd()
{
    for(int k=1;k<=n;k++)
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&q);
    for(int i=1;i<=n;i++)//初始化
        for(int j=1;j<=n;j++)
        if(i==j) d[i][j]=0;
        else d[i][j]=INF;
    while(m--)//建图
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        d[a][b]=min(d[a][b],c);//防止重边
    }
    floyd();
    while(q--)//询问
    {
        int a,b;
        scanf("%d%d",&a,&b);
        int t=d[a][b];
        if(t>INF/2) puts("impossible");
        else printf("%d\n",t);
    }
    return 0;
}

2.spfa求最短路

#include <bits/stdc++.h>

using namespace std;
const int N=1e6+10;//变成稀疏图，使用邻接表
int n,m;
int h[N],w[N],e[N],ne[N],idx;
int dist[N];
bool st[N];
void add(int a,int b,int c)
{
    e[idx]=b;w[idx]=c;ne[idx]=h[a];h[a]=idx++;
}
int spfa()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    queue<int>q;
    q.push(1);
    st[1]=true;
    while(q.size())
    {
        int t=q.front();
        q.pop();
        st[t]=false;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[t]+w[i])
            {
                dist[j]=dist[t]+w[i];
                if(!st[j])
                {
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }
    if(dist[n]==0x3f3f3f3f) return -1;
    else return dist[n];
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    if(spfa()==-1) puts("impossible");
    else cout<<spfa()<<endl;
    return 0;
}

3.spfa判断负环

#include <bits/stdc++.h>

using namespace std;
const int N = 2010, M = 10010;
int n, m;
int h[N], w[M], e[M], ne[M], idx;
int dist[N], cnt[N];
bool st[N];
void add(int a,int b,int c)
{
    e[idx]=b;w[idx]=c;ne[idx]=h[a];h[a]=idx++;
}
int spfa()
{

    queue<int>q;
   for(int i=1;i<=n;i++)
   {
       st[i]=true;
       q.push(i);
   }
    while(q.size())
    {
        int t=q.front();
        q.pop();
        st[t]=false;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[t]+w[i])
            {
                dist[j]=dist[t]+w[i];
                cnt[j]=cnt[t]+1;
                if(cnt[j]>=n) return true;
                if(!st[j])
                {
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }
    return false;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    if(spfa()) puts("Yes");
    else puts("No");
    return 0;
}

4.Prim求最小生成树：加点法

#include <bits/stdc++.h>

using namespace std;
const int N=510,INF=0x3f3f3f3f;
int n,m;
int g[N][N];
int dist[N];//每个点到集合的距离
bool st[N];
int prim()
{
    memset(dist,0x3f,sizeof dist);
    int res=0;//存储最短边的权值之和
    for(int i=0;i<n;i++)//要将n个点加入集合，所以都判断一遍
    {
        int t=-1;//找权值最小的点的过程
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
            t=j;
        if(i&&dist[t]==INF) return INF;//不是第一个点并且不与任何一点连通
        if(i) res+=dist[t];//不是第一个点
        st[t]=true;
        for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);
        //用加进来的点来更新最小距离
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof g);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        g[a][b]=g[b][a]=min(g[a][b],c);//创建一个无向图
    }
    int t=prim();
    if(t==INF) puts("impossible");
    else printf("%d\n",t);
    return 0;
}

5.kurskal求最小生成树：并查集的应用，加边法

#include <bits/stdc++.h>

using namespace std;
const int N=100010,M=200010,INF=0x3f3f3f3f;
int n,m;
int p[N];
struct Edge
{
    int a,b,w;
    bool operator <(const Edge &W)const//重载一个按照w来排序的规则
    {
        return w<W.w;
    }
}edges[M];
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int kruskal()
{
    sort(edges,edges+m);
    for(int i=1;i<=n;i++) p[i]=i;//初始化并查集
    int res=0,cnt=0;//初始化权值之和以及加入集合的边数
    for(int i=0;i<m;i++)
    {
        int a=edges[i].a,b=edges[i].b,w=edges[i].w;
        a=find(a),b=find(b);
        if(a!=b)//不在同一个集合
        {
            p[a]=b;
            res+=w;
            cnt++;
        }
    }
    if(cnt<n-1) return INF;
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        edges[i]={a,b,w};
    }
    int t=kruskal();
    if(t==INF) puts("impossible");
    else printf("%d\n",t);
    return 0;
}

6.匈牙利算法求二分图的最大匹配数——>用y总的月老方法理解

#include <bits/stdc++.h>

using namespace std;
const int N=510,M=100010;
int n1,n2,m;
int h[N],e[M],ne[M],idx;
int match[N];
bool st[N];
void add(int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
bool find(int x)
{
    for(int i=h[x];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(!st[j])
        {
            st[j]=true;
            if(match[j]==0||find(match[j]))
            {
                match[j]=x;
                return true;
            }
        }
    }
    return false;
}
int main()
{
    scanf("%d%d%d",&n1,&n2,&m);
    memset(h,-1,sizeof h);
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    int res=0;//记录当前匹配数
    for(int i=1;i<=n1;i++)
    {
        memset(st,false,sizeof st);
        if(find(i)) res++;
    }
    printf("%d\n",res);
    return 0;
}

7.1224. 交换瓶子

与1215. 小朋友排队非常类似，一开始我甚至想直接去复制代码\笑死，但是题目要求稍有不同（本题可以交换任意两个位置），使用图论的知识进行解答：每个瓶子应该在的编号和它当前的位置编号进行连线，会形成许多环
下面分为两种情况：
①交换同一个环内的点——>裂成2个环
②交换不同环中的点——>合并两个环
问题转化为：将当前图中的环全部变成自环需要走多少步。如果是分裂每次都会多一个环，所以只需要求当前图中k个环变成n个环的步数n-k,问题再次转换成求当前图中环的数量。【妙啊】

#include <bits/stdc++.h>

using namespace std;
const int N=10010;
int n,b[N];
bool st[N];

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    int cnt=0;
    for(int i=1;i<=n;i++)
        if(!st[i])
    {
        cnt++;//环数+1
        for(int j=i;!st[j];j=b[j])//把j点连接的所有点（一个环上的点）都变成true
            st[j]=true;
    }
    printf("%d\n",n-cnt);
    return 0;
}