题意:
求有向图中最短路和比最短路大1的路径数量。
思路:
需要理解dijkstra算法中dis[n]数组的含义,设cnt[i]表示到点i的最短路径数量,cnt1[i]表示到点i比最短路大1的路径数量。在运行dijkstra算法的过程中每次获得最小dis[i]的时候可以对所有dis[v]+w(v,i)==dis[i]的v做如下更新cnt[i]+=cnt[v],cnt1[i]+=cnt1[v]。而当所有值为某数的dis[i]计算完成时也就是对任意i,dis[i]为同一值且不再变化时,可以对这些满足dis[i]+1=dis[v]+w(v,i)的点i做如下更新:cnt1[i]+=cnt[v];最后结果为cnt[f]+cnt1[f]。
//poj 3463
//sepNINE
#include <iostream>
#include <queue>
using namespace std;
const int maxN=1024;
const int maxM=10024;
int head[maxN];
int head1[maxN];
int dis[maxN];
int cnt[maxN];
int cnt1[maxN];
int n,m,e,e1,s,f;typedef pair<int,int> pii;
priority_queue<pii,vector<pii>,greater<pii> > q;struct Edge
{int v,w,next;
}edge[maxM],edge1[maxM];
void update(int maxDis)
{int i;for(int i=1;i<=n;++i){if(dis[i]==maxDis){for(int e=head1[i];e!=-1;e=edge1[e].next){int v=edge1[e].v,w=edge1[e].w;if(dis[v]+w==dis[i]+1)cnt1[i]+=cnt[v]; } }}}
int dij(int s,int t)
{ memset(cnt,0,sizeof(cnt));memset(cnt1,0,sizeof(cnt1));int i,maxDis;for(i=1;i<=n;++i)dis[i]=INT_MAX; dis[s]=0;cnt[s]=1;maxDis=0;q.push(make_pair(dis[s],s));while(!q.empty()){pii u=q.top();q.pop();int x=u.second;if(u.first>dis[x])continue;if(maxDis<dis[x]){update(maxDis); maxDis=dis[x];}for(int e=head1[x];e!=-1;e=edge1[e].next){int v=edge1[e].v,w=edge[e].w;if(dis[v]+w==dis[x]){cnt[x]+=cnt[v]; cnt1[x]+=cnt1[v];}}for(int e=head[x];e!=-1;e=edge[e].next)if(dis[x]+edge[e].w<dis[edge[e].v]){dis[edge[e].v]=dis[x]+edge[e].w;q.push(make_pair(dis[edge[e].v],edge[e].v));}}update(maxDis);return dis[t];
}int main()
{int cases;scanf("%d", &cases);while(cases--){memset(head,-1,sizeof(head));memset(head1,-1,sizeof(head1));e=e1=0;scanf("%d%d",&n,&m);while(m--){int a,b,c;scanf("%d%d%d",&a,&b,&c);edge[e].v=b;edge[e].w=c;edge[e].next=head[a];head[a]=e++;edge1[e1].v=a;edge1[e1].w=c;edge1[e1].next=head1[b];head1[b]=e1++;}scanf("%d%d",&s,&f);dij(s,f);printf("%d\n",cnt[f]+cnt1[f]);}return 0;
}
代码: