动态规划——树形DP&记忆化搜索

树形DP

树形DP就是在一棵树上进行动态规划，根据一个节点的孩子树递归获取信息，然后进行状态转移。
没有上司的舞会

• 题目描述

Ural 大学有 $N$ 名职员，编号为 $1\sim N$ 。
他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。
每个职员有一个快乐指数，用整数 $H_i$ 给出, 其中 $1\le i\le N$ 。
现在要召开一场周年庆宴会, 不过, 没有职员愿意和直接上司一起参会。
在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大 值。

• 分析

对一个节点，有两个状态：取或不取，取用1表示，不取用0表示，则f[u,0]表示编号是u的节点不取，f[u,1]表示编号为u的节点取。如果一个节点取，则它的直接孩子不能用，所以f[u,1]的所有情况就是他所有直接孩子不取的和，即f[u,1]=sum(f[i,0]),如果一个节点不取，则它的直接孩子取不取都行，所以f[u,0]=sum(max(f[i,0],f[i,1])).

• 代码

#include<bits/stdc++.h>
using namespace std;
const int N = 6010;
int happy[N];
int h[N],e[N],ne[N],idx;
bool father[N];
int f[N][2];
int n;

void Insert(int a,int b){
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

void dfs(int u){
    f[u][1] = happy[u];
    for(int i=h[u];i!=-1;i=ne[i]){
        int j = e[i];
        dfs(j);
        f[u][0] += max(f[j][0],f[j][1]);
        f[u][1] += f[j][0];
    }
}

int main()
{
    scanf("%d",&n);
    memset(h,-1,sizeof(h));
    for(int i=1;i<=n;i++)scanf("%d",&happy[i]);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        father[a] = true;
        Insert(b,a);
    }
    int root = 1;
    while(father[root])root++;
    dfs(root);
    printf("%d",max(f[root][1],f[root][0]));
    return 0;
}

记忆化搜索

记忆化搜索就是用一个数组存储一个点是否被访问用过。

滑雪

• 题目描述

给定一个 $R$ 行 $C$ 列的矩阵，表示一个矩形网格滑雪场。
矩阵中第 $i$ 行第 $j$ 列的点表示滑雪场的第 $i$ 行第 $j$ 列区域的高度。
一个人从滑雪场中的某个区域内出发，每次可以向上下左右任意一个方向滑动一个单位距离。
当然，一个人能够滑动到某相邻区域的前提是该区域的高度低于自己目前所在区域的高度。

• 分析

用f[i,j]表示坐标是(i,j)的点出发的所有情况，f[i,j]由他滑到的下一个点更新。
由于每次可能向四个方向滑行，且可能重复，因此用f记录是否被走过。

• 代码

#include<bits/stdc++.h>
using namespace std;
const int  N = 310;
int f[N][N],h[N][N];
int n,m;

int dx[4] = {0,1,0,-1},dy[4] = {1,0,-1,0};

int dp(int x,int y){
    int &v = f[x][y];
    if(v!=-1)return v;
    v = 1;
    for(int i=0;i<4;i++){
        int a = x+dx[i],b = y+dy[i];
        if(a>=1&&a<=n&&b>=1&&b<=m&&h[a][b]<h[x][y])
            v = max(v,dp(a,b)+1);//状态转移。
    }
    return v;
}

int main()
{
    scanf("%d%d",&n,&m);
    memset(f,-1,sizeof(f));
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",&h[i][j]);
    
    int res = 0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            res = max(res,dp(i,j));
            
    printf("%d",res);
    return 0;
}