Navigation System-最短路计数

CF1320B Navigation System

题目描述

题解

最短路计数
首先反向建边，并对$p_i$维护其下一个数即$p_{i+1}=nxt[p_i]$
然后从终点跑最短路，如果$nxt[p_i]$在一条终点到$p_i$的最短路上，给$p_i$打上标记
最后统计答案
注意：更新最短路时，要消去原来的标记

代码实现

#include<bits/stdc++.h>
#define M 1000009
using namespace std;
int nxt[M],to[M],tot,num[M],first[M],pd[M],d[M],vis[M],n,m,k,ans1,ans2,t,s,nx[M],bj[M],p[M];
const int inf=1e9;
int read(){
	int f=1,re=0;
	char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
void add(int x,int y){
	nxt[++tot]=first[x];
	first[x]=tot;
	to[tot]=y;
}
void dj(){
	for(int i=1;i<=n;i++) d[i]=inf;
	priority_queue<pair<int,int> >q;
	q.push(make_pair(d[t]=0,t));vis[t]=1;
	while(q.size()){
		int u=q.top().second;q.pop();
		for(int i=first[u];i;i=nxt[i]){
			int v=to[i];
			if(d[v]==d[u]+1){
				if(bj[v]&&u==nx[v]) pd[v]=1;
				num[v]++;
				if(!vis[v]) q.push(make_pair(-d[v],v)),vis[v]=1;
			}
			if(d[v]>d[u]+1){
				pd[v]=0;
				if(bj[v]&&u==nx[v]) pd[v]=1;
				d[v]=d[u]+1;
				num[v]=1;
				if(!vis[v]) q.push(make_pair(-d[v],v)),vis[v]=1;
			}	
		}
	}
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=m;i++){
		int x=read(),y=read();
		add(y,x);
	}k=read();
	for(int i=1;i<=k;i++) p[i]=read(),bj[p[i]]=1;
	for(int i=1;i<=k;i++) nx[p[i]]=p[i+1];
	s=p[1],t=p[k],dj();
	for(int i=1;i<=k;i++){
		//printf("%d ",num[p[i]]);
		if(!pd[p[i]]) ans1++;
		if(!pd[p[i]]||num[p[i]]>1) ans2++;
	}printf("%d %d\n",ans1-1,ans2-1);
	return 0;
}