比赛 |
9.27练习赛 |
评测结果 |
AAAAAAAAAAAT |
题目名称 |
观光 |
最终得分 |
92 |
用户昵称 |
袁书杰 |
运行时间 |
3.159 s |
代码语言 |
C++ |
内存使用 |
51.10 MiB |
提交时间 |
2024-09-27 21:39:01 |
显示代码纯文本
#include<bits/stdc++.h>
#define int long long
using namespace std;
struct edge{
int u,v,w,nxt;
}e[1000005];
int etot,head[1000005],dis[1000005],now,ans;
bool vis[1000005],flag[1000005];
void adde(int u,int v,int w){
e[++etot]={u,v,w,head[u]};
head[u]=etot;
}
struct node{
int u,dis;
bool operator<(const node&a)const{
return a.dis<dis;
}
};
priority_queue<node> q;
void dij(int s){
for(int i=0; i<=1000000; i++) {
dis[i]=5e18;
}
q.push(node{s,0});
dis[s]=0;
while(!q.empty()){
int u=q.top().u;
q.pop();
if(vis[u]){
continue;
}
vis[u]=1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].v,w=e[i].w;
if(dis[v]>dis[u]+w){
dis[v]=dis[u]+w;
q.push(node{v,dis[v]});
}
}
}
}
int n,m,s,t;
void dfs(int u,int father,int len){
if(u==t){
if(len==now||len==now-1){
ans++;
return;
}
}
if(len>now){
return;
}
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].v;
if(!flag[v]){
flag[u]=true;
dfs(v,u,len+e[i].w);
flag[u]=false;
}
}
}
signed main(){
freopen("sightseeing.in","r",stdin);
freopen("sightseeing.out","w",stdout);
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
int T;
cin>>T;
while(T--){
cin>>n>>m;
memset(vis,0,sizeof(vis));
memset(flag,0,sizeof(flag));
memset(head,0,sizeof(head));
etot=0;
now=0;
ans=0;
memset(e,0,sizeof(e));
for(int i=1;i<=m;i++){
int u,v,w;
cin>>u>>v>>w;
adde(u,v,w);
}
cin>>s>>t;
dij(s);
now=dis[t]+1;
dfs(s,0,0);
cout<<ans<<'\n';
}
return 0;
}