【dp】——随机的机器人

题目来自牛客网：

 【随机的机器人】有一条无限长的纸带,分割成一系列的格子,最开始所有格子初始是白色。现在在一个格子上放上一个萌萌的机器人(放上的这个格子也会被染红),机 器人一旦走到某个格子上,就会把这个格子涂成红色。现在给出一个整数n,机器人现在会在纸带上走n步。每一步,机器人都会向左或者向右走一个格子,两种情况概率相等。机器人做出的所有随机选择都是独立的。现在需要计算出最后纸带上红色格子的期望值。

输入描述： 输入包括一个整数n(0 ≤ n ≤ 500),即机器人行走的步数。

输出描述： 输出一个实数,表示红色格子的期望个数,保留一位小数。

示例：

输入：4    输出：3.4 

#include<iostream>
using namespace std;

double dp[2][505][505]={0};
int main()
{
	int n;
	cin>>n;
	dp[0][1][1]=1.0;
	int last,now;
	for(int i=1;i<=n;i++)
	{
		last=(i+1)%2;
		now=i%2;
		for(int j=1;j<=i+1;j++)
			for(int k=0;k<=j;k++)
				dp[now][j][k]=0;
		for(int j=1;j<=i+1;j++)
			for(int k=1;k<=j;k++)
			{
				// go to left
				if(k==1)
					dp[now][j+1][k]+=dp[last][j][k]*0.5;
				else
					dp[now][j][k-1]+=dp[last][j][k]*0.5;
				// go to right
				if(k==j)
					dp[now][j+1][k+1]+=dp[last][j][k]*0.5;
				else
					dp[now][j][k+1]+=dp[last][j][k]*0.5;
			}
	}
	double ans=0;
	for(int j=1;j<=n+1;j++)
		for(int k=1;k<=j;k++)
			ans+=dp[now][j][k]*j;
	printf("%.1f",ans);
	return 0;
}

dp要声明为全局变量，因为它需要的内存空间超过了VS默认分配给局部变量的空间大小