动态规划之—状态机模型(大盗阿福、没有上司的舞会)

大盗阿福

题目链接:acwing1049. 大盗阿福
题目描述:

分析：
这是一道比较简单的动态规划—状态机的问题，可以用线性DP的思想或者DP状态机的思想来做。
首先分析线性DP的做法:

状态表示:
用f[i],表示考虑前i家商铺，大盗阿福能偷盗的最多现金
状态转移:
有两种选择，不偷第i个店铺和偷第i个店铺，两种选择中取max
f[i]=max(f[i-1],f[i-2]+x)//x为在商铺i能偷到的现金
因为不能连着偷两家商铺，所以如果：
不偷:f[i]=f[i-1]
偷:f[i]=f[i-2]+x

代码如下:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;

int f[N];

int main(){
    int t;
    cin>>t;
    while(t--){
        int n;
        cin>>n;
        for(int i=2;i<=n+1;i++){
            int x;
            scanf("%d",&x);
            f[i]=max(f[i-1],f[i-2]+x);
        }
        cout<<f[n+1]<<endl;
    }
    return 0;
}

这种需要向前看两步，所以为了方便我们遍历从2~n+1。
再分析用DP状态机的思想来解决:

状态表示:
用f[i][0]表示不偷第i个商铺，在前i个商铺中能偷到的最多现金
用f[i][1]表示偷第i个商铺，在前i个商铺中能偷到的最多现金

状态转移:
f[i][0]=max(f[i-1][0],f[i-1][1]);//不偷第i个商铺
f[i][1]=f[i-1][0]+x;//偷第i个商铺,则不能偷第i-1个商铺

代码如下:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
int f[N][2];
 
int main(){
    int t;
    cin>>t;
    while(t--){
        int n;
        cin>>n;
        for(int i=1;i<=n;i++){
            int x;
            scanf("%d",&x);
            f[i][0]=max(f[i-1][0],f[i-1][1]);
            f[i][1]=f[i-1][0]+x;
        }
        cout<<max(f[n][0],f[n][1])<<endl;
    }
    return 0;
}

DP状态机来表示状态一般比较清晰

没有上司的舞会

题目链接:acwing285. 没有上司的舞会
题目描述:

分析:

状态表示:
这是一道树上DP的问题,我们用f[i]来表示员工i以即其属下的子树
f[i][0]表示不选员工i,这棵子树的最大happy
f[i][1]表示选员工i,这棵子树的最大happy

状态转移:
f[i][1]表示选了员工i，则其直接属下不能选
f[i][0]表示没选员工i，则其直接属下可选可不选
f[x][1]+=f[v][0];
f[x][0]+=max(f[v][0],f[v][1]);

这里存图我用的是链式前向星
代码如下:

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
const int N=6e3+10;

int f[N][2];
int happy[N];
int father[N];

struct E{
    int v,w,nxt;
}Edge[N];
int Head[N],idx=0;

void add(int u,int v){
    Edge[idx].v=v;
    // Edge[idx].w=w;
    Edge[idx].nxt=Head[u];
    Head[u]=idx++;
}

void init(int n){
    for(int i=0;i<=n;i++) Head[i]=-1;
}

void dfs(int x){
    f[x][1]=happy[x];
    for(int i=Head[x];~i;i=Edge[i].nxt){
        int v=Edge[i].v;
        dfs(v);
        f[x][1]+=f[v][0];
        f[x][0]+=max(f[v][0],f[v][1]);
    }
}
int main(){
    int n;
    cin>>n;
    init(n);
    for(int i=1;i<=n;i++) 
        scanf("%d",&happy[i]);
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&v,&u);
        father[v]++;
        add(u,v);
    }
    int root=-1;
    for(int i=1;i<=n;i++)
        if(!father[i]){
            root=i;
            break;
        }
    dfs(root);
    cout<<max(f[root][0],f[root][1]);
    return 0;
}