n个数,长度为k的缓存,每次询问,每个数以pi的概率被选,如果不在缓存区则加入,如果缓存区满了,则第一个进缓存的出来,问10^100次询问以后每个数在缓存的概率
题解
把问题倒过来想。假设从最晚访问的元素开始向前加元素,加到k个为止。求第i个出现的概率。这样就不同考虑从缓存区有文件出来了。
将20个元素压位,dp[mask]记录mask状态(由 0/1 表示的状态)出现的可能性。当1<< i在mask中为0,也就是第i个元素没有出现过时,dp[mask]可以转移到dp[mask+(1<< i)],表示第i个元素在mask中所有已存在元素之前加入,转移时要乘上概率p(i)/(p(i)+p(x)),其中p(x)是除了mask中元素和i元素的其他元素出现的概率,这个式子的含义是在除了已经在mask中出现的元素中,i第一个出现,所以用i的概率除以还没被选入mask中元素的概率和。最后答案只要统计一下有k个元素且第i个元素存在的状态的概率和。
要注意的是如果有概率为0的要直接去掉,因为这个元素根本不能进缓存,如果没有去掉,答案肯定是会出问题的。而且去掉这些元素后,要保证k<=n
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define L(i) i<<1
#define R(i) i<<1|1
#define INF 0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-9
#define maxn 10000100
#define MOD 1000000007
int n,k;
double f[1<<21],ans[22];
double a[22];
int main()
{
int t;
while(scanf("%d%d",&n,&k) != EOF)
{
int m = n;
for(int i = 0; i < n; i++)
{
scanf("%lf",&a[i]);
if(a[i] < eps)
m--;
}
k = min(k,m);
memset(ans,0,sizeof(ans));
memset(f,0,sizeof(f));
f[0] = 1;
for(int i = 0; i < (1<<n); i++)
{
int num = 0;
double cnt = 0;
for(int j = 0; j < n; j++)
{
if(i & (1<<j))
num++;
else
cnt += a[j];
}
if(num > k)
continue;
if(num == k)
{
for(int j = 0; j < n; j++)
if(i & (1<<j))
ans[j] += f[i];
continue;
}
for(int j = 0; j < n; j++)
{
if(i & (1<<j))
continue;
f[i|(1<<j)] += f[i] * a[j] / cnt;
}
}
for(int i = 0; i < n; i++)
printf("%f ",ans[i]);
printf("\n");
}
return 0;
}