图论模板(二)

SPFA（bellman_ford优化）

1. 将标准点（源点）入队，把队列内的元素都标记，不再入队
2. 如果此元素被遍历，则出队，取消标记

#include<iostream>
#include<cstring>

using namespace std;
const int N=2e5+10;
int n,m;
int e[N],ne[N],w[N],h[N],d[N],q[N],idx;
bool st[N];

void add(int a,int b,int c)
{
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}

void spfa()
{
    memset(d,0x3f,sizeof(d));
    int hh=0,tt=-1;
    q[++tt]=1;
    d[1]=0;
    while(hh<=tt)
    {
        int t=q[hh++];
        st[t]=false;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(d[j]>d[t]+w[i])
            {
                d[j]=d[t]+w[i];
                if(!st[j])
                {
                    q[++tt]=j;
                    st[j]=1;
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    
    memset(h,-1,sizeof(h));
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    
    spfa();
    
    if(d[n]==0x3f3f3f3f) cout<<"impossible";
    else cout<<d[n];
}

SPFA求负环

1. 在SPFA的基础上增加一个cnt变量保存每一个点的入度，每次找到一个边入度加一。
2. 如果入度大于等于N则已经有环了。

#include<iostream>
#include<cstring>

using namespace std;
const int N=1e6+10;
int e[N],ne[N],w[N],h[N],d[N],cnt[N],idx,q[N*2];
int n,m;
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(d,0x3f,sizeof(d));
    
    
    int hh=0,tt=-1;
    
    for (int i = 1; i <= n; i ++ )
    {
        st[i] = true;
        q[++tt]=i;
    }
    while(hh<=tt)
    {
        int t=q[hh++];
        st[t]=0;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(d[j]>d[t]+w[i])
            {
                d[j]=d[t]+w[i];
                cnt[j]=cnt[t]+1;
                if(cnt[j]>=n) 
                {
                    cout<<"Yes";
                    return 0;
                }
                if(!st[j])
                {
                    q[++tt]=j;
                    st[j]=1;
                }
                
            }
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    
    memset(h,-1,sizeof(h));
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    
    if(!spfa())return 0;
    cout<<"No";
}

Floyd（多源最短路） 

#include<iostream>
#include<cstring>

using namespace std;
const int N=300;
int n,m,k;
int d[N][N];
int main()
{
    cin>>n>>m>>k;
    
    memset(d,0x3f,sizeof(d));
    
    for(int i=1;i<=n;i++)
    {
        d[i][i]=0;
    }
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        d[a][b]=min(d[a][b],c);
    }
    
    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]);
            }
        }
    }
    
    while(k--)
    {
        int x,y;
        cin>>x>>y;
        if(d[x][y]>=0x3f3f3f3f/2) cout<<"impossible"<<endl;
        else cout<<d[x][y]<<endl;
    }
}