DP动态规划学习笔记——高级篇上

说了要肝的怎么能咕咕咕呢？

不了解DP或者想从基础开始学习DP的请移步上一篇博客：DP动态规划学习笔记

这一篇博客我们将分为上中下三篇（这样就不用咕咕咕了...），上篇是较难一些树形DP，中篇则是数位和状压DP，下篇则是各种DP的优化手段。

——正片开始—— （为啥我最近的博客都喜欢写这个）

背包类树形DP，树形DP里一种很鬼畜的题目。

简单点讲就是：树上的分组背包。不知道分组背包的也请前往上一篇学习。

我们先来看一道板子题：选课

然后我们一起分析一下这道题（最好自己先想一想），由于每门课的先修课只有一门，所以我们很容易想到用树形结构储存这种关系——>只能从父节点到子节点。但是，这一题的数据会形成一个森林。想一想，为什么？

没错，可能不止一门课没有先修课。但是我们只学过树形DP，怎么办呢？简单，我们自己整一棵树出来呗。

但是原来的父子关系又不能变，怎么整呢？我们新建一个“虚拟课程”0出来，作为“没有先修课的课程”的先修课。也就是说我们把森林里的每一棵树都连到一个新的根节点——0节点上，这样我们就得到了一棵树。

这个想法是不是很妙？然后这题就转换成了一个在树上运行的分组背包。

然后我们来设状态。不会？那你别学了。上一篇博客看了没？快去看。

我们上一篇讲过了，树形DP一般以每个节点x作为第一维...

然后呢？（锤），分组背包啊。哦...

于是我们就得到状态了：设f[x,j]表示我们从以x为根的子树中选j门课能获得的最高学分。

很简单是不是？状态都出来了还不知道方程咩？那我们一起来推一推吧。

我们定义几个变量方便描述，设son(x)表示x的子节点集合，siz(x)表示x的子节点个数，如果选修x这门课，那我们就对于任意y∈son(x)，可以从以y为根的的子树中选出若干门课(记为ci)来修...在满足Σci=t-1的基础上，我们尽量要修最高的学分。我们有siz(x)组物品，每组物品有j-1个，其中第i组物品的第k个物品的体积为k，价值为f[y,k]，背包的总容积为j-1。

我们从每组中选出至多一个物品，即每个子节点最多转移一个状态到x。我们修完x后，还可以选j-1门课，所以我们要在“物品体积”不超过j-1的情况下，使学分（价值）最大。也就是说我们把“上多少y内的课”作为物品。

然后我们进行树形DP，得到答案：

 

void dp(int x){
    f[x][0]=0;
    for(int i=0;i){
        int y=son[x][i];
        dp(y);
        for(int j=m+1;j>=1;t--)//虚拟节点必选，所以是m+1
            for(int k=0;k<=j-1;k++)//组内的物品，第x组第k个物品（选k门课）的体积为k
                f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
    }
}

 

主函数部分，我们读入每个f[i][1]作为第i门课的价值——>解释：从以i为子树的课中选1门——>价值就是f[i][1]。

接下来我们还是继续讨论树形DP。

我们目前做的树形DP题都是在有根树上进行树形DP的，如果题目给定是一棵无根树呢？

难道我们需要对以每个点为根都做一次树形DP统计答案吗？当然不是，要是这么做还不如暴力呢。

一般对于这种题目，我们用二次扫描（换根法）来解决。

这种DP又被称为换根DP，什么？你想要题？

没有，（爆锤狗头）...拿去 Accumulation Degree。

这好像是我第一次给POJ上的题，皮欧勾的题我都没怎么刷过，不过貌似质量挺好的。

但是我永远喜欢洛谷.jpg，emm，你说你看不懂？Chrome自带翻译，不皮了不皮了。

喏。

给你一个树形的水系（废话那么多懒得翻译了）。

有一个有n个节点，n-1条河道的树形水系，每个河道有一个最大容水量c[x][y]c[x][y]表示点x到y的最大容水量，源点可以源源不断出水，以源点作为根节点的树的叶子结点可以无限接纳水，而一个节点水的流量等于流过其儿子节点的水的流量之和，儿子节点水的流量不能超过其与父亲连边的最大容水量，询问最大的源点水流量，n≤2×10^5。

其实简单点说就是求树形结构上的最大流，但是我们不知道源点。为什么不用网络流？嘘，数据太狗了。

于是我们来快乐地DP啊。不给我源点？我每个点都DP一遍。

要是这样能过我还写它干嘛。

不过还是先讲讲怎么DP吧。假设我们知道根节点是x。

