Description
n个视频,长度为k的缓存,每次询问,每个视频以pi的概率被选,如果不在缓存区则加入,如果缓存区满了,则最先进缓存的出来,问10^100次操作以后每个视频在缓存的概率
Input
第一行两个整数n和k,第二行n个数表示每个视频被选中的概率
(1<=k<=n<=20,0<=pi<=1,sum(pi)=1)
Output
输出10^100次操作后每个视频在缓存中出现的概率
Sample Input
3 1
0.3 0.2 0.5
Sample Output
0.3 0.2 0.5
10^100次方说明这个缓存肯定是填满了,那么也就是说只有最后几个操作可以影响到缓存中存在东西的概率 那么问题就转化成了装满缓存后每个物品存在的概率 就是说只要我加满了k 就停止。那么这个东西我们可以装压dp dp[i]表示i集合出现的概率。那么怎么转移呢?dp[i | 1 << j] = dp[i] * p[j] / sum (sum = sigma(p[k]), k不属于i, j属于i) 为什么呢 因为如果缓存中已经存在了某些东西,那么我们加进去是无效的,又因为我们加了很多次,那么这些无效的操作可以忽视(不是很懂,自己yy的)所以我们要除去选中缓存中已有物品的概率,剩下的概率就是从没选中的东西中选j的概率。那么这个dp就很好理解了。统计答案时如果这个集合的元素个数=k那么每个元素出现的概率加上这个dp值 还需要注意几点,1.sum不能很小; 2.如果p=0 那么这个物品肯定不会选中,所以我们先统计一下p不等于0的元素,和k比较一下谁更小,更小的作为k。
想想这还是我第一次打cf的题目呢,当时觉得这是什么东西,现在好像有了一些的进步。
#include<bits/stdc++.h> using namespace std; const int N = 23; const double eps = 1e-6; int n, k, all; double dp[1 << N], p[N], ans[N]; inline double getsum(int x) { double ret = 0; for(int i = 0; i < n; ++i) if(x & (1 << i)) ret += p[i]; return 1 - ret; } int main() { scanf("%d%d", &n, &k); all = 1 << n; int m = 0; for(int i = 0; i < n; ++i) { scanf("%lf", &p[i]); if(fabs(p[i]) > eps) ++m; } k = min(k, m); dp[0] = 1; for(int i = 0; i < all; ++i) { double sum = getsum(i); for(int j = 0; j < n; ++j) if(!(i & (1 << j))) dp[i | (1 << j)] += dp[i] * p[j] / sum; } for(int i = 0; i < all; ++i) if(__builtin_popcount(i) == k) for(int j = 0; j < n; ++j) if(i & (1 << j)) ans[j] += dp[i]; for(int i = 0; i < n; ++i) printf("%.7f ", ans[i]); return 0; }