【BZOJ 1415】 1415: [Noi2005]聪聪和可可 （bfs+记忆化搜索+期望）

1415: [Noi2005]聪聪和可可

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1640  Solved: 962

Description

Input

数据的第1行为两个整数N和E，以空格分隔，分别表示森林中的景点数和连接相邻景点的路的条数。 第2行包含两个整数C和M，以空格分隔，分别表示初始时聪聪和可可所在的景点的编号。 接下来E行，每行两个整数，第i+2行的两个整数Ai和Bi表示景点Ai和景点Bi之间有一条路。 所有的路都是无向的，即：如果能从A走到B，就可以从B走到A。 输入保证任何两个景点之间不会有多于一条路直接相连，且聪聪和可可之间必有路直接或间接的相连。

Output

输出1个实数，四舍五入保留三位小数，表示平均多少个时间单位后聪聪会把可可吃掉。

Sample Input

【输入样例1】
4 3
1 4
1 2
2 3
3 4
【输入样例2】
9 9
9 3
1 2
2 3
3 4
4 5
3 6
4 6
4 7
7 8
8 9

Sample Output

【输出样例1】
1.500
【输出样例2】
2.167

HINT

【样例说明1】
开始时，聪聪和可可分别在景点1和景点4。
第一个时刻，聪聪先走，她向更靠近可可(景点4)的景点走动，走到景点2，然后走到景点3；假定忽略走路所花时间。
可可后走，有两种可能：
第一种是走到景点3，这样聪聪和可可到达同一个景点，可可被吃掉，步数为1，概率为 。
第二种是停在景点4，不被吃掉。概率为 。
到第二个时刻，聪聪向更靠近可可(景点4)的景点走动，只需要走一步即和可可在同一景点。因此这种情况下聪聪会在两步吃掉可可。
所以平均的步数是1* +2* =1.5步。

对于所有的数据，1≤N,E≤1000。
对于50%的数据，1≤N≤50。

Source

【分析】

　　先n次bfs求出聪聪会走哪里。

　　然后直接记忆化搜索求期望就好了。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<queue>
 using namespace std;
 #define Maxn 1010
 #define Maxm 1010

 int mymin(int x,int y) {return x<y?x:y;}

 struct node
 {
     int x,y,next;
 }t[Maxm*];
 int first[Maxn],len;
 int d[Maxn];

 void ins(int x,int y)
 {
     t[++len].x=x;t[len].y=y;
     t[len].next=first[x];first[x]=len;
     d[x]++;
 }

 int n,m;
 int dis[Maxn][Maxn],pre[Maxn],wk[Maxn][Maxn];
 queue<int > q;
 void bfs(int nw)
 {
     for(int i=;i<=n;i++) dis[nw][i]=-;
     while(!q.empty()) q.pop();
     dis[nw][nw]=;q.push(nw);
     while(!q.empty())
     {
         int x=q.front();q.pop();
         for(int i=first[x];i;i=t[i].next)
         {
             int y=t[i].y;
             if(dis[nw][y]!=-&&dis[nw][x]+==dis[nw][y]) pre[y]=mymin(pre[y],x);
             else if(dis[nw][y]==-)
             {
                 dis[nw][y]=dis[nw][x]+;
                 pre[y]=x;
                 q.push(y);
             }
         }
     }
     for(int i=;i<=n;i++) wk[i][nw]=pre[i];
 }

 double f[Maxn][Maxn];
 bool vis[Maxn][Maxn];

 double ffind(int st,int ed)
 {
     if(st==ed) return f[st][ed]=;
     if(dis[st][ed]<=) return f[st][ed]=1.0;
     if(vis[st][ed]) return f[st][ed];
     f[st][ed]=;vis[st][ed]=;
     int to=wk[wk[st][ed]][ed];
     for(int i=first[ed];i;i=t[i].next)
     {
         int y=t[i].y;
         f[st][ed]+=(ffind(to,y)+)*1.0/(d[ed]+);
     }
     f[st][ed]+=(ffind(to,ed)+)*1.0/(d[ed]+);
     return f[st][ed];
 }

 int main()
 {
     int st,ed;
     scanf("%d%d%d%d",&n,&m,&st,&ed);
     len=;
     memset(first,,sizeof(first));
     for(int i=;i<=m;i++)
     {
         int x,y;
         scanf("%d%d",&x,&y);
         ins(x,y);ins(y,x);
     }
     for(int i=;i<=n;i++) bfs(i);
     memset(vis,,sizeof(vis));
     ffind(st,ed);
     printf("%.3lf\n",f[st][ed]);
     return ;
 }

2017-04-22 08:51:55