标签：背包 int siz 复杂度 树形 P3177 HAOI2015 子树
惊奇的发现躺在任务计划里几个月的“\(P3177\)树上染色”变成蓝题了，既然前几天清北学堂提到这题了，那就看一眼。（众所周知，洛谷任务计划是个栈）
本人十分不擅长\(DP\)，尤其像什么树形背包题，尤其是为啥看上去 \(n^3\) 的复杂度都说是 \(n^2\) 的，让我十分害怕，因为别人说的经典句子什么"两个点只会在 \(LCA\) 处合并"，这 \(nm\) 我怎么总感觉是错的啊。
我曾经以为的优化就是指枚举到子树大小与父亲的 \(min\) 即可，但很长时间以来都被这个玄学的复杂度所困扰，因为这明显一条链可以卡的掉啊，换上链的复杂度是 \(n*(n-1)+(n-1)*(n-2)+(n-2)*(n-3)...+2*1\)，难道这东西确实是 \(n^2\) 的？（雾。
我信你个鬼，很明显不是啊~。。我个大彩笔。
那么产生分歧的地方很明显：
我做的时候是将整个树的 \(siz[rt]\) 去与子树的 \(siz[v]\) 进行合并，它张这样：
大家都说只会在\(LCA\)处合并，相互产生贡献的肯定是像下面这样的：
每个子树都不交叉，它们的\(LCA\)都是这棵子树的根，设这个子树的第 \(i\) 个儿子大小为 \(siz[i]\)，那么每次的复杂度就是
\(siz[2]*siz[1]+siz[3]*(siz[2]+siz[1])+siz[4]*(siz[3]+siz[2]+siz[1])+...+siz[n]*(siz[n-1]+siz[n-2]+...+siz[1])\)
\(=\sum\limits_{i=1}^n \sum\limits_{j=1}^n siz[i]*siz[j] \ \ (i!=j)\)
就是两两子树相乘。既然每个点当根的复杂度不好证，那么就考虑将每个 \(siz\) 看做这个子树中的每个点，看整棵树中所有的点对复杂度的贡献。
考虑到子树不交叉，那么这棵树的任意两个点对背包进行更新答案时，肯定要保证这两个点：1、都在当前根的子树中；2、不在当前根的同一个儿子子树中。那么显然，这个根是它俩的 \(LCA\)，是这个意思。写的好的话复杂度基本 \(n^2\) 不多不少。
所以，为了保证 \(n^2\) ，我们不能提前处理出所有子树的大小，而是边合并边做背包。
关于树形背包的心路历程呢还得看这道题，就是刚刚从任务计划里出来的（从栈底开了个洞钻出来的）这道\(P3177\)。
有些树形背包题数据范围是 \(n^3\) 的，不用考虑这些。这题是 \(n^2\)，而且最近洛谷放进去一个加强数据，果然就是一条链，卡掉了一部分题解。
由于数据加强了，过会儿考虑发给洛谷一个题解，咕值快要掉出前100了，qwq。
题意：
给你一颗 \(n\) 个点带边权的树，要你将 \(k\) 个点涂成黑色，其余点为白色，整棵树的价值为任意两个黑点的距离之和，加上任意两个白点的距离之和。\((0\leq k\leq n\leq 2000)\)
一看数据范围就很友善（雾，显然没有 \(O(n)\) 的算法或者其他神奇的乱搞，所以考虑 \(dp\)，\(f[i][j]\)表示 \(i\) 子树中选了 \(j\) 个黑点的最。。。（愣住）。发现子树最优解并不独立，它还会与其他点产生贡献，我们需要考虑全局最优。
既然任意两点之间的贡献很难处理，那我们将所有的贡献转移到边上，突然就发现简单轻松了许多：一条边的贡献为两边的黑点数相乘加上白点数相乘。
而且恰好，在枚举子树内有多少黑点时，我们同样知道子树外有多少黑点（因为整棵树正好要选 \(k\) 个），还知道子树内白点个数，以及子树外白点个数，这条连接当前根与儿子的边的贡献就可以算出来了。
考虑这样做的可行性，因为一条边我们已经固定了这棵子树内黑点的个数，而外面的黑点无论在什么位置，这条边的价值是不变的。所以我们放心地让每棵子树内的边权和最大，最后一定会是最优解。
于是来到了树形背包的环节，接下来是几分钟就能写好的代码：
#include <bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int n,m;
struct E{ int to,we; };
vector <E> e[2333];
int siz[2333];
int f[2333][2333];
inline int read() {
int sum = 0, f = 1; char c = getchar();
while(c<'0' || c>'9') { if(c=='-') f = -1; c = getchar(); }
while(c>='0'&&c<='9') { sum = sum * 10 + c - '0'; c = getchar(); }
return sum * f;
}
inline int MIN(int aa,int bb) { return aa<bb?aa:bb; }
void DFS(int u,int fa) {
siz[u] = 1;
for(int i=0;i<e[u].size();i++) {
int v = e[u][i].to;
if(v==fa) continue;
DFS(v,u);
siz[u] += siz[v];
}
f[u][0] = f[u][1] = 0;
for(int i=0;i<e[u].size();i++) {
int v = e[u][i].to, w = e[u][i].we;
if(v==fa) continue;
for(int j=MIN(siz[u],m); ~j; --j) {
for(int k=0; k<=MIN(siz[v],j); k++) {
int val = w * ( k*(m-k) + (siz[v]-k)*(n-m-siz[v]+k) );
f[u][j] = max( f[u][j], f[u][j-k]+f[v][k]+val );
}
}
}
}
int main () {
memset(f,-0x3f,sizeof(f));
int x,y,z;
n = read(); m = read();
for(int i=1;i<n;i++) {
x = read(); y = read(); z = read();
e[x].push_back( (E){y,z} );
e[y].push_back( (E){x,z} );
}
DFS(1,0);
cout<<f[1][m];
return 0;
}
如果树形背包不太熟练的人，像我，很有可能发现自己的代码交上去WA了或TLE了，然而这种题就算下载数据再小也是基本没法调试的，建议WA或TLE的同学看看下面这三条：
1、由于树形背包是已经滚掉一维的（枚举到当前第几个儿子），我们枚举j要倒着枚举，因为我们一直用小的 \(j\) 更新大的 \(j\) ，如果先更新了小的 \(j\) ，可能会出现重复选取的情况。
2、再个就是枚举 \(v\) 中的黑点 \(k\) ，直觉上来看按说看似这层正着倒着都行，因为 \(j\) 是倒着枚举的，比 \(j\) 大的状态是已经合并了 \(v\) 子树的（上一层状态），比 \(j\) 小的状态一定是没有合并 \(v\) 子树的（本层状态），用 \(f[u][j-k]\) 去更新 \(f[u][j]\)，肯定是用没有合并的状态去更新合并后的状态。但是，这题得考虑个东西：当 \(k=0\) 的情况，会出现这么一种转移：
\(f[u][j]=max(f[u][j],f[u][j]+f[v][k]+val)\)。
你以为你的 \(f[u][j-k]\) 是上一层的状态？它可能已经被更新成这一层了！所以 \(f[u][j]\) 确实需要先转移一下，否则被新的一层更新掉之后，你就再也找不到这个状态了，而且不光找不到，还会把本层的 \(f[u][j]\) 进行重复转移，与上述第一条的原理类似，相当于重复选取，只不过看上去 \(k=0\) 没有选黑点，但是白点也是有贡献的啊！
所以，在做DP题尤其像为了这种省空间的背包题时，要滚掉一维的话细节是要仔细考虑的，否则丢掉上一层状态和同层状态互相转移是经常会发生的，如果实在认怂，就开成 \(f[u][j][0/1]\) 这样就不用担心上述 1 与 2 的枚举顺序了，不过貌似也没人这么做。
3、洛谷新增加了一组数据，好多份题解都TLE了，但不代码他们代码就是错的，因为树形背包题随机数据上述写法是很快的。但是有一种数据可以卡掉：一条链。如果每次j都枚举到 \(siz[u]\) 的话，复杂度确实是有问题的，为了让复杂度成为树形背包中常说的那句：“任意两个点在其LCA处合并”，我们不能提前处理出子树的 \(siz\)，而是边做背包边合并，这样可以保证两个点会在不同的儿子子树之间合并。
这种写法由于是将做差转移改为了求和转移，我们也不用考虑 \(f[u][j-k]\) 的状态不存在的情况了，就不用赋初值了。仿佛更好写了。
下面是AC代码：
#include <bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int n,m;
struct E{ int to,we; };
vector <E> e[N];
int siz[N];
int f[2333][2333];
inline int read() {
int sum = 0, f = 1; char c = getchar();
while(c<'0' || c>'9') { if(c=='-') f = -1; c = getchar(); }
while(c>='0'&&c<='9') { sum = sum * 10 + c - '0'; c = getchar(); }
return sum * f;
}
inline int MIN(int aa,int bb) { return aa<bb?aa:bb; }
void DFS(int u,int fa) {
siz[u] = 1;
for(int i=0;i<e[u].size();i++) {
int v = e[u][i].to, w = e[u][i].we;
if(v==fa) continue;
DFS(v,u);
for(int j=siz[u]; ~j; --j) {
if(m<j) continue;
for(int k=MIN(m-j,siz[v]); ~k; --k) {
int val = w * ( (n-m-siz[v]+k)*(siz[v]-k) + (m-k)*k );
f[u][j+k] = max( f[u][j+k], f[u][j] + f[v][k] + val );
}
}
siz[u] += siz[v];
}
}
int main () {
int x,y,z;
n = read(); m = read();
for(int i=1;i<n;i++) {
x = read(); y = read(); z = read();
e[x].push_back( (E){y,z} );
e[y].push_back( (E){x,z} );
}
DFS(1,0);
cout<<f[1][m];
return 0;
}
如果是像我这样树形背包不熟练的同学，建议将每次\(WA\)的原因找出来，而不是乱改了一番\(AC\)了之后就走了。否则如果理解不是很清晰的话（这种题细节还是蛮多的说），下次再遇到此类的题，一遍\(AC\)就像冒险一样。
标签：背包,int,siz,复杂度,树形,P3177,HAOI2015,子树 来源： https://www.cnblogs.com/clever-sheep/p/12870572.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。