动态规划刷题

P1122最大子树和

通过链式前向星建树，然后通过遍历每一个点作为根节点然后进行剪枝，通过dfs找子树f[i]<0的就减去。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
const int N=160010;
int a[N];
int h[N],e[N],ne[N],idx;
int res=-2147483647;
int f[N];
void add(int a,int b){
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int fa){
	for(int i=h[u];i!=-1;i=ne[i]){
		int j=e[i];
		if(j!=fa){
			dfs(j,u);
			if(f[j]>0)f[u]+=f[j];
		}
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		f[i]=a[i];
	}
	memset(h,-1,sizeof(h));
	for(int i=1;i<n;i++){
		int a,b;cin>>a>>b;
		add(a,b),add(b,a);
	}
	dfs(1,0);
	for(int i=1;i<=n;i++)res=max(res,f[i]);
	cout<<res<<endl;
	return 0;
} 

洛谷P1352没有上司的舞会

通过链式前向星建树，然后记录建树时的父节点has_father[]变成true，通过while遍历has_father数组进行找根，通过dfs进行根节点遍历，然后分有没有包括根节点自身f[u][1]和f[u][0]进行分类状态转移，最后输出两个的最大值。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
const int N=6010;
int happy[N];
int f[N][2];
int h[N],e[N],ne[N],idx;
bool has_father[N];
void add(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][1]+=f[j][0];
		f[u][0]+=max(f[j][0],f[j][1]);
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>happy[i];
	memset(h,-1,sizeof(h));
	for(int i=1;i<n;i++){
		int a,b;cin>>a>>b;
		has_father[a]=true;
		add(b,a);
	}
	int root=1;
	while(has_father[root])root++;
	dfs(root);
	cout<<max(f[root][0],f[root][1])<<endl;
	
	return 0;
}