树型动态规划

xingxg.

已于 2022-08-11 13:41:22 修改

阅读量792

点赞数 1

分类专栏： # 动态规划文章标签：动态规划算法

于 2022-05-06 09:23:01 首次发布

本文链接：https://blog.csdn.net/PURSUE__LIFE/article/details/124602568

版权

动态规划专栏收录该内容

6 篇文章 0 订阅

订阅专栏

统计人数

// sgy 佚名
// 树形动态规划
// problem： 统计人数
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 100005;
struct edge{
	int to, next;
}e[N];
int head[N];
int n,cnt;
ll f[N];

inline void makelist(int u,int v){
	e[++cnt].to = v;
	e[cnt].next = head[u];
	head[u] = cnt;
}

// dfs写法
inline void solve(int i){
	f[i] = 1;
	for(int x = head[i] ; x ; x = e[x].next){
		int to= e[x].to;
		solve(to);
		f[i] += f[to];
	}
}
int main(){
	cin>>n;
	cnt = 0;
	for(int i = 2;i<=n;++i){
		int x;cin>>x;
		makelist(x,i);
	}
	// dfs写法
	solve(1);
	for(int i = 1; i <= n; ++i)cout<<f[i]<<" ";
	cout<<endl;

	return 0;
}

没有上司的舞会I

// sgy  
// 树形动态规划
// problem ： 没有上司的舞会I
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 100005;
struct edge{
	int to, next;
}e[N];
int head[N];
int n,cnt,v[N];
ll f[N][2];  // 0: 不参加   1:参加

inline void makelist(int u,int v){
	e[++cnt].to = v;
	e[cnt].next = head[u];
	head[u] = cnt;
}

// dfs写法
inline void solve(int i){
	f[i][1] = v[i];
    f[i][0] = 0;
	for(int x = head[i] ; x ; x = e[x].next){
		int to= e[x].to;
		solve(to);
		f[i][0] += max(f[to][0],f[to][1]);
		f[i][1] += f[to][0];
	}
}
int main(){
	cin>>n;
	for(int i = 0;i<N;++i)e[i].next = 0;
	cnt = 0;
	for(int i = 2;i<=n;++i){
		int x;cin>>x;
		makelist(x,i);
	}
	for(int i = 1;i<=n;++i)cin>>v[i];
	// dfs写法
	solve(1);
	cout<<max(f[1][0],f[1][1])<<endl;
	cout<<endl;

	return 0;
}

没有上司的舞会II

// sgy  
// 树形动态规划
// problem ： 没有上司的舞会II
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 505;
struct edge{
	int to, next;
}e[N];
int head[N];
int n,m,cnt,v[N];
ll f[N][N][2];  // 0: 不参加   1:参加

inline void makelist(int u,int v){
	e[++cnt].to = v;
	e[cnt].next = head[u];
	head[u] = cnt;
}

// dfs写法
inline void solve(int i){
	for(int x = head[i] ; x ; x = e[x].next){
		int to= e[x].to;
		solve(to);
		for(int j = m; j >= 0; --j){
			for(int k = 0; k <= j; ++k){
				f[i][j][0] = max(f[i][j][0], f[i][j-k][0] + max(f[to][k][0], f[to][k][1]));
				f[i][j][1] = max(f[i][j][1], f[i][j-k][1] + f[to][k][0]);
			}
		}
	}
	// i 这个人最后要考虑一下，因为之前都是考虑他的下属所带来的快乐值。
	for(int j = m; j ; --j){
		f[i][j][1] = f[i][j-1][1] + v[i];
		f[i][j][0] = f[i][j][0] + 0;
	}
}
int main(){
	cin>>n>>m;
	for(int i = 0;i<N;++i)e[i].next = 0;
	cnt = 0;
	for(int i = 2;i<=n;++i){
		int x;cin>>x;
		makelist(x,i);
	}
	for(int i = 1;i<=n;++i)cin>>v[i];
	// dfs写法
	solve(1);
	cout<<max(f[1][m][0],f[1][m][1])<<endl;
	cout<<endl;

	return 0;
}

树上背包1

problem：在u的子树中选取包含u的一个大小为m的连通块，要求权值最大。

solve：

dp【u】【i】定义为在u这个子树选取大小为i的连通块

u这个节点肯定是要选的，可以先放着。假设u有5个分支，可以先用分支1去合并分支2，将分支1和分支2合并后，再将合并后的与分支3进行合并......直至合并完所有的子树，最后加上u这个节点

时间复杂度：O(n^2)

// problem :  

