首页 > 代码库 > zoj 3640 Help Me Escape (概率dp 递归求期望)
zoj 3640 Help Me Escape (概率dp 递归求期望)
题目链接
Background
If thou doest well, shalt thou not be accepted? and if thou doest not well, sin lieth at the door. And unto thee shall be his desire, and thou shalt rule over him.
And Cain talked with Abel his brother: and it came to pass, when they were in the field, that Cain rose up against Abel his brother, and slew him.
And the LORD said unto Cain, Where is Abel thy brother? And he said, I know not: Am I my brother‘s keeper?
And he said, What hast thou done? the voice of thy brother‘s blood crieth unto me from the ground.
And now art thou cursed from the earth, which hath opened her mouth to receive thy brother‘s blood from thy hand;
When thou tillest the ground, it shall not henceforth yield unto thee her strength; a fugitive and a vagabond shalt thou be in the earth.
—— Bible Chapter 4
Now Cain is unexpectedly trapped in a cave with N paths. Due to LORD‘s punishment, all the paths are zigzag and dangerous. The difficulty of the ith path is ci.
Then we define f as the fighting capacity of Cain. Every day, Cain will be sent to one of the N paths randomly.
Suppose Cain is in front of the ith path. He can successfully take ti days to escape from the cave as long as his fighting capacity f is larger than ci. Otherwise, he has to keep trying day after day. However, if Cain failed to escape, his fighting capacity would increase ci as the result of actual combat. (A kindly reminder: Cain will never died.)
As for ti, we can easily draw a conclusion that ti is closely related to ci. Let‘s use the following function to describe their relationship:
After D days, Cain finally escapes from the cave. Please output the expectation of D.
Input
The input consists of several cases. In each case, two positive integers N and f (n ≤ 100, f ≤ 10000) are given in the first line. The second line includes N positive integers ci (ci ≤ 10000, 1 ≤ i ≤ N)
Output
For each case, you should output the expectation(3 digits after the decimal point).
Sample Input
3 11 2 3
Sample Output
6.889
题意:
有一个吸血鬼被困了,有n条路可以逃出去,每条路有一个难度c[],他初始的战斗力是f,对于第i条路,若f > c[i]他花t[i]天就能出去,否则,他就停留一天,同时战斗力增加c[i]然后再选一条路走出去,他走每条路的概率是相同的。问他逃出去的天数的期望。
注意t[i]是整数。
分析:
d[i]表示战斗力为 i 的时候的逃出去的期望。
用递归的好处是思路比较清楚,就是每次按照概率加 什么时候能逃出去的期望。
这个题也有逆推 递推的做法。
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <cstdlib> 5 #include <queue> 6 #include <cmath> 7 #include <algorithm> 8 #define LL __int64 9 const int maxn = 100+10;10 using namespace std;11 double d[20000+10], t[maxn]; //注意数组开到了2倍,因为终态可能是20000.12 int c[maxn], n;13 14 double dfs(int f)15 {16 if(d[f]>0) return d[f]; //相当于剪枝,已经计算过的不再计算17 18 for(int i = 0; i < n; i++)19 {20 int tmp = (int)t[i];21 if(f>c[i])22 d[f] += 1.0/n*(double)tmp;23 else24 d[f] += 1.0/n*(1.0+dfs(f+c[i]));25 }26 return d[f];27 }28 29 int main()30 {31 int f, i;32 while(~scanf("%d%d", &n, &f))33 {34 for(i = 0; i < n; i++)35 {36 scanf("%d", &c[i]);37 t[i] = (double)c[i]*c[i]*1.0*(1.0+sqrt(5.0))/2.0;38 }39 memset(d, 0, sizeof(d));40 printf("%.3lf\n", dfs(f));41 }42 return 0;43 }
zoj 3640 Help Me Escape (概率dp 递归求期望)