本文来自: http://blog.csdn.net/napoleon_acm/article/details/24583767
题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=1688
题目: 给定一有向图 点和边个数 2 ≤ N ≤ 1,000 and 1 ≤ M ≤ 10, 000, 给定 起点和终点,求 最短路 和比最短路大1个单位 的路径的 路径数之和。
首先要会Dijkstra 求最短路, 之后修改一下就可以了。
dis[i][2],dp[i][2] 分别表示 起点到i点的 最短 及次短路径长度,以及方案数。
求最短路是dijkstra 算法 是循环n次 一次更新每个点的最短距离, 现在求最短以及次短就要 循环2*n次。
对于每次更新:
1.新路径比最短路径长度要小,那么最短路和次短路的长度和次数都要更新。
2.新路径等于最短路的长度,那么只需要更新最短路的条数。
3.新路径比最短路要长但是比次短路要短,那么需要更新次短路的长度和条数。
4.新路径等于次短路径的长度,那么只需要更新次短路径的条数。
之后判断终点的 次短路是否比最短路大1就可以了。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define INF 1000000000
#define N 1005
int t,tot,hh[N],n,m,dp[N][3],dis[N][3],S,T,vis[N][3];
struct node
{
int v,next,w;
}edge[100000];
void init()
{
memset(hh,-1,sizeof(hh));
tot=0;
}
void add(int u,int v,int w)
{
edge[tot].v=v; edge[tot].w=w;
edge[tot].next=hh[u]; hh[u]=tot++;
}
void dijk()
{
memset(dp,0,sizeof(dp));
for(int i=0;i<=n;i++) dis[i][0]=dis[i][1]=INF;
memset(vis,0,sizeof(vis));
dp[S][1]=1;
dis[S][1]=0; //不要先进行对起点的更新,因为可能有重边
for(int q=1;q<=2*n;q++)
{
int ma,ad,type;
ma=INF; ad=-1; type=-1;
for(int i=1;i<=n;i++)
{
if(!vis[i][1] && ma>dis[i][1])
{
ad=i; ma=dis[i][1]; type=1;
}
else if(!vis[i][0] && ma>dis[i][0])
{
ad=i; ma=dis[i][0]; type=0;
}
}
if(ad==-1) break;
vis[ad][type]=1;
for(int i=hh[ad];i!=-1;i=edge[i].next)
{
int cost=edge[i].w;
int vv=edge[i].v;
if(dis[vv][1]>ma+cost)
{
dis[vv][0]=dis[vv][1];
dp[vv][0]=dp[vv][1];
dis[vv][1]=ma+cost;
dp[vv][1]=dp[ad][type];
}
else if(dis[vv][1]==ma+cost)
dp[vv][1]+=dp[ad][type];
else if(dis[vv][0]>ma+cost)
{
dis[vv][0]=ma+cost;
dp[vv][0]=dp[ad][type];
}
else if(dis[vv][0]==ma+cost)
dp[vv][0]+=dp[ad][type];
}
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen ("in.txt" , "r" , stdin);
#endif
scanf("%d",&t);
while(t--)
{
init();
scanf("%d%d",&n,&m);
int u,v,w;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
}
scanf("%d%d",&S,&T);
dijk();
int sum=dp[T][1];
if(dis[T][1]+1==dis[T][0]) sum+=dp[T][0];
printf("%d\n",sum);
}
return 0;
}