AW383 观光

题目地址


易错点:

  • 预处理时cnt[s][0]应当初始化为1.

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXN=2e3,MAXM=2e4;
struct Edge{
	int from,to,w,nxt;
}e[MAXM];
int head[MAXN],edgeCnt=1;
void addEdge(int u,int v,int w){
	e[++edgeCnt].from=u;
	e[edgeCnt].to=v;
	e[edgeCnt].w=w;
	e[edgeCnt].nxt=head[u];
	head[u]=edgeCnt;
}
struct Node{
	int v,dis;
	bool operator <(Node another)const{
		return dis>another.dis;
	}
};
int s,t;
int dis[MAXN][2],cnt[MAXN][2];
int dijkstra(){
	priority_queue<Node> q;
	dis[s][0]=dis[s][1]=0;
	cnt[s][0]=1;//
	q.push(Node{s,0});
	while(!q.empty()){
		Node nowNode=q.top();q.pop();
		int nowU=nowNode.v;
		int k=-1;
		if(dis[nowU][0]==nowNode.dis)k=0;
		else if(dis[nowU][1]==nowNode.dis)k=1;
		if(k==-1)continue;
		for(int i=head[nowU];i;i=e[i].nxt){
			int nowV=e[i].to,nowDis=nowNode.dis+e[i].w;
			if(dis[nowV][0]>nowDis){
				dis[nowV][1]=dis[nowV][0];
				cnt[nowV][1]=cnt[nowV][0];
				dis[nowV][0]=nowDis;
				cnt[nowV][0]=cnt[nowU][0];
				q.push(Node{nowV,nowDis});
			}else if(dis[nowV][0]==nowDis){
				cnt[nowV][0]+=cnt[nowU][0];
			}else if(dis[nowV][1]>nowDis){
				dis[nowV][1]=nowDis;
				cnt[nowV][1]=cnt[nowU][k];
				q.push(Node{nowV,nowDis});
			}else if(dis[nowV][1]==nowDis){
				cnt[nowV][1]+=cnt[nowU][k];
			}
		}
	}
	if(dis[t][0]==dis[t][1]-1)return cnt[t][0]+cnt[t][1];
	return cnt[t][0];
}
void init(){
	s=0,t=0;
	memset(dis,0x3f,sizeof(dis));
	memset(cnt,0,sizeof(cnt));
	memset(head,0,sizeof(head));
	edgeCnt=1;
}
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		init();
		int n,m;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=m;i++){
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			addEdge(u,v,w);
		}
		scanf("%d%d",&s,&t);
		int ans=dijkstra();
		printf("%d\n",ans);
	}
	return 0;
} 

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值