The Preliminary Contest for ICPC Asia Shenyang 2019 B. Dudu's maze

题意：一张图，n个点，m条边，n个点中有k个怪兽节点，除了怪兽节点其他节点都有糖果，从1号点出发，遇到怪兽点就得停下，但你有一次使用传送门的机会，传送门会等概率的将你传送到这个点相连的任意一个点中去，问你最大糖果数的期望值是多少？

思路：幷查集找连通块，再找出所有与1号节点所在连通块相连的怪兽节点，枚举这些怪兽节点，找出收益最大的那个

代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=1e5+9;
struct Edge{
    int to,next;
}edge[maxn*4];
struct node{
    int x,y;
}ed[maxn*2];
int cnt=0,f[maxn],head[maxn],sz[maxn],vis[maxn],flag[maxn],book[maxn];
vector<int>vec;
inline void add(int u,int v){
    edge[cnt].next=head[u];
    edge[cnt].to=v;
    head[u]=cnt++;
}
inline int Find(int v){
    if(f[v]==v)return v;
    f[v]=Find(f[v]);
    return f[v];
}
void Merge(int u,int v){
    int t1=Find(u),t2=Find(v);
    if(t1!=t2){
        if(t1>t2)swap(t1,t2);
        f[t2]=t1;
        sz[t1]+=sz[t2];
        sz[t2]=0;
    }
}

int main(){
    int i,j,k,n,m,t;
    scanf("%d",&t);
    while(t--){
        vec.clear();
        cnt=0;
        memset(vis,0,sizeof(vis));
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=n;i++){
            sz[i]=1;
            f[i]=i;
            flag[i]=0;
            head[i]=-1;
        }
        int x,y;
        for(i=0;i<m;i++){
            scanf("%d%d",&x,&y);
            ed[i].x=x;
            ed[i].y=y;
                add(x,y);add(y,x);
          //  }
        }
        for(i=0;i<k;i++){
            scanf("%d",&x);
            flag[x]=1;
        }
        for(i=0;i<m;i++){
            int x=ed[i].x,y=ed[i].y;
            if(!flag[x]&&!flag[y])Merge(x,y);
        }
        double ans=sz[1];
        for(i=1;i<=n;i++){
            if(flag[i]){
                for(j=head[i];j!=-1;j=edge[j].next){
                    int v=edge[j].to;
                    if(Find(v)==1){
                        vec.push_back(i);
                        break;
                    }
                }
            }
        }
     //   cout<<ans<<endl;
        for(i=0;i<vec.size();i++){
            int u=vec[i];
            double sum=0;
            int et=0;
            for(j=head[u];j!=-1;j=edge[j].next){
                int v=edge[j].to;
                et++;
                if(Find(v)!=1){
                    if(!flag[v])
                    sum+=sz[Find(v)];
                }
            }
//            cout<<sum<<' '<<et<<' '<<sum/et<<endl;
            ans=max(ans,sz[1]+sum/et);
        }
        printf("%.7f\n",ans);
    }
}