【BZOJ1415】 [Noi2005]聪聪和可可 概率与期望

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #define N 1001
 5 using namespace std;
 6 double f[N][N];
 7 int n,m,cnt,a,b;
 8 struct E{int next,to;}e[2*N+10];
 9 int head[N],ds[N],q[N],dis[N][N],p[N][N];
10 inline int read()
11 {
12     int ans=0,f=1;
13     char c;
14     while (!isdigit(c=getchar())) if (c==‘-‘) f=-1;
15     ans=c-‘0‘;
16     while (isdigit(c=getchar())) ans=ans*10+c-‘0‘;
17     return ans*f;
18 }
19 void insert(int u,int v)
20 {
21     cnt++; e[cnt].next=head[u]; head[u]=cnt; e[cnt].to=v; ds[u]++;
22     cnt++; e[cnt].next=head[v]; head[v]=cnt; e[cnt].to=u; ds[v]++;
23 }
24 double DP(int x,int y)
25 {
26     if (f[x][y]) return f[x][y];
27     if (x==y) return 0;
28     if (p[x][y]==y || p[p[x][y]][y]==y) return f[x][y]=1;
29     double tot=DP(p[p[x][y]][y],y);
30     for (int i=head[y];i;i=e[i].next)
31         tot+=DP(p[p[x][y]][y],e[i].to);
32     return f[x][y]=tot/(ds[y]+1)+1; 
33 }
34 void Bfs(int x)
35 {
36     int t=0,w=1;
37     q[t]=x;
38     dis[x][x]=0;    
39     while (t!=w)
40     {
41         int now=q[t],tmp=p[x][now];
42         t++; if (t==1001) t=0;
43         for (int i=head[now];i;i=e[i].next)
44             if (dis[x][e[i].to]==-1 || (dis[x][e[i].to]==dis[x][now]+1 && tmp<p[x][e[i].to]))
45             {
46                 dis[x][e[i].to]=dis[x][now]+1;
47                 p[x][e[i].to]=tmp;
48                 if (!tmp) p[x][e[i].to]=e[i].to;
49                 q[w]=e[i].to;
50                 w++; if (w==1001) w=0;
51             }
52     }
53 } 
54 int main()
55 {
56     memset(dis,-1,sizeof(dis));
57     n=read(); m=read(); a=read(); b=read();
58     for (int i=1;i<=m;i++)
59     {
60         int u,v;
61         u=read(); v=read();
62         insert(u,v);
63     }
64     for (int i=1;i<=n;i++) Bfs(i);
65     printf("%.3lf",DP(a,b));
66     return 0;
67 }