随机的机器人--七月牛客模拟笔试题解（三）

随机的机器人

| 时间限制：2秒 | 内存限制：32768K | 语言限制： 不限
【随机的机器人】有一条无限长的纸带,分割成一系列的格子,最开始所有格子初始是白色。现在在一个格子上放上一个萌萌的机器人(放上的这个格子也会被染红),机器人一旦走到某个格子上,就会把这个格子涂成红色。现在给出一个整数n,机器人现在会在纸带上走n步。每一步,机器人都会向左或者向右走一个格子,两种情况概率相等。机器人做出的所有随机选择都是独立的。现在需要计算出最后纸带上红色格子的期望值。
输入描述：输入包括一个整数n(0 ≤ n ≤ 500),即机器人行走的步数。
输出描述：输出一个实数,表示红色格子的期望个数,保留一位小数。

示例1：

输入：4
输出：3.4

参考答案

作者：NotDeep
链接：https://www.nowcoder.com/discuss/208143
来源：牛客网

考虑dp[i % 2][j][k]表示走了i步之后恰好有j个红色格子,并且此时机器人正好站在第k个红色格子上的概率。最后求一个可能的状态的带权和就是所求期望。时间复杂度O(n ^ 3)

这个题应该还有很多做法,甚至可以做到O(n^2),O(n),欢迎讨论.

#include <bits/stdc++.h>

using namespace std;

const int maxn = 500 + 5;

double dp[2][maxn][maxn];
int n ;
int main() {
    cin >> n;
    doubl