树上状态机

没有上司的舞会

每一条边至少选择一个点

0表示不选，1表示选

#include <bits/stdc++.h>
#define ll long long
#define INF 0x7f7f7f7f
using namespace std;

const int N=6000+10;

int n,m,ecnt,ans;

int head[N];
int f[N][2],w[N];
int vis[N];//判断是否为根节点
struct edge {
	int u,v,next;
} E[N<<1];

void add(int u,int v) {
	E[++ecnt].u=u;
	E[ecnt].v=v;
	E[ecnt].next=head[u];
	head[u]=ecnt;
}

void dfs(int u) {

	f[u][1]=w[u];//初始化
	for(int i=head[u]; i; i=E[i].next) {
		int v=E[i].v;
		dfs(v);
		f[u][0]+=max(f[v][1],f[v][0]);//不选u
		f[u][1]+=f[v][0];//选u
	}
}




int main() {


	scanf("%d",&n);
	for(int i = 1; i <= n; i ++) scanf("%d",&w[i]); //输入每个人的高兴度
	for(int i = 1; i < n; i ++) {
		int a,b; //对应题目中的L,K,表示b是a的上司
		scanf("%d%d",&a,&b); //输入~
		vis[a] = true; //说明a他有爸爸（划掉）上司
		add(b,a); //把a加入到b的后面
	}
	int root = 1; //用来找根节点
	while(vis[root]) root ++; //找根节点
	dfs(root); //从根节点开始搜索
	printf("%d\n",max(f[root][0],f[root][1])); //输出不选根节点与选根节点的最大值
	return 0;



	return 0;

}

战略游戏

每一个边都必须都一个看守

#include <bits/stdc++.h>
#define ll long long
#define INF 0x7f7f7f7f
using namespace std;

const int N=1500+10;

int n,m,ecnt,ans;

int head[N];
int f[N][2];
int vis[N];//判断是否为根节点
struct edge {
	int u,v,next;
} E[N<<1];

void add(int u,int v) {
	E[++ecnt].u=u;
	E[ecnt].v=v;
	E[ecnt].next=head[u];
	head[u]=ecnt;
}

void dfs(int u) {
	f[u][0]=0,f[u][1]=1;
	for(int i=head[u]; i; i=E[i].next) {
		int v=E[i].v;
		dfs(v);
		f[u][0]+=f[v][1];
		f[u][1]+=min(f[v][0],f[v][1]);
	}
}




int main() {


	while(scanf("%d",&n)==1) {
		memset(head,0,sizeof(head));
		memset(vis,0,sizeof(vis)); 
		ecnt=0;//别忘记 
		for(int i=0; i<n; i++) {
			int u,v,cnt;
			scanf("%d:(%d)",&u,&cnt);
			while(cnt--) {
				cin>>v;
				add(u,v);
				vis[v]=1;
			}
		}
		int root=0;
		while(vis[root])root++;//找到根节点
		dfs(root);
		cout<<min(f[root][0],f[root][1])<<endl;
	}





	return 0;

}

皇宫看守

每一个点都要被观察到


f(i，1) 中需要枚举每个点照亮父亲的情况 (i为父亲节点，j为子节点)
公式中min的右侧等价于f(i，0)-min(f(j，1)，f(j，2))+f(k，2) (j不包含k)

#include <bits/stdc++.h>
#define ll long long
#define INF 0x7f7f7f7f
using namespace std;

const int N=1500+10;

int n,m,ecnt,ans;

int head[N];
int f[N][3],w[N];
int vis[N];//判断是否为根节点
struct edge {
	int u,v,next;
} E[N<<1];

void add(int u,int v) {
	E[++ecnt].u=u;
	E[ecnt].v=v;
	E[ecnt].next=head[u];
	head[u]=ecnt;
}

void dfs(int u) {
	f[u][0]=0;//放在父亲上
	f[u][1]=INF;//放在儿子上
	f[u][2]=w[u];//放在自己上

	for(int i=head[u]; i; i=E[i].next) {
		int v=E[i].v;
		dfs(v);
		f[u][0]+=min(f[v][1],f[v][2]);//v的父亲u,没有放
		f[u][2]+=min(min(f[v][0],f[v][1]),f[v][2]);//v的父亲u,放置一个
	}
	//放在儿子上
	//枚举放在哪一个儿子上
	for(int i=head[u]; i; i=E[i].next) {
		int v=E[i].v;
		f[u][1]=min(f[u][1],f[v][2]+f[u][0]-min(f[v][1],f[v][2]));//公式推导
	}
}




int main() {


	cin>>n;
	int u,k,v;
	for(int i=1; i<=n; i++) {
		cin>>u;
		cin>>w[u]>>k;
		while(k--) {
			cin>>v;
			add(u,v);
			vis[v]=1;
		}
	}

	int root=1;
	while(vis[root])root++;//寻找根节点
	dfs(root);
	cout<<min(f[root][1],f[root][2])<<endl;//根节点无父亲节点




	return 0;

}