【BZOJ 3036】 3036: 绿豆蛙的归宿 （概率DP）

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 using namespace std;
 7 #define Maxn 100010
 8 #define Maxm 200010
 9 
10 struct node
11 {
12     int x,y,c,next;
13 }t[Maxm],tt[Maxm];
14 int len,first[Maxn],d[Maxn];
15 
16 void ins(int x,int y,int c)
17 {
18     d[x]++;
19     t[++len].x=x;t[len].y=y;t[len].c=c;
20     t[len].next=first[x];first[x]=len;
21 }
22 int ln,ft[Maxn];
23 void INS(int x,int y){tt[++ln].x=x;tt[ln].y=y;tt[ln].next=ft[x];ft[x]=ln;}
24 
25 bool vis[Maxn];
26 double f[Maxn],g[Maxn];
27 
28 void dfs(int x)
29 {
30     if(vis[x]) return;vis[x]=1;
31     for(int i=ft[x];i;i=tt[i].next)
32     {
33         int y=tt[i].y;
34         dfs(y);
35         g[x]+=g[y]*1.0/d[y];
36     }
37 }
38 
39 int n,m;
40 double ffind(int x)
41 {
42     if(x==n) return f[x]=0;
43     if(vis[x]) return f[x];vis[x]=1;
44     for(int i=first[x];i;i=t[i].next)
45     {
46         int y=t[i].y;
47         ffind(y);
48         if(f[y]==0&&y!=n) continue;
49         f[x]+=(f[y]+t[i].c)*1.0/d[x];
50     }
51 }
52 
53 int main()
54 {
55     scanf("%d%d",&n,&m);
56     len=0;ln=0;
57     memset(first,0,sizeof(first));
58     memset(ft,0,sizeof(ft));
59     memset(d,0,sizeof(d));
60     for(int i=1;i<=m;i++)
61     {
62         int x,y,c;
63         scanf("%d%d%d",&x,&y,&c);
64         ins(x,y,c);INS(y,x);
65     }
66     g[1]=1;
67     memset(vis,0,sizeof(vis));vis[1]=1;
68     dfs(n);
69     memset(vis,0,sizeof(vis));
70     ffind(1);
71     printf("%.2lf\n",f[1]/g[n]);
72     return 0;
73 }

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 using namespace std;
 7 #define Maxn 100010
 8 #define Maxm 200010
 9 
10 struct node
11 {
12     int x,y,c,next;
13 }t[Maxm],tt[Maxm];
14 int len,first[Maxn],d[Maxn];
15 
16 void ins(int x,int y,int c)
17 {
18     d[x]++;
19     t[++len].x=x;t[len].y=y;t[len].c=c;
20     t[len].next=first[x];first[x]=len;
21 }
22 
23 bool vis[Maxn];
24 double f[Maxn];
25 
26 int n,m;
27 double ffind(int x)
28 {
29     if(vis[x]) return f[x];vis[x]=1;
30     for(int i=first[x];i;i=t[i].next)
31     {
32         int y=t[i].y;
33         ffind(y);
34         f[x]+=(f[y]+t[i].c)*1.0/d[x];
35     }
36 }
37 
38 int main()
39 {
40     scanf("%d%d",&n,&m);
41     len=0;
42     memset(first,0,sizeof(first));
43     for(int i=1;i<=m;i++)
44     {
45         int x,y,c;
46         scanf("%d%d%d",&x,&y,&c);
47         ins(x,y,c);
48     }
49     memset(vis,0,sizeof(vis));
50     ffind(1);
51     printf("%.2lf\n",f[1]);
52     return 0;
53 }