传送门:观光
思路:这里因为是有边权的最短路,使用堆优化的dijkstra,保证每个顶点的次短路都只会被入队出队一次,又因为每次取出来的都是dist最小的一个,具有拓扑性质。
代码:
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
const int N=2e4+10;
int n,m,s,t;
bool st[N][2];
int dist[N][2],cnt[N][2];
int idx,e[N*2],ne[N*2],h[N],w[2*N];
struct Ver
{
int id,type,dist;
bool operator> (const Ver &W) const
{
return dist > W.dist;
}
};
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dijkstra()
{
memset(dist,0x3f,sizeof dist);
memset(st,0,sizeof st);
memset(cnt,0,sizeof cnt);
dist[s][0]=0;
cnt[s][0]=1;
priority_queue<Ver, vector<Ver>, greater<Ver>> heap;
heap.push({s,0,0});
while(heap.size())
{
Ver p=heap.top();
heap.pop();
int ver=p.id,type=p.type,distance=p.dist,cont=cnt[ver][type];
if(st[ver][type]) continue;
st[ver][type]=true;
for(int i=h[ver];i!=-1;i=ne[i])
{
int j=e[i];
if(dist[j][0]>distance+w[i])
{
dist[j][1]=dist[j][0],cnt[j][1]=cnt[j][0];
heap.push({j,1,dist[j][1]});
dist[j][0]=distance+w[i],cnt[j][0]=cont;
heap.push({j,0,dist[j][0]});
}else if(dist[j][0]==distance+w[i]) cnt[j][0]+=cont;
else if(dist[j][1]>distance+w[i])
{
dist[j][1]=distance+w[i], cnt[j][1]=cont;
heap.push({j, 1, dist[j][1]});
}else if(dist[j][1]==distance+w[i]) cnt[j][1]+=cont;
}
}
int res=cnt[t][0];
if(dist[t][0]+1==dist[t][1]) res+=cnt[t][1];
return res;
}
int main()
{
int T;
cin>>T;
while(T--)
{
cin>>n>>m;
memset(h,-1,sizeof h);
idx=0;
while(m--)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
}
cin>>s>>t;
cout<<dijkstra()<<endl;
}
return 0;
}