zoukankan      html  css  js  c++  java
  • caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)

    这道题的难点在于状态怎么设计
    这道题要求全部都是安全的,所以我们做的时候自底向上每一个结点都要是安全的
    结合前一题当前结点选和不选,我们可以分出四种情况出来
    选 安全
    选 不安全
    不选 安全
    不选 不安全

    显然选 不安全是不可能的,那么就去掉
    所以我们就可以设计状态为
    f[0][i]表示i放人且安全
    f[1][i]表示i不放人且安全
    f[2][i]表示i不放人且不安全

    那么状态转移方程最关键的就是怎么保证回溯的时候都是安全的。
    我们只考虑以u为结点的子树,不考虑i的父亲
    我们要让u的子树除了u以外全部是安全的,u自己安全和不安全分开讨论

    对于 f[0][u], u放人且安全
    那么显然这时儿子无论如何都是安全的(就算原来他是不安全的)
    那么有f[0][u] = sum(min(f[0][v], min(f[1][v], f[2][v])))
    这里v是u的儿子,这时三种情况都可以,取最小

    对于f[2][u] i不放人且不安全
    那么为了保证u不安全肯定儿子不能放人,而这时我们要保证
    儿子都安全,所以f[2][u] = sum(f[1][v])

    对于f[1][u] i不放人且安全
    儿子一定要安全的话有
    f[1][u] =sum(min(f[0][v], f[1][v]))
    但是要保证u安全,v中至少有一个放人
    这就比较麻烦了,我们要专门来判断v中有没有放人
    如果没有的话,就加上min( f[0][v] - f[1][v])
    也就是以最小的费用使一个儿子从不放人到放人

    最后还有一个小细节,之前我写树形dp搜到叶子都直接return的
    这里不行,因为这时f[1][u]是不存在的(不考虑u的父亲)
    所以这时要把f[1][u]初始化为最大值(代码中体现为最后加上mint

    #include<cstdio>
    #include<vector>
    #include<algorithm>
    #define REP(i, a, b) for(int i = (a); i < (b); i++)
    using namespace std;
    
    const int MAXN = 2123;
    int f[3][MAXN], a[MAXN], b[MAXN], n;
    vector<int> g[MAXN];
    
    void dfs(int u)
    {
    	f[0][u] = a[u]; 
    	f[1][u] = f[2][u] = 0; 	
    	
    	int mint = 1e8, ok = 0;
    	REP(i, 0, g[u].size())
    	{
    		int v = g[u][i];
    		dfs(v);
    		f[0][u] += min(f[0][v], min(f[1][v], f[2][v]));
    		f[1][u] += min(f[0][v], f[1][v]);
    		if(f[0][v] <= f[1][v]) ok = 1;
    		mint = min(mint, f[0][v] - f[1][v]);
    		f[2][u] += f[1][v];
    	}
    	if(!ok) f[1][u] += mint;
    } 
    
    int main()
    {
    	scanf("%d", &n);
    	REP(i, 1, n + 1) 
    	{
    		int u, k, son; 
    		scanf("%d", &u);
    		scanf("%d%d", &a[u], &k);
    		REP(j, 0, k)
    		{
    			scanf("%d", &son);
    			g[u].push_back(son);
    			b[son] = 1;
    		}	
    	}
    	
    	REP(i, 1, n + 1)
    		if(!b[i])
    		{
    			dfs(i);
    			printf("%d
    ", min(f[0][i], f[1][i]));
    			break;
    		}
    
    	return 0;	
    } 
  • 相关阅读:
    Swizzle在OC问题排查中的应用
    MacOS中系统提供的音频单元
    Mac catalyst 使用iOS-AudioUnit的音频采集、播放
    删除单向链表中的某一个节点
    C语言的的free和c++的delete的区别
    Mac下使用源码编译安装TensorFlow CPU版本
    ROC曲线与AUC值
    Linux中如何产生core文件?
    更改Linux默认栈空间的大小
    互信息(Mutual Information)
  • 原文地址:https://www.cnblogs.com/sugewud/p/9819393.html
Copyright © 2011-2022 走看看