2021-02-01树形DP

一.没有上司的舞会

1.题意：

略

2.题解：

建树->树的遍历->对每一个节点选择or不选

3.ac代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int value[N],father[N];
int dp[N][2];//选择or不选 
vector<int> tree[N];
void dfs(int u){
	dp[u][0]=0;
	dp[u][1]=value[u];
	int i;
	int son;
	int sum=tree[u].size();
	for(i=0;i<sum;i++){//遍历该节点的子节点
	     son=tree[u][i];
	     dfs(son); 
	     dp[u][1]=dp[u][1]+dp[son][0];//选父不选子 
	     dp[u][0]=dp[u][0]+max(dp[son][0],dp[son][1]);//不选父，子选不选都行 
		 
	}
}

int main(){
	int n;
	int i;
	int a,b;
	while(cin>>n){
		for(i=1;i<=n;i++){//录入value值 
			cin>>value[i];
			tree[i].clear();
			father[i]=-1;
		}
		while(cin>>a>>b){//用vector建树 
			if(a==0&&b==0)
			 break;
			tree[b].push_back(a);
			father[a]=b;
		}
		int t=1;
		while(father[t]!=-1){//找树根 
			t=father[t];
		}
		dfs(t);//从树根遍历树 ，进行dp
		cout<< max(dp[t][0],dp[t][1])<<endl;
	}
} 

二.computer

1.题意：

输出距离第i个节点最远距离

2.题解：

1.分为两部分，第i个节点的子树到达他的最长距离，还有剩余节点到达他的最长距离
so记录每个节点的子节点最长距离，子节点第二长距离，还有其他节点到达他的最长距离
第i个节点的最远距离为max（子节点最长距离，其他节点到达他的最长距离）
2.其他节点到达他的最远距离求法：
父节点与该节点路权值+父节点的最长/第二长距离
（1）如果该节点在父节点的最长距离上，父节点与该节点路权值+父节点的第二长距离
（2）不在，父节点与该节点路权值+父节点的最长距离

3.ac代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
struct node{
	int cost;//路权值 
	int id;//第几号节点 
};
int value[N];
int dp[N][3];//第一长，第二长，上面的长 
vector<node> tree[N];
int n;
void init(){//初始化，录入节点，父节点，和权值 
	int i;
	int x,y;
	for(i=1;i<=n;i++){
		tree[i].clear();
	}
	for(i=2;i<=n;i++){
		cin>>x>>y;
		node tmp;
		tmp.cost=y;
		tmp.id=i;
		tree[x].push_back(tmp);		
	}
}

void dfs1(int u){//找出每个节点的第一长和第二长 
	int i;
	int one=0;
	int two=0;
	node son;
	int sum=tree[u].size();
	for(i=0;i<sum;i++){
		son=tree[u][i];
		dfs1(son.id);
		int temp=dp[son.id][0]+son.cost;
		if(temp>=one){
			two=one;
			one=temp;
		}
		if(temp<one&&temp>two){
			two=temp;
		}
		
	}
	    dp[u][0]=one;
		dp[u][1]=two;
}
void dfs2(int u){//对每个父节点，求子节点的dp[][2] 
    node son;
    int sum=tree[u].size();
    for(int i=0;i<sum;i++){
    	son=tree[u][i];
    	if(dp[son.id][0]+son.cost==dp[u][0]){//该子节点在父节点的最长距离上 
    		dp[son.id][2]=max(dp[u][1],dp[u][2])+son.cost;
		}else{//不在 
			dp[son.id][2]=max(dp[u][0],dp[u][2])+son.cost;
		}
		dfs2(son.id);
	}
	
}

int main(){

	int i;
	while(cin>>n){
		memset(dp,0,sizeof(dp));
		init();
		dp[1][2]=0;
		dfs1(1);
		dfs2(1);
		int x;
		for(i=1;i<=n;i++){
			x=max(dp[i][0],dp[i][2]);
			cout<<x<<endl;
		}
		
	}
	

}

三.选课

1.题意：

2.题解：

树形背包dp

3.错误代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300;
struct node{
	int cost;//点权值 
	int id;//第几号节点 
};
int n,m;
int value[N];
vector<node> tree[N];
int dp[N][N];//dp[i][j]以i为根的子树选了i时，供选择j门课的最大学分 
void init(){//初始化，录入节点，父节点，和权值 
	int i;
	int x,y;
	for(i=1;i<=n;i++){
		tree[i].clear();
	}
	for(i=1;i<=n;i++){
		cin>>x>>y;
		node tmp;
		tmp.cost=y;
		tmp.id=i;		
		tree[x].push_back(tmp);		
		value[i]=y;
	}
}
void dfs(int u){
	node son;
	int i,j;
	int sum=tree[u].size();
	for(i=0;i<sum;i++){
		son=tree[u][i];
		dfs(son.id);
		for(j=m;j>0;j--){//01背包,u节点，对它分给该子节点【0，m）节课 ，第二维相当于01背包的一维 
			for(int k=0;k<j;k++){
				dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[son.id][k]);
			}
		}
		
	} 
		for(int i=1;i<=m;i++)
        printf("%d ",dp[u][i]);
        printf("\n");
        
		for(i=1;i<=m;i++){//加上根节点u的权值 
			dp[u][i]=dp[u][i-1]+value[u];
		
		for(int i=1;i<=m;i++)
        printf("%d ",dp[u][i]);
        printf("\n");
	}
}
int main(){
	while(cin>>n>>m){
		init();
		memset(dp,0,sizeof(dp));
		dfs(0);
		cout<<dp[0][m]<<endl;
	}
	
	
}