2019icpc沈阳网络赛B. Dudu's maze

https://nanti.jisuanke.com/t/41402
题意：有一张图，有的点有一个糖果，其他点有怪兽。一个人从1出发，若走到安全点，则可以自己选择一个相邻点继续走，若第1次走到怪兽点，则强制随机走到一个相邻点，第二次走到怪兽点或者随时可以结束游戏，求得到糖果数的最大期望。
思路：将怪兽点割掉，并查集求连通块，每到达一个连通块的任意一点，那么这个联通块的所有点都可以不经过怪兽而到达。因为只能到达一次怪兽，那么dfs求出所有可能的第一次到达的怪兽，分别算期望，最大的一个就是答案。注意从怪兽点走到相邻安全点/怪兽点/1所在连通块这3种情形。注意图有环，需要vis数组记录。

#include<bits/stdc++.h>
using namespace std;
const int maxn=100000+100;

int T,n,m,k,is[maxn],flag[maxn],p[maxn],sz[maxn];
vector<int> G[maxn];
bool vis[maxn];
double maxx;

void dfs(int u,int fa)
{
    if(is[u]){flag[u]=1;return;}
    if(vis[u])return;
    vis[u]=1;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa)continue;
        dfs(v,u);
    }
    //vis[u]=0;
}

int findset(int x){return (x==p[x])?x:(p[x]=findset(p[x]));}

int main()
{
    cin>>T;
    while(T--)
    {
        int a,b;
        cin>>n>>m>>k;
        memset(is,0,sizeof(is));
        memset(flag,0,sizeof(flag));
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)p[i]=i,sz[i]=1;
        for(int i=1;i<=n;i++)G[i].clear();
        for(int i=1;i<=m;i++)scanf("%d%d",&a,&b),G[a].push_back(b),G[b].push_back(a);
        for(int i=1;i<=k;i++)scanf("%d",&a),is[a]=1;
        dfs(1,0);
        for(int i=1;i<=n;i++)for(int j=0;j<G[i].size();j++)
        {
            int v=G[i][j];
            if(is[i]||is[v])continue;
            int x=findset(i),y=findset(v);
            if(x!=y)
            {
                p[x]=y;
                sz[y]+=sz[x];
            }
        }
        maxx=sz[findset(1)];
        for(int i=1;i<=n;i++)if(flag[i])
        {
            double sum=0;
            for(int j=0;j<G[i].size();j++)
            {
                if(is[G[i][j]])sum+=sz[findset(1)];
                else if(findset(G[i][j])!=findset(1))
					sum+=sz[findset(G[i][j])]+sz[findset(1)];
                else sum+=sz[findset(1)];
            }
				
            if (G[i].size()) sum/=G[i].size();
            maxx=max(maxx,sum);
        }
        printf("%.8f\n",maxx);
    }
    return 0;
}