codeforce #331D Wilbur and Trees (记忆化搜索DP)
2015-11-24 20:16
573 查看
题目:http://codeforces.com/contest/596/problem/D
题意:有n课树并排排列,没课树高h,砍树。每次,从剩余的树中选最左边的或最右边的(概率相等,0.5),砍掉一棵树,树倒向左边的概率为p,倒向右边的概率为1-p。如果树和相邻树的距离小于h,树倒时会把相邻的树撞倒。问,砍完所有树,树干覆盖的草地长度的期望值。
思路:
dp[ l ][ r ] 表示看完从 i 到 j 这段的树的期望值,易得 dp[ l ][ r ] 应由 dp[ i ][ j ] (l <= i <= j <= r) 转移得到。
l 到 r 这段有4种情况:
1)砍l,向左倒; 2)砍r, 向右倒;3)砍l,向右倒; 4)砍r, 向左倒。
其中1)2)两种情况不会引起连锁反应
引起连锁反应时,找到影响到的最后一个点,所以需要预处理出每棵树 向左倒影响到的最左边点fr[ i ] 和 向右倒影响到的最右边的点 fl[ i ]。
砍 l 和 r 增加的覆盖长度又是与 l - 1 和 r + 1 倒地的情况有关的,所以要增加维度,用dp[ l ][ r ][ f1 ][ f2 ] ,f1 和 f2 取值为0或1,表示 l - 1 和 r + 1 是否向中间倒。
题意:有n课树并排排列,没课树高h,砍树。每次,从剩余的树中选最左边的或最右边的(概率相等,0.5),砍掉一棵树,树倒向左边的概率为p,倒向右边的概率为1-p。如果树和相邻树的距离小于h,树倒时会把相邻的树撞倒。问,砍完所有树,树干覆盖的草地长度的期望值。
思路:
dp[ l ][ r ] 表示看完从 i 到 j 这段的树的期望值,易得 dp[ l ][ r ] 应由 dp[ i ][ j ] (l <= i <= j <= r) 转移得到。
l 到 r 这段有4种情况:
1)砍l,向左倒; 2)砍r, 向右倒;3)砍l,向右倒; 4)砍r, 向左倒。
其中1)2)两种情况不会引起连锁反应
引起连锁反应时,找到影响到的最后一个点,所以需要预处理出每棵树 向左倒影响到的最左边点fr[ i ] 和 向右倒影响到的最右边的点 fl[ i ]。
砍 l 和 r 增加的覆盖长度又是与 l - 1 和 r + 1 倒地的情况有关的,所以要增加维度,用dp[ l ][ r ][ f1 ][ f2 ] ,f1 和 f2 取值为0或1,表示 l - 1 和 r + 1 是否向中间倒。
#include <stdio.h> #include <string.h> #include <algorithm> #include <math.h> #include <stdlib.h> #define INF 0x7fffffff #define MOD 1000000007 using namespace std; typedef long long ll; double p, dp[2005][2005][2][2]; int a[2005], h, fl[2005], fr[2005], vis[2005][2005][2][2]; double dfs(int l, int r, int f1, int f2) { //f1,f2表示l-1和r+1是否向中间倒 if(vis[l][r][f1][f2]) return dp[l][r][f1][f2]; if(l > r) return 0; double ans = 0; double h1 = min(h, a[l] - a[l - 1] - f1 * h), h2 = min(h, a[r + 1] - a[r] - f2 * h); ans += p * 0.5 * (dfs(l + 1, r, 0, f2) + h1);//l向左倒 ans += (1 - p) * 0.5 * (dfs(l, r - 1, f1, 0)+ h2); //r向右倒 //l向右倒 if(fl[l] >= r) ans += (1 - p) * 0.5 * (a[r] - a[l] + h2); else ans += (1 - p) * 0.5 * (a[fl[l]] - a[l] + h + dfs(fl[l] + 1, r, 1, f2)); //r向左倒 if(fr[r] <= l) ans += p * 0.5 * (a[r] - a[l] + h1); else ans += p * 0.5 * (a[r] - a[fr[r]] + h + dfs(l, fr[r] - 1, f1, 1)); vis[l][r][f1][f2] = 1; return dp[l][r][f1][f2] = ans; } int main() { #ifdef LOCAL freopen("data.in", "r", stdin); #endif int n; while(scanf("%d%d%lf", &n, &h, &p) != EOF) { for(int i = 1; i <= n; i++) scanf("%d", &a[i]); sort(a + 1, a + n + 1); a[0] = -1000000000; a[n + 1] = 1000000000; fl = n; for(int i = n; i >= 1; i--) { if(a[i] - a[i - 1] < h) fl[i - 1] = fl[i]; else fl[i - 1] = i - 1; } fr[1] = 1; for(int i = 1; i <= n; i++) { if(a[i + 1] - a[i] < h) fr[i + 1] = fr[i]; else fr[i + 1] = i + 1; } memset(dp, 0, sizeof(dp)); memset(vis, 0, sizeof(vis)); double fe = dfs(1, n, 0, 0); printf("%.9f\n", fe); } return 0; }
相关文章推荐
- ibatis缓存优化
- 黑马程序员——C语言——数组
- EEG 学习记录
- 【人月神话】第二章:人月神话
- 家庭作业(更新中)
- Android应用启动优化:一种DelayLoad的实现和原理
- PopupWindow
- Java反射机制简介
- Linux.BackDoor.Gates/Chikdos Attack And Defense Analysis
- 我的余弦相似性,基于TFIDF结果
- 日志分析系统
- 如何实现 TreeView Items 全部展开和全部收缩(WPF)
- android 位移、缩放动画
- POJ2387 Til the Cows Come Home(SPFA + dijkstra + BallemFord 模板)
- uva 120 C - Stacks of Flapjacks
- 如何实现 TreeView Items 全部展开和全部收缩(WPF)
- 第二个Sprint冲刺第二天
- 读其他博客有感~
- Winform 子线程更新 控件
- css杂记