hdu5036(概率dp+bitset优化)

链接：点击打开链接

题意：n个房间，每个房间都有若干个钥匙打开其他的门，如果手上没有钥匙可以选择炸开一个门，求打开所有门的情况下炸弹的期望

代码：

#include <bitset>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
bitset<1005> s[1005];
int main(){
    double ans;
    int t,n,i,j,v,num,sum,cas;
    scanf("%d",&t);
    for(cas=1;cas<=t;cas++){                    //用期望的可加性，因此只需对每一个房间
        scanf("%d",&n);                         //单独考虑，假如有k个房间可到一个房间，
        for(i=0;i<n;i++)                        //则期望加1/k
        s[i].reset();
        for(i=0;i<n;i++){
            s[i][i]=1;
            scanf("%d",&num);
            while(num--){
                scanf("%d",&v);
                s[i][v-1]=1;
            }
        }                                       //用bitset优化，并表示可达房间
        for(i=0;i<n;i++)
        for(j=0;j<n;j++)
        if(s[j][i])
        s[j]|=s[i];                             //求传递闭包求出所有可达房间
        ans=0;
        for(i=0;i<n;i++){
            sum=0;
            for(j=0;j<n;j++)
            if(s[j][i])
            sum++;
            ans+=(1.0/sum);
        }
        printf("Case #%d: %.5lf\n",cas,ans);
    }
    return 0;
}