那么我们来拆问题了。你看这个问题它又大又烦，不如把它拆掉。

考虑用树形DP拆问题。对于每个节点x，我们发现它只能流向自己的子树，于是可以这样设状态，我们用f[x]表示在以x为根的子树中，以x为源点流向子树的最大流量。然后我们对于每个子树都可以拆成小子树，再拆，再拆...就到叶节点了，问题就得解了。

切，刚刚还那么傲娇的大问题现在还不是被脱的一件不剩，看到DP的魅力了吧？再大的问题，只要你是DP题，我就能把你拆掉。

然后我们就很自然的得到了转移方程式：

这里我不多写一个latex下面的就会炸不知道为什么，你们就当没看见那个error吧...抱歉抱歉

$$f[x]= \sum\limits_{\text{y∈son(x)}} \begin{cases} min(f[x],c(x,y))& \text{x=0}\\ c(x,y)& \text{x!=0} \end{cases}$$

$$f[x]= \sum\limits_{\text{y∈son(x)}} \begin{cases} min(f[x],c(x,y))& \text{x=0}\\ c(x,y)& \text{x!=0} \end{cases}$$

Latex新手写上面那个公式写了半个小时...

DP代码：

void dp(int x){
    vis[x]=1;
    f[x]=0;
    for(int i=head[x];i;i=nxt(i)){
        int y=to(i);
        if(vis[y])continue;
        dp(y);
        if(deg[y]==1)f[x]+=val(i);
        else f[x]+=min(f[y],val(i));
    }
}

 如果我们直接枚举源点...时间复杂度是O(n^2)的，接受不了（不是我接受不了，是出题人接受不了）

（如果我自己要求，O(n^n)的算法我都开开心心用...）

于是我们使用换根DP来O(n)的解决这道题目。

我们任选一个源点作为根，记为rt。然后我们进行一次上面的DP，得到f数组。

设g[x]表示把x作为源点，流向整个水系，流量最大是多少。对于根节点rt，显然有f[rt]=g[rt]。

假设g[x]已经被正确地求出了。开始了，万能的假设法...

我们考虑一下它的子节点y，g[y]我们并不知道，我们来分析一下g[y]。

如果我们把根换成y，那么g[y]包含两部分：

1. 从y流向以y为根的子树的流量，即我们上面算出来的f[x]

2. 从y沿着原父节点x流向水系中的其他部分的流量

为什么可以像2这样流呢？因为我们把根换成y了。这是很多博客没有提到的，容易让人看得很懵逼。

我们把x作为源点的总流量是g[x]，从x流向y的流量就是min(f[y],c(x,y))。

这个很好理解吧，y是x的子节点，x流向子树的最大流量是f[y]，流量限制是c(x,y)，哪个小流量就是哪个。

所以从x流向除了y以外其他部分的流量就是两者之差。这个也很好理解吧。

于是我们把y作为源点，先流向x，再流向其他部分的流量就是两者之差和c(x,y)之间的最小值。

同样的，对于度数为1的x节点，我们也要特判——>节点x没法流向其他地方了。

于是我们得到了g[x]的计算方法：

又要写Latex...我裂开了

$$g[y]=f[y]+ \begin{cases} min( g[x]-min( f[y], c(x,y) ), c( x,y ) )& \text{x的度数>1}\\ c(x,y)& \text{x的度数=1} \end{cases}$$

g[y]就是把根从x换成y后流量的计算结果，由于这是一个从上至下的递推方程，所以我们可以通过一遍dfs得出g数组。

又到了你们最爱的放代码时间：

 

void dfs(int x){
    vis[x]=1;
    for(int i=head[x];i;i=nxt(i)){
        int y=to(i);
        if(vis[y])continue;
        if(deg[x]==1)g[y]=f[x]+val(i);
        else g[y]=f[y]+min(g[x]-min(f[y],val(i)),val(i));
        dfs(y);
    }
}

 

解释一下，val(i)就是上面说的c(x,y)，我把它当作边权存了起来。

然后我们在main函数里面这样搞就可以得到答案了。

int rt=1;
dp(rt);
memset(vis,0,sizeof vis);
g[rt]=f[rt];
dfs(rt);
int ans=-1;
for(int i=1;i<=n;i++)
    ans=max(ans,g[i]);

OK，那么希望你们通过这道例题，对换根DP有了一个初步的了解。

其实换根法的思想就是通过差值来更新答案，而不必一遍遍枚举计算重复信息。

那么，我们中篇再见（咕咕咕）。