19 瀋陽網絡賽 B. Dudu's maze （dfs 並查集）

傳送門

題意：n個點的無向圖，m條路徑 （n-1<m<=2*m）n個點有些存在糖果，一些存在怪獸，1號點肯定存在糖果，當你從1點開始，詢問找到最大糖果的期望，你很聰明並且知道地圖。當你第一次遇見怪獸，你有一次機會逃出該點。

思路：首先dfs所有和1點相連的點，記錄下和1點連通塊相連的怪獸的點，之後遍歷這些怪獸點，更新最大值。

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <map>
#include <iterator>
#include <vector>
#include <set>
#include <bitset>
#include <stack>
#define ull unsigned long long
#define Mod(x) (x+mod)%mod
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define mems(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const ll mod=1e9+7;;
const int N=1e5+5;
const double pi=acos(-1);
const int inf=0x3f3f3f3f;
const int M=50000+5;
int num;
int vi[N],vis[N],fi[N];
int ans[N],tot;
vector<int>ve[N];
void dfs(int u,int fa)
{
    vis[u]=1;
    num++;
    for(int i=0;i<ve[u].size();i++)
    {
        int v=ve[u][i];
        if(v==fa)
            continue;
        if(vi[v])
        {
            ans[tot++]=v;
            continue;
        }

        if(!vis[v])
        {
            dfs(v,u);
        }
    }
}
void dfs1(int u,int fa)
{
    if(vis[u]||vi[u])
        return;
    fi[u]=fa;
    num++;
    for(int i=0;i<ve[u].size();i++)
    {
        int v=ve[u][i];
        if(fi[v]==fa)
            continue;
        dfs1(v,fa);
    }
}
void init(int n)
{
    for(int i=0;i<=n;i++)
    {
        ve[i].clear();
        vi[i]=0;
        vis[i]=0;
        fi[i]=0;
    }
    num=0;
    tot=0;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        init(n);
        for(int i=1;i<=m;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            ve[a].push_back(b);
            ve[b].push_back(a);
        }
        for(int i=1;i<=k;i++)
        {
            int a;
            scanf("%d",&a);
            vi[a]=1;
        }
        dfs(1,-1);
        int ljt=num;
        double f=0.0;
        int v=0;
        for(int i=0;i<tot;i++)
        {
            num=0;
            for(int j=0;j<ve[ans[i]].size();j++)
            {
                dfs1(ve[ans[i]][j],++v);
//                printf("%d %d**\n",ve[ans[i]][j],num);
            }
            double ff=(double)num/ve[ans[i]].size();
//            printf("%d %d %d %.6f*\n",ans[i],num,ve[ans[i]].size(),ff);
            if(ff>f)
                f=ff;
        }
        printf("%.6f\n",ljt+f);
    }
}