动态规划“数塔”类型题目总结

前几天做了好几个DP题目,感觉都是一个类型的,因此有必要总结一下。

 

数塔问题 :要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?

分析:站在位置9,我们可以选择沿12方向移动,也可以选择沿着15方向移动,现在我们假设“已经求的”沿12方向的最大值x和沿15方向的最大值y,那么站在9的最大值必然是:Max(x,y) + 9。

因此不难得出,对于任意节点i,其状态转移方程为:m[i] = Max(a[i的左孩子] , a[i的右孩子]) + a[i]

代码
#include <stdio.h>

#define N 10000
#define Max(a,b) ((a) > (b) ? (a) : (b))

int a[N];

int main(void)
{
int n , m , i , k , j;

scanf(
"%d",&m);
while(m-- > 0)
{
scanf(
"%d",&n);
k
= (1 + n) * n / 2;
for(i = 1 ; i <= k; i++)
{
scanf(
"%d",a+i);
}

k
= k - n;
for(i = k , j = 0 ; i >= 1 ; i--)
{
a[i]
= a[i] + Max(a[i+n],a[i+n-1]);
if(++j == n -1)
{
n
--;
j
= 0;
}
}
printf(
"%d\n",a[1]);

}

return 0;
}

首先什么是“数塔类型”?从某一点转向另一点或者说是从某一状态转向另一状态,有多种选择方式(比如这里的9->12 , 9->15),从中选取一条能产生最优值的路径。

这类问题的思考方法:假设后续步骤的结果已知,比如这里假设已经知道沿12方向的最大值x和沿15方向的最大值y。

接下来看几个题,加深印象吧

 

1.免费馅饼问题

                     5 (起始位置)

     4       |      5       |       6

3   4   5  |  4   5   6  |  5   6   7

..................

和“数塔”一样,它也是从某一点出发,有多个选择的问题(往前走一步,呆在原地,往后走一步)从中选择一条最优值路径(获得馅饼最多)。还是按照“数塔”的思考方式,我们可以假设“已经求得”下一个站在位置4获得的最大值x和呆在原地获得的最大值y以及站在位置6获得的最大值z,那么对于起始位置5获得最大值就是Max(x,y,z) ,因此可以得到状态转移方程为:m[t][x] = Max(m[t+1][x-1] , m[t+1][x] , m[t+1][x+1])

并且我们可以通过“列表格”的方式,自底向上求解:

代码
#include <stdio.h>
#include
<string.h>

#define N 100000
int a[N][11];


int Max(int a , int b , int c)
{
int n;

n
= a > b ? a : b;

return n > c ? n : c;
}


int main(void)
{
int n , x , t , max , i;


while(scanf("%d",&n))
{
if(!n) break;

max
= 0;
memset(a ,
0 , sizeof(a));
for(i = 0 ; i < n ; i++)
{
scanf(
"%d%d",&x,&t);
a[t][x]
+= 1;
if(t > max) max = t;
}


//DP
for(t = max - 1 ; t >= 0 ; t--)
{
a[t][
0] += Max(0 , a[t + 1][0] , a[t + 1][1]) ;

for(x = 1 ; x < 10 ; x++)
{
a[t][x]
+= Max(a[t + 1][x - 1] , a[t + 1][x] , a[t + 1][x + 1]) ;
}
a[t][
10] += Max(a[t + 1][9] , a[t + 1][10] , 0) ;
}
printf(
"%d\n",a[0][5]);
}

return 0;
}

2.滑雪问题

      上

左   A    右

  下

依然和“数塔”一样,从某一点出发,面临多个选择(往上,往左,往下,往右)从中选择一条最优值路径(滑雪距离最长)

若对A点求,很显然它的最大值就为: Max(上,右,下,左) + 1

因此对于任意位置[i,j], 其状态转移方程为:m[i][j] = Max(m[i-1][j] , m[i][j+1] , m[i+1][j] , m[i][j-1]) + 1

由于这道题很难画出它的路径图(起点和终点都不知道)因此很难用“列表格”的方式自底向上求解,因此我们采用备忘录法:

代码
#include <stdio.h>
#include
<string.h>
#define N 101

int a[N][N] , m[N][N] , r , c;


int OK(int i ,int j)
{
return (i >= 1 && i <= r && j >= 1 && j <= c);
}

int search(int i , int j)
{
int k;

if(m[i][j] > 0) return m[i][j];

if(OK(i - 1, j) && a[i][j] > a[i - 1][j])
{
k
= search(i - 1, j) + 1;
m[i][j]
= m[i][j] < k ? k : m[i][j];
}
if(OK(i, j + 1) && a[i][j] > a[i][j + 1])
{
k
= search(i, j + 1) + 1;
m[i][j]
= m[i][j] < k ? k : m[i][j];
}
if(OK(i + 1, j) && a[i][j] > a[i + 1][j])
{
k
= search(i + 1, j) + 1;
m[i][j]
= m[i][j] < k ? k : m[i][j];
}
if(OK(i, j - 1) && a[i][j] > a[i][j - 1])
{
k
= search(i, j - 1) + 1;
m[i][j]
= m[i][j] < k ? k : m[i][j];
}

return m[i][j] ;
}

int main(void)
{
int i , j , k , t;

while(scanf("%d%d", &r, &c) != EOF)
{
for(i = 1 ; i <= r ; i++)
for(j = 1 ; j <= c ; j++)
scanf(
"%d", &a[i][j]);

memset(m,
0, sizeof(m)); k = 0;
for(i = 1 ; i <= r ; i++)
for(j = 1 ; j <= c ; j++)
{
t
= search(i, j);
k
= k < t ? t : k;
}

printf(
"%d\n", k + 1);

}
return 0;
}

3.Worm问题,这题和免费馅饼几乎是一样的,我们同样可以使用“列表格”的方式自底向上求解:

代码
#include <stdio.h>
#include
<string.h>
#define N 100

int a[N][N];

int main(void)
{
int t , x , n , p , m , T;

while(scanf("%d%d%d%d", &n, &p, &m, &T) != EOF) //苹果树n,毛毛虫其实位置p,m分钟,终点位置T
{
memset(a,
0, sizeof(a));
a[m][T]
= 1;
for(t = m - 1; t >= 0 ; t--)
{
a[t][
1] += a[t + 1][2];
for(x = 2 ; x < n ; x++)
a[t][x]
+= a[t + 1][x - 1] + a[t + 1][x + 1];
a[t][n]
+= a[t + 1][n - 1];
}
printf(
"%d\n", a[0][p]);
}
return 0;
}

 

 

 

 

 

 

 

 

 

 

posted on 2010-04-18 18:42  DiaoCow  阅读(5178)  评论(0编辑  收藏  举报

导航