[BZOJ4871][树形DP]SHOI2017:摧毁“树状图”

BZOJ4871

题外话:这是魔禁里的那个树状图么
题内话:建议大家找个有图的题解

很容易看出来是树形DP,设计状态:
f [ x ] [ 0 ] f[x][0] f[x][0]:穿过x向上的半条链
f [ x ] [ 1 ] f[x][1] f[x][1]:不穿过x且完全在子树内的一条链
f [ x ] [ 2 ] f[x][2] f[x][2]:穿过x且完全在子树内的一条链
f [ x ] [ 3 ] f[x][3] f[x][3]:穿过x向上的半条链以及完全在子树内的一条链

Code:

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=5e5+5;
vector<int>e[N];
inline void add(int u,int v){e[u].push_back(v);}
int f[N][4],ans;
int in[N];
inline void upd(int &a,const int &b){a=max(a,b);}
void dfs(int v,int fa) {
    bool isrt=v==1;
    int deg=e[v].size()-!isrt;
    f[v][0]=f[v][2]=f[v][3]=deg;f[v][1]=1;
    int maxn=0;
    for(int i=0;i<e[v].size();i++) {
        int y=e[v][i];
        if(y==fa) continue;
        dfs(y,v);
        upd(ans,f[v][0]+f[y][3]-isrt);
        upd(ans,f[v][3]+f[y][0]-isrt);
        upd(ans,f[v][1]+f[y][1]-1);
        upd(ans,f[v][1]+f[y][2]);
        upd(ans,f[v][2]+f[y][1]-isrt);
        upd(ans,f[v][2]+f[y][2]-isrt);
        upd(f[v][1],f[y][1]);
        upd(f[v][1],f[y][2]+1);
        upd(f[v][3],f[y][3]+deg-1);
        upd(f[v][3],f[y][0]+maxn+deg-2);
        upd(f[v][3],f[v][0]+f[y][1]-1);
        upd(f[v][3],f[v][0]+f[y][2]-1);
        upd(f[v][3],f[v][2]+f[y][0]-1);
        upd(f[v][2],f[v][0]+f[y][0]-1);
        upd(f[v][0],f[y][0]+deg-1);
        upd(f[v][2],f[v][0]);
        upd(f[v][3],f[v][2]);
        upd(maxn,f[y][1]);
        upd(maxn,f[y][2]);
    }
}
int main(){
    int t=read(),x=read();
    while(t--){
        int n=read();
        for(int i=0;i<x*2;i++) read();
        for(int x,y,i=1;i<n;i++){x=read(),y=read();add(x,y);add(y,x);in[x]++,in[y]++;}
        ans=0;
        dfs(1,0);
        cout<<ans<<"\n";
        for(int i=1;i<=n;i++){e[i].clear();}
    }
    return 0;
}
已标记关键词 清除标记
相关推荐
【为什么还需要学习C++?】 你是否接触很多语言,但从来没有了解过编程语言的本质? 你是否想成为一名资深开发人员,想开发别人做不了的高性能程序? 你是否经常想要窥探大型企业级开发工程的思路,但苦于没有基础只能望洋兴叹?   那么C++就是你个人能力提升,职业之路进阶的不二之选。 【课程特色】 1.课程共19大章节,239课时内容,涵盖数据结构、函数、类、指针、标准库全部知识体系。 2.带你从知识与思想的层面从0构建C++知识框架,分析大型项目实践思路,为你打下坚实的基础。 3.李宁老师结合4大国外顶级C++著作的精华为大家推出的《征服C++11》课程。 【学完后我将达到什么水平?】 1.对C++的各个知识能够熟练配置、开发、部署; 2.吊打一切关于C++的笔试面试题; 3.面向物联网的“嵌入式”和面向大型化的“分布式”开发,掌握职业钥匙,把握行业先机。 【面向人群】 1.希望一站式快速入门的C++初学者; 2.希望快速学习 C++、掌握编程要义、修炼内功的开发者; 3.有志于挑战更高级的开发项目,成为资深开发的工程师。 【课程设计】 本课程包含3大模块 基础篇 本篇主要讲解c++的基础概念,包含数据类型、运算符等基本语法,数组、指针、字符串等基本词法,循环、函数、类等基本句法等。 进阶篇 本篇主要讲解编程中常用的一些技能,包含类的高级技术、类的继承、编译链接和命名空间等。 提升篇: 本篇可以帮助学员更加高效的进行c++开发,其中包含类型转换、文件操作、异常处理、代码重用等内容。
©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页