题目
思路
考场想用dijkstra求最短路得方法算出最大概率,但是错了:因为如果有两条路可以到达该点,那么到达这个点的概率应该是走这两条路的概率之和,而不是选择概率大的一条路。所以T1只有5分qwq
正解
直接暴力dfs算概率,最后把概率排序增加陷阱
代码
#include <bits/stdc++.h>
using namespace std;
#define N 100005
#define db double
int n,m,p,s,t,tot;
int v[N],head[N],to[N],Next[N],du[N];
db val[N],d[N],ans;
bool vis[N];
inline int read(){
int x=0,f=0; char ch=getchar();
while(ch<'0'||ch>'9') f|=ch=='-',ch=getchar();
while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return f?-x:x;
}
inline void add(int x,int y,db z){
to[++tot] = y;
val[tot] = z;
Next[tot] = head[x];
head[x] = tot;
++du[y];
}
inline void Dfs(int x,db last){
if(x != s) d[x] += last;
ans += (v[x]*last);
for(int i=head[x]; i; i=Next[i]){
int y = to[i];
double z = val[i];
Dfs(y,last*z);
}
}
int main()
{
// freopen("trap.in","r",stdin);
// freopen("trap.out","w",stdout);
n=read(); m=read(); p=read(); s=read(); t=read();
for(int i=1; i<=n; i++)
v[i] = read();
for(int i=1,ai,bi; i<=m; i++){
db ci;
ai=read();bi=read();scanf("%lf",&ci);
add(ai,bi,ci);
}
Dfs(s,1);
sort(d+1,d+1+n);
for(int i=n; i>=n-t+1; i--)
ans += (d[i]*p);
printf("%.3lf",ans);