#include <bits/stdc++.h>
using namespace std;
#define ll long long
typedef pair<int, int> PII;
#define pb push_back
const int inf = 1 << 29;
const int N = 2010;
int n, q, dp[N][N], a[N], sz[N];
vector<int> son[N];
void dfs(int u) {
    sz[u] = 0;
    static int tmp[N];
    for (auto v : son[u]) {
        dfs(v);
        
        for (int i = 0; i <= sz[u] + sz[v]; ++i) {
            tmp[i] = -inf;
        }
        for (int i = 0; i <= sz[u]; ++i) {
            for (int j = 0; j <= sz[v]; ++j) {
                tmp[i + j] = max(tmp[i + j], dp[u][i] + dp[v][j]);
            }
        }
        for (int i = 0; i <= sz[u] + sz[v]; ++i) {
            dp[u][i] = tmp[i];
        }
        sz[u] += sz[v];
    }
    sz[u]++;
    for (int i = sz[u]; i >= 1; --i) {
        dp[u][i] = dp[u][i - 1] + a[u];
    }
}
int main(){
    scanf("%d %d", &n, &q);
    for (int i = 2; i <= n; ++i) {
        int f; scanf("%d", &f);
        son[f].push_back(i);
    }   
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
    }
    dfs(1);
    for (int i = 0; i < q; ++i) {
        int u, m; scanf("%d %d", &u, &m);
        printf("%d\n", dp[u][m]);
    }
    return 0;
}

树上背包2

跟树上背包的题目一样，只不过数据的范围发生了变化，N巨大，m小于等于100。由m的范围，我们可以确定dp[N][M] 只要维护从u子树选取 <= M 个节点的数量连通块即可。

也相当于是暴力吧，但就是过的了。没想到就不行。时间复杂度 O(nm)

// problem :  

#include <bits/stdc++.h>
using namespace std;
#define ll long long
typedef pair<int, int> PII;
#define pb push_back
const int inf = 1 << 29;
const int N = 50010, M = 100;
int n, q, dp[N][M + 10], a[N], sz[N];
vector<int> son[N];
void dfs(int u) {
    sz[u] = 0;
    static int tmp[M + 10];
    for (auto v : son[u]) {
        dfs(v);
        
        for (int i = 0; i <= sz[u] + sz[v] && i <= M; ++i) {
            tmp[i] = -inf;
        }
        for (int i = 0; i <= sz[u] && i <= M; ++i) {
            for (int j = 0; j <= sz[v] && i + j <= M; ++j) {
                tmp[i + j] = max(tmp[i + j], dp[u][i] + dp[v][j]);
            }
        }
        for (int i = 0; i <= sz[u] + sz[v] && i <= M; ++i) {
            dp[u][i] = tmp[i];
        }
        sz[u] += sz[v];
    }
    sz[u]++;
    for (int i = min(sz[u], M); i >= 1; --i) {
        dp[u][i] = dp[u][i - 1] + a[u];
    }
}
int main(){
    scanf("%d %d", &n, &q);
    for (int i = 2; i <= n; ++i) {
        int f; scanf("%d", &f);
        son[f].push_back(i);
    }   
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
    }
    dfs(1);
    for (int i = 0; i < q; ++i) {
        int u, m; scanf("%d %d", &u, &m);
        printf("%d\n", dp[u][m]);
    }
    return 0;
}

树上背包3

与前两种题型不同的是，前两种都是求连通块的大小，一个节点的大小固定为1，而这题是求重量为某一个值。如果节点为1 2 4 ，那么节点个数为3，可是重量的状态就已经达到了8种。这题有个很巧妙的技巧，也就是利用dfs序。如果一个节点没有被选中，那么这个节点的孩子节点也不可能被选上。直接跳到这个节点的兄弟节点的dfs序上去；如果这个节点选上了，那就正常的dp下去。

// problem :  

#include <bits/stdc++.h>
using namespace std;
#define ll long long
typedef pair<int, int> PII;
#define pb push_back
const int inf = 1 << 29;
const int N = 1010, M = 10010;
vector<int> son[N];
int n, m, a[N], w[N];
int l[N], r[N], id[N], tot;
int dp[N][M];
void dfs(int u) {
    l[u] = ++tot;
    id[tot] = u;
    for (auto v : son[u]) {
        dfs(v);
    }
    r[u] = tot;
}
int main(){
    scanf("%d %d", &n, &m);
    for (int i = 2; i <= n; ++i) {
        int f; scanf("%d", &f);
        son[f].push_back(i);
    }   
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
    }
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &w[i]);
    }
    dfs(1);

    for (int j = 1; j <= m; ++j) {
        dp[n + 1][j] = -inf;
    }
    for (int i = n; i >= 1; --i) {
        int u = id[i];
        for (int j = 0; j <= m; ++j) {
            dp[i][j] = dp[r[u] + 1][j];
            if (j >= w[u]) {
                dp[i][j] = max(dp[i][j], dp[i + 1][j - w[u]] + a[u]);
            }
        }
    }
    for (int i = 0; i <= m; ++i) {
        if (dp[1][i] >= 0) printf("%d\n", dp[1][i]);
        else printf("0\n");
    }
    return 0;
}