树形背包

P2014 选课

题目描述

在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有N门功课，每门课有个学分，每门课有一门或没有直接先修课（若课程a是课程b的先修课即只有学完了课程a，才能学习课程b）。一个学生要从这些课程里选择M门课程学习，问他能获得的最大学分是多少？

简述： 有许多不同的课有不同的学分，有些课只能上过其先修课才可以上，问上m门课的最大学分

思路：首先是一个树形dp，然后其限制就是将其变成了一个背包

 

#include<bits/stdc++.h>
#define maxn 350
using namespace std;
const int INF = 0x3f3f3f3f;
struct edge{
	int to,w;
};
vector<edge>g[maxn];
int dp[maxn][maxn];//背包
int n,m;
void dfs(int root){
	for(int i=0;i<g[root].size();i++)
	{
		int to = g[root][i].to , w = g[root][i].w;
		if(to==root)continue; //不是他本身
		for(int j=m;j>=0;j--)dp[to][j]=dp[root][j]+w;
		dfs(to);
		for(int k=m;k>=0;k--)
		dp[root][k]=max(dp[root][k],dp[to][k-1]);
	}
	return ;
}
int main() {
   	
   	int x,w;
   	scanf("%d%d",&n,&m);
   	for(int i=1;i<=n;i++)
   	{
   	   scanf("%d%d",&x,&w);
   	   g[x].push_back({i,w});
   	}
   	dfs(0);
   	printf("%d",dp[0][m]);
    return 0;
}

 

P3177 [HAOI2015]树上染色

题目描述

有一棵点数为 N 的树，树边有边权。给你一个在 0~ N 之内的正整数 K ，你要在这棵树中选择 K个点，将其染成黑色，并将其他 的N-K个点染成白色 。 将所有点染色后，你会获得黑点两两之间的距离加上白点两两之间的距离的和的受益。问受益最大值是多少。

输入格式

第一行包含两个整数 N, K 。接下来 N-1 行每行三个正整数 fr, to, dis ， 表示该树中存在一条长度为 dis 的边 (fr, to) 。输入保证所有点之间是联通的。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 2020 + 5;
struct node{
	int v,w,next; 
}e[maxn*2];
int head[maxn];
int st[maxn]={0};
ll dp[maxn][maxn];
int n,m,tot;
  void add(int u,int v,int w)
  {
          e[tot].w=w,e[tot].v=v,e[tot].next=head[u];head[u]=tot++;
  }
 void dfs(int u,int father)
 {
 	st[u] = 1; 
 	dp[u][0]=dp[u][1] = 0; 

 	for(int i=head[u];~i;i=e[i].next)
	 {
		 int v=e[i].v;
		 if(v ==father)continue;
		 dfs(v,u);
		 st[u]+=st[v];
 		for(int j=min(m,st[u]);j>=0;--j)
		 {  
            if(dp[u][j]!=-1)  
                dp[u][j]+=dp[v][0]+(ll)st[v]*(n-m-st[v])*e[i].w;
            for(int k=min(j,st[v]);k;--k){
                if(dp[u][j-k]==-1)continue;
                ll val=(ll)(k*(m-k)+(st[v]-k)*(n-m-st[v]+k))*e[i].w;  
                dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]+val);
            }
        }
	 } 
 }
int main(){
	int u,v,w;
	memset(head,-1,sizeof head);
	scanf("%d%d",&n,&m);
	if(n-m<m)m=n-m;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	} 
	memset(dp,-1,sizeof dp);
	dfs(1,-1);
	printf("%lld",dp[1][m]);
	return 0; 
}