#4072 最短路计数

WOJ

描述

给出一个N个点M条边的有向带权图，点编号1~N.问从点1到点N的最短路与最短路的条数。

两个不同的最短路方案要求：路径相同（都是最短路长度）且至少有一条边不重合

输入

第一行是N,M

接下来M行，每行3个数字u，v，w表示从u到v有边权为w

保证没有自环，不保证重边

输出

输出一行，如果没有最短路输出No answer

否则输出2个数，第一个数是最短路径长度，第二个数是不同最短路数，中间空格隔开

样例输入

5 4
1 5 4
1 2 2
2 5 2
4 1 1

样例输出

4 2

提示

• 30%数据N<=20

• 100%数据1<=N<=2000 ,0<=M<=N*(N-1),1<=Wi<=10

解析：

这道题还是比较基础，只不过要注意：

• 题中给的是有向图，则在跑SPFA时要注意方向，若是正向建边就只能从1向n正向跑一遍SPFA，当然为了防止毒瘤数据卡SPFA，可以反向建边，从n到1反向跑一遍SPFA
• 由于题中要求计数最短边方案数，且所给数据不大，可以开一个二维vis数组，表示满足节点v最短路时与u相连的方案数vis[v][0]表示节点v所有最短路的方案数
• 更新vis数组前，要判断与v相连目标节点u节点的最短路方案数是否更新过，或vis数组从未更新过与u相连的方案数
• 更新vis数组前，先要减去原来与u节点相连的方案数，再加上更新后的方案数

CODE

#include<bits/stdc++.h>
using namespace std;
const int N=2005,M=4000005;
int read(){
	int s=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') f=-f;
		ch=getchar();
	}
	while(isdigit(ch)){
		s=(s<<3)+(s<<1)+ch-'0';
		ch=getchar();
	}
	return s*f;
}
struct fjy{
	int nxt,v,w;
}e[M];
int cnt;
int first[N];
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].nxt=first[u];
	first[u]=cnt;
}
int dis[N],val[N],vis[N][N];//vis[i][j]表示当满足当前最短路时，v的最短路径过u点的方案数 
void spfa(int x){
	queue<int> q;
	q.push(x),val[x]=1,dis[x]=0,vis[x][0]=vis[x][x]=1;
	while(!q.empty()){
		int u=q.front();q.pop();val[u]=0;
		for(int i=first[u];i;i=e[i].nxt){
			int v=e[i].v,w=e[i].w;//printf("yeah");
			if(dis[v]>dis[u]+w){
				memset(vis[v],0,sizeof vis[v]);
				vis[v][0]=vis[u][0];
				vis[v][u]=vis[u][0];
				dis[v]=dis[u]+w;
				if(!val[v]) q.push(v),val[v]=1/*,printf("v=%d\n",v)*/;
			}
			else	if(dis[v]==dis[u]+w&&vis[v][u]!=vis[u][0]){//当满足最短路条件时，若节点v未存放了连接节点u的方案数，或节点u最短路的总方案数更新过，进行以下操作 
				vis[v][0]-=vis[v][u],vis[v][u]=vis[u][0];vis[v][0]+=vis[v][u];//先要减去旧的连接u点的方案数，再加上新的连接u点的方案数 
				if(!val[v]) q.push(v),val[v]=1/*,printf("v=%d\n",v)*/;
			}
		}
	}
}
int n,m,ui,vi,wi;
int main(){
	n=read(),m=read();
	for(int i=1;i<=m;i++){
		ui=read(),vi=read(),wi=read();
		add(vi,ui,wi);
	}
	memset(dis,0x7f7f7f7f,sizeof dis);
	spfa(n);
	if(dis[n]==0x7f7f7f7f){
		printf("No answer");
		return 0;
	}
	printf("%d %d",dis[1],vis[1][0]);
	return 0;
}