GYM 101149 L.Right Build（SPFA）

Description
给出一个n个点m条边的有向连通图每次只有起点可以用终点才可以用，初始时0点可以用，问最少要用多少点才能用到a点和b点
Input
第一行四个整数n,m,a,b分别表示点数，边数，要用的两个点，之后m行每行两个整数u和v表示u->v有一条有向边(2<=n,m<=2e5)
Output
输出最少要用多少点
Sample Input
6 8 4 6
0 1
0 2
1 2
1 5
2 3
2 4
3 6
5 6
Sample Output
4
Solution
如果只是要用一个点a，那么求0->a的最短路即可，但是要用两个点就可能0->a和0->b都不是最短路，但是由于路径重叠很多点可以被用两次以节省用的点数，但是注意到两条路径只会是开始时重叠在一次，之后分开就再也不会相交，因为如果再次相交不如之前全部重叠更优，那么我们枚举这个叉点c，求出0->c,a->c,b->c的最短路来更新最优解即可
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
#define INF 0x3f3f3f3f
struct edge
{
    int to,next;
    int cost;
}g[maxn];
int head[maxn],tol;
void init()//初始化 
{
    memset(head,-1,sizeof(head));
    tol=0;
}
void add(int u,int v,int c)//单向边,从u到v,权值为c 
{
    g[tol].cost=c;
    g[tol].to=v;
    g[tol].next=head[u];
    head[u]=tol++;
}
void spfa(int s,int *dis)//单源最短路,s是起点
{
    bool vis[maxn];
    memset(vis,false,sizeof(vis));
    queue<int>que;
    for(int i=0;i<maxn;i++)dis[i]=INF;
    dis[s]=0;
    vis[s]=true;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=g[i].next)
        {
            int v=g[i].to;
            int c=g[i].cost;
            if(dis[v]>dis[u]+c)
            {
                dis[v]=dis[u]+c;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                }
            }
        }
    }
}
int n,m,a,b,e[maxn][2],dis1[maxn],dis2[maxn],dis3[maxn];
int main()
{
    while(~scanf("%d%d%d%d",&n,&m,&a,&b))
    {
        init();
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&e[i][0],&e[i][1]);
            add(e[i][0],e[i][1],1); 
        }
        spfa(0,dis1);
        init();
        for(int i=1;i<=m;i++)add(e[i][1],e[i][0],1);
        spfa(a,dis2),spfa(b,dis3);
        int ans=n;
        for(int i=0;i<=n;i++)
            ans=min(dis1[i]+dis2[i]+dis3[i],ans);
        printf("%d\n",ans);
    }
    return 0;
}