题意:给出N个不同的串,长度一样,别人等概率选一个,你要询问到能确定那个串才能停止,问询问的期望。
这题我一开始的做法是枚举那个选定的串,然后还需要枚举某个状态S能不能选出这个串,之后dp[s]由所有没有判断出来的dp[s^(1<<i)]转移过来,再计算。这样复杂度高达(n^2m2^m)。果然超时。
绞尽脑汁没想出优化方法,仔细解读了下CF题解。
首先用一个数组a[s]代表询问的状态为s不能分辨的串(用二进制表示),这个如何递推?枚举i和j,找2个相同串位置的的二进制集合s,然后a[s] |= (1<<i) | (1<<j),但是这样仅仅只是相同状态刚刚好为s不能分辨的串,比如baaa,caaa,在a[0111]里,但是也同样应该在a[0110]或者a[0101]等等子集里,所以再来一次逆推,让子集包含本集。
dp[s]代表状态为s到达的概率,对于这个状态s,假设a[s]里有k个1代表有k个串不能分辨,那么对于询问这k个串其中一个,询问期望就是dp[s]×1,其他0的位置就是已经不需要询问了,1的位置这次询问是这次必须的,之后还询问不询问不知道,还询问不询问是后面的状态所决定的,对于k个询问次数期望就是dp[s]*k,对于最后的答案自然是加dp[s]*k/n。其实主要思想就是利用期望的可加性,分开来来算。
AC代码:
//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<ctype.h>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdlib>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#include<ctime>
#include<string.h>
#include<string>
#include<sstream>
#include<bitset>
using namespace std;
#define ll long long
#define ull unsigned long long
#define eps 1e-11
#define NMAX 200005
#define MOD 1000000007
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI acos(-1)
template<class T>
inline void scan_d(T &ret)
{
char c;
int flag = 0;
ret=0;
while(((c=getchar())<'0'||c>'9')&&c!='-');
if(c == '-')
{
flag = 1;
c = getchar();
}
while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
if(flag) ret = -ret;
}
char ch[55][25];
double dp[(1<<20)+5];
ll a[(1<<20)+5];
int main()
{
#ifdef GLQ
freopen("input.txt","r",stdin);
// freopen("o4.txt","w",stdout);
#endif // GLQ
int n;
scanf("%d",&n);
for(int i = 0; i < n; i++)
scanf("%s",ch[i]);
int len = strlen(ch[0]);
double ans = 0;
for(int i = 0; i < n; i++)
for(int j = i+1; j < n; j++)
{
int tmp = 0;
for(int k = 0; k < len; k++) if(ch[i][k] == ch[j][k])
tmp |= 1<<k;
a[tmp] |= (1LL<<i)|(1LL<<j);
}
for(int s = (1<<len)-1; s >= 0; s--)
for(int i = 0; i < len; i++) if(s&(1<<i))
a[s^(1<<i)] |= a[s];
dp[0] = 1;
for(int s = 0; s < (1<<len); s++)
{
int nct = 0,gg = 0;
for(int i = 0; i < len; i++) if(s&(1<<i)) nct++;
for(int i = 0; i < len; i++) if(!(s&(1<<i)))
dp[s^(1<<i)] += dp[s]/(double)(len-nct);
for(int i = 0; i < n; i++) if((a[s]&(1LL<<i))) gg++;
ans += dp[s]*(double)gg/(double)n;
}
printf("%.10lf\n",ans);
return 0;
}