首页 > 代码库 > HDU 1520 Anniversary party 树DP水题
HDU 1520 Anniversary party 树DP水题
非常水的树DP,状态为当前为i,上级来没来
然后跑一遍记忆化搜索即可
#include <cstdio>#include <cstring>#include <cmath>#include <algorithm>#include <climits>#include <string>#include <iostream>#include <map>#include <cstdlib>#include <list>#include <set>#include <queue>#include <stack>using namespace std;typedef long long LL;const int maxn = 6000 + 5;int fa[maxn],n,c[maxn];vector<int> ch[maxn];int f[maxn][2];int dfs(int now,bool prev) { int m = ch[now].size(),ret; if(m == 0) return prev ? 0 : c[now]; int ¬e = f[now][prev]; if(note != -1) return note; int ret1 = c[now],ret2 = 0; for(int i = 0;i < m;i++) { int id = ch[now][i]; ret1 += dfs(id,1); ret2 += dfs(id,0); } if(prev) ret = ret2; else ret = max(ret1,ret2); return note = ret;}int main() { while(scanf("%d",&n) != EOF) { memset(f,-1,sizeof(f)); memset(fa,0,sizeof(fa)); for(int i = 1;i <= n;i++) scanf("%d",&c[i]); for(int i = 1;i <= n;i++) ch[i].clear(); int a,b; while(scanf("%d%d",&a,&b), a) { fa[a] = b; ch[b].push_back(a); } int root; for(int i = 1;i <= n;i++) if(fa[i] == 0) root = i; int ret = dfs(root,0); printf("%d\n",ret); } return 0;}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。