Codeforces Round #625 (Div. 2, based on Technocup 2020 Final Round) D - Navigation System(最短路 BFS)

传送门

题意:

给一个有向图,然后有一条长度为k要走的路径,起点,终点确定
如果在路径上走的时候,发现有可以到达终点的更短的路径,可以重新选择路径
问重新选择次数的最小值和最大值

思路:

先反向建边,bfs算出终点到每个点的最短路径
考虑重新选择路径的情况
1,如果在走的路径上的下一个点到终点不是最短的且有其他比他更短的路径,那么一定要重新选择
2,如果在走的路径上的下一个点到终点是最短的且还有其他和他一样短的路径,那么可以换也可以不换,如果一直不换就是最小值,如果一直换就是最大值

代码:

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#include <set>
#include <stack>
typedef long long ll;
#define PII make_pair
#define pb push_back
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int MAXN=2e5+50;
const int inf=0x3f3f3f3f;
const int M=5000*4;
int n,m,k;
vector<int>p[MAXN],pf[MAXN];
int a[MAXN];
int dis[MAXN];
int vis[MAXN];
void bfs(){
    queue<int>q;
    q.push(a[k]);
    vis[a[k]]=1;
    while(!q.empty()){
        int x=q.front();
        q.pop();
        for(int i=0;i<p[x].size();i++){
            if(!vis[p[x][i]]){
                vis[p[x][i]]=1;
                dis[p[x][i]]=dis[x]+1;
                q.push(p[x][i]);
            }
        }
    }
}
int main()
{   
    scanf("%d%d",&n,&m);
    while(m--){
        int u,v;
        scanf("%d%d",&u,&v);
        pf[u].pb(v);
        p[v].pb(u);
    }
    scanf("%d",&k);
    rep(i,1,k)scanf("%d",&a[i]);
    bfs();//求终点到每个点的最短路径dis
    int ans1=0,ans2=0;
    rep(i,1,k-1){
        int f1=0,f2=0;
        for(int j=0;j<pf[a[i]].size();j++){
            int u=a[i],v=pf[a[i]][j];
            if(dis[u]==dis[v]+1&&v!=a[i+1])f1=1;//一条其他符合的最短路
            if(dis[u]!=dis[v]+1&&v==a[i+1])f2=1;//下一个点不是最短路,必须要变
        }
        if(f2)ans1++;//一定要选择
        if(f1==1&&f2==0)ans2++;//可选可不选(f2==0说明下一个点是最短的)
        								//因为循环中肯定有v==a[i+1]的时候,那时候f2都不等于1,说明v==a[i+1]的时,dis[u]==dis[v]+1
    }
    printf("%d %d\n",ans1,ans1+ans2);
    return 0;
} 

刚才那个不好理解的话循环里55-63行的代码可以换成下面的,一个意思

		int f1=0,f2=0,f=0;
        for(int j=0;j<pf[a[i]].size();j++){
            int u=a[i],v=pf[a[i]][j];
            if(dis[u]==dis[v]+1&&v==a[i+1])f=1;//下一个点是最短路
            if(dis[u]==dis[v]+1&&v!=a[i+1])f1=1;//一条其他符合的最短路
            if(dis[u]!=dis[v]+1&&v==a[i+1])f2=1;//下一个点不是最短路,必须要变
        }
        if(f2)ans1++;//一定要选择
        if(f==1&&f1==1)ans2++;//可选可不选
发布了142 篇原创文章 · 获赞 13 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44091178/article/details/104622877