記憶化搜尋題目總結(1)
記憶化搜尋題目總結(1)
一. 概述
演算法上依然是搜尋的流程,但是搜尋到的一些解用動態規劃的那種思想和模式作一些儲存。在求解的時候還是按著自頂向下的順序,但是每求解一個狀態,就將它的解儲存下來,以後再次遇到這個狀態的時候,就不必重新求解了。這種方法綜合了搜尋和動態規劃兩方面的優點。
二. 引題
POJ1579 題目連結
if a <= 0 or b <= 0 or c <= 0, then w(a, b, c) returns 1
if a > 20 or b > 20 or c > 20, then w(a, b, c) returns w(20, 20, 20)
if a < b and b < c, then w(a, b, c) returnsw(a, b, c-1) + w(a, b-1, c-1) - w(a, b-1, c)
otherwise it returns w(a-1, b, c) + w(a-1, b-1, c) + w(a-1, b, c-1) - w(a-1, b-1, c-1)
DFS做法
long long dfs(int x, int y, int z)
{
if(x <= 0 || y <= 0 || z <= 0){
return 1;
}
if(x > 20 || y > 20 || z > 20){
x = y = z = 20;
}
if(x < y && y < z){
return dfs(x, y, z - 1) + dfs(x, y - 1, z - 1) - dfs(x, y - 1, z);
}
else {
return dfs(x - 1, y, z) + dfs(x - 1, y - 1, z) + dfs(x - 1, y, z - 1) - dfs(x - 1, y - 1, z - 1);
}
}
上述的dfs做法可以發現計算了大量的重疊子問題,如在計算dfs(4, 4, 4)和dfs(3, 4, 5)時都需要計算dfs(3, 4, 4),都會遍歷dfs(3, 4, 4)這整顆子樹,花費大量時間,因此我們想到將每個狀態的值記錄下來,以後求解時可以直接得到該值。
記憶化搜尋解法
#include<iostream>
#include<cstring>
#include<stdio.h>
using namespace std;
long long dp[21][21][21];
long long dfs(int x, int y, int z)
{
if(x <= 0 || y <= 0 || z <= 0){
return 1;
}
if(dp[x][y][z]){
return dp[x][y][z];
}
if(x < y && y < z){
dp[x][y][z] = dfs(x, y, z - 1) + dfs(x, y - 1, z - 1) - dfs(x, y - 1, z);
}
else{
dp[x][y][z] = dfs(x - 1, y, z) + dfs(x - 1, y - 1, z) + dfs(x - 1, y, z - 1) - dfs(x - 1, y - 1, z - 1);
}
return dp[x][y][z];
}
int main()
{
int i;
int x, y, z;
memset(dp, 0, sizeof(dp));
while(1)
{
cin >> x >> y >> z;
if(x == -1 && y == -1 && z == -1)
{
break;
}
long long ans;
if(x <= 0 || y <= 0 || z <= 0){
ans = 1;
}
else if(x > 20 || y > 20 || z > 20){
ans = dfs(20, 20, 20);
}
else{
ans = dfs(x, y, z);
}
printf("w(%d, %d, %d) = %lld\n", x, y, z, ans);
}
return 0;
}
三. 例題
數字三角形
POJ1163 題目連結
給定一個由n行數字組成的數字三角形。試設計一個演算法,計算出從三角形的頂至底的一條路徑,使該路徑經過的數字總和最大。
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
思路:
定義dp[i][j]為第i行第j列到底部的最大數字和,
狀態轉移方程為dp[i][j] = max(dp[i + 1][j], dp[i + 1][j + 1]) + array[i][j]
#include<cmath>
#include<iostream>
using namespace std;
int a[10][10];
int d[10][10];
int dp(int i,int j,int n)
{
if(d[i][j] > 0)
return d[i][j];
if(i == n-1)
return a[i][j];
else
{
return d[i][j] = a[i][j] + max(dp(i+1,j,n),dp(i+1,j+1,n));
}
}
int main()
{
int i,j,n;
cin>>n;
for(i=0; i<n; i++)
for(j=0; j<=i; j++){
cin>>a[i][j];
}
for(i=0; i<n; i++)
for(j=0; j<=i; j++){
d[i][j]=0;
}
dp(0,0,n);
cout << d[0][0];
return 0;
}
滑雪
POJ1088 題目連結
Michael喜歡滑雪百這並不奇怪, 因為滑雪的確很刺激。可是為了獲得速度,滑的區域必須向下傾斜,而且當你滑到坡底,你不得不再次走上坡或者等待升降機來載你。Michael想知道載一個區域中最長底滑坡。區域由一個二維陣列給出。陣列的每個數字代表點的高度。下面是一個例子
1 2 3 4 5
16 17 18 19 6
15 24 25 20 7
14 23 22 21 8
13 12 11 10 9
一個人可以從某個點滑向上下左右相鄰四個點之一,當且僅當高度減小。
思路:
定義dp[i][j]為從第i行第j列到最後的最長長度
狀態轉移方程:dp[i][j] = max(dp[i-1][j], dp[i][j-1], dp[i+1][j], dp[i][j+1]) + 1
#include<iostream>
using namespace std;
int array[101][101];
int dp[101][101];
int r, c;
int dict[4][2] = {{-1, 0}, {0, -1}, {1, 0}, {0, 1}}; //四個方向
int dfs(int x, int y)
{
if(dp[x][y] > 0){
return dp[x][y];
}
dp[x][y] = 1;
int i;
for(i=0; i<4; i++)
{
int dx = x + dict[i][0];
int dy = y + dict[i][1];
if(dx >= 1 && dx <= r && dy >= 1 && dy <= c)
{
if(array[dx][dy] < array[x][y])
{
dp[x][y] = max(dp[x][y], dfs(dx, dy) + 1);
}
}
}
return dp[x][y];
}
int main()
{
int i, j, k;
cin >> r >> c;
for(i=1; i<=r; i++)
for(j=1; j<=c; j++)
{
cin >> array[i][j];
}
int maxx = 0;
for(i=1; i<=r; i++)
for(j=1; j<=c; j++)
{
int cur = dfs(i, j);
if(cur > maxx){
maxx = cur;
}
}
cout << maxx << endl;
}
矩形巢狀
nyoj16 題目連結
有n個矩形,每個矩形可以用a,b來描述,表示長和寬。矩形X(a,b)可以巢狀在矩形Y(c,d)中當且僅當a < c, b < d或者b < c, a < d(相當於旋轉X90度)。例如(1,5)可以巢狀在(6,2)內,但不能巢狀在(3,4)中。你的任務是選出儘可能多的矩形排成一行,使得除最後一個外,每一個矩形都可以巢狀在下一個矩形內。
定義d[i]為第i個矩陣可巢狀的最大個數,b[i][j]為第i個矩陣是否可以巢狀進第j個矩陣。
狀態轉移方程:d[i] = max(d[j] ) + 1,其中b[i][j] = 1;
#include<cmath>
#include<iostream>
using namespace std;
int a[1001][2];
int b[1001][1001]; //鄰接矩陣
int d[1001];
int dp(int i, int n)
{
if(d[i] > 0)
return d[i];
int j;
d[i] = 1;
for(j=0; j<n; j++)
{
if(b[i][j] == 1)
{
d[i] = max(d[i], dp(j,n)+1);
}
}
return d[i];
}
//先構造圖
void init(int n)
{
int i, j;
for(i=0; i<n; i++)
for(j=0; j<n; j++)
{
if((a[i][0] < a[j][0] && a[i][1] < a[j][1]) ||
(a[i][0] < a[j][1] && a[i][1] < a[j][0]))
b[i][j] = 1;
}
}
int main()
{
int N, n, i, j, k, max=0;
cin >> N;
for(i=1; i<=N; i++)
{
max = 0;
cin >> n;
for(j=0; j<n; j++)
cin >> a[j][0] >> a[j][1];
for(j=0; j<n; j++)
for(k=0; k<n; k++){
b[j][k] = 0;
}
for(j=0; j<n; j++){
d[j] = 0;
}
init(n);
for(j=0; j<n; j++){
dp(j,n);
}
for(j=0; j<n; j++){
if(d[j] > max)
{
max = d[j];
}
}
cout << max << endl;
}
}
硬幣問題
題目:有n種硬幣,面值分別為V1,V2,…Vn,每種都有無限多。給定非負整數S,可以選用多少個硬幣,使得面值之和恰好為S?輸出硬幣數目的最小值。
思路:
定義d[s]為面值s到面值0最少可用的硬幣數目
狀態轉移方程:d[s] = max(d[s - v[i]] + 1),其中s >= v[i];
int dfs(int s)
{
if(vis[s]){
return dp[s];
}
dp[s] = -(1 << 30);
int i;
for(i=0; i<n; i++){
if(s >= v[i])
{
dp[s] = max(dp[s], dfs(s - v[i]) + 1);
}
}
return dp[s];
}
New Year and Fireworks
參考文獻:《演算法競賽入門經典》