题目大意
有一种生成\(n\)个点的树的方法为：
一开始有一个点，\(n-1\)次操作，每次可以有两种操作：1.选一个点，用一条红边将它与新点连接；2.将新点放在一条红边上，新点与这条红边两端点直接的连边变成蓝色
给出一个\(n(n\leq 2\times10^5)\)个点的树，问如何分配边的颜色使这棵树能用上述方法生成，而且蓝边权值之和最大
题解
（想要简明且优秀的题解点这里）
考虑那种生成方法，发现每条蓝边肯定是由一条红边“分裂”的，那就可以把“连红边”和“分裂红边”放在一起考虑
将初始点当根，那么想要生成蓝边时，一定会生成一条长度为2的蓝色直链
即确定根之后，存在一种方案将所有蓝边划分成若干条长为2的直链
考虑换根dp，发现不想写
先随便找一个点当“假根”
发现在考虑点\(u\)时，根具体在哪不影响答案，但根是在 \(u\)除自己以外的子树中 还是 在\(u\)或\(u\)的子树外 会影响过点\(u\)的蓝直链形态
\(u\)是否是蓝直链的中点也会影响
那么就可以设\(f(u,i,j)\)表示考虑点\(u\)时，根在/不在\(u\)除自己以外的子树中，且\(u\)是/不是中点
又发现当根在\(u\)除自己以外的子树中，且\(u\)为中点，且过它的蓝直链在假根看来是弯链，这样它的转移就会和 过它的蓝直链在假根看来是弯链 不同
所以\(u\)就会有5个状态：\(f(u,0,0),f(u,0,1),f(u,1,0),f(u,1,1)\)（过它的直链在假根看来是弯链）\(,f(u,1,2)\)（过它的直链在假根看来是直链）
多乎哉？不多也
下面考虑如何得到\(f(u,i,j)\)
发现得到\(f(u,1,1)\)需要像儿子中连2条作为过它的蓝直链的边，\(f(u,0,1),f(u,1,2)\)需要连1条作为过它的蓝直链的边，那就需要把“实际连了几条边”记在状态里
当根在\(u\)除自己以外的子树中时，还要记当前是否有根出现在儿子的子树中
所以为了求出\(f(u,i,j)\)，要再给\(u\)设\(g(u,i,j,k)\)，其中\(k\)表示是否已经有根出现在子树中：\(g(u,0,0,k)\)，\(g(u,0,1,k)\)，\(g(u,0,2,k)\)，\(g(u,1,0,k)\)，\(g(u,1,1,k)\)，\(g(u,1,2,k)\)，\(g(u,1,3,k)\)，\(g(u,1,4,k)\)，\(g(u,1,5,k)\)
其中：
\(g(u,0,0,k)\)同\(f(u,0,0)\)
\(g(u,0,1,k),g(u,0,2,k)\)表示想要达到\(f(u,0,1)\)的状态，实际连了0或1条作为过它的蓝直链的边
\(g(u,1,0,k)\)同\(f(u,1,0)\)
\(g(u,1,1,k),g(u,1,2,k),g(u,1,3,k)\)表示想要达到\(f(u,1,1)\)的状态，实际连了0或1或2条作为过它的蓝直链的边
\(g(u,1,4,k),g(u,1,5,k)\)表示想要达到\(f(u,1,2)\)的状态，实际连了0或1条作为过它的蓝直链的边
多乎哉？不多也
接下来考虑转移，设\(v\)为当前考虑到的\(u\)的儿子
要注意的是，\(v\)作为中点且过\(v\)的蓝直链在假根看来也是直链时，\(u,v\)之间必须为蓝边
如果\(u,v\)都不是中点，那么它们之间的边不能为红
具体转移见代码
最后的答案是\(max( f(假根,0, 0), f(假根, 1, 0), f(假根, 1, 1) )\)，因为假根没有父亲，而\(f(u,0,1),f(u,1,2)\)需要\(u\)与\(u\)的父亲之间连蓝边
代码
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#define rep(i,x,y) for(register int i=(x);i<=(y);++i)
#define dwn(i,x,y) for(register int i=(x);i>=(y);--i)
#define view(u,k) for(int k=fir[u];~k;k=nxt[k])
#define maxn 200010
#define maxm (maxn<<1)
#define inf 2147483647
using namespace std;
int read()
{
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)&&ch!='-')ch=getchar();
if(ch=='-')f=-1,ch=getchar();
while(isdigit(ch))x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
return x*f;
}
void write(int x)
{
if(x==0){putchar('0'),putchar('\n');return;}
int f=0;char ch[20];
if(x<0)putchar('-'),x=-x;
while(x)ch[++f]=x%10+'0',x/=10;
while(f)putchar(ch[f--]);
putchar('\n');
return;
}
int n,g[3][6][2],t[3][6][2],f[maxn][2][3],fir[maxn],v[maxm],nxt[maxm],cnt,w[maxm];
void ade(int u1,int v1,int w1){v[cnt]=v1,nxt[cnt]=fir[u1],w[cnt]=w1,fir[u1]=cnt++;}
int add(int x,int y){if(x==-inf||y==-inf)return -inf;return x+y;}
#define upd(x,y) x=max(x,y)
void getans(int u,int fa)
{
view(u,k)if(v[k]!=fa)getans(v[k],u);
rep(i,0,1)rep(j,0,5)rep(k,0,1)g[i][j][k]=-inf;
g[0][0][0]=g[0][1][0]=g[1][0][0]=g[1][1][0]=g[1][4][0]=0;
view(u,k)if(v[k]!=fa)
{
rep(i,0,2)t[0][i][0]=g[0][i][0];
rep(i,0,5)rep(l,0,1)t[1][i][l]=g[1][i][l];
//一、边(u,v)为红：
//1.根不在v子树中
rep(i,0,2)upd(g[0][i][0],add(t[0][i][0],f[v[k]][0][0]));
rep(i,0,5)rep(l,0,1)upd(g[1][i][l],add(t[1][i][l],f[v[k]][0][0]));
//2.根在v子树中
//当过u的蓝直链在假根看来是直链时，在v中的根看来一定是弯链，所以没有到g(u,1,1/2/3/4/5,0/1)的转移
upd(g[1][0][1],add(t[1][0][0],max(f[v[k]][1][1],f[v[k]][1][0])));
//二、边(u,v)为蓝
//1.根不在v子树中
rep(i,0,2)upd(g[0][i][0],add(add(t[0][i][0],f[v[k]][0][1]),w[k]));
rep(i,0,5)rep(l,0,1)upd(g[1][i][l],add(add(t[1][i][l],f[v[k]][0][1]),w[k]));
rep(l,0,1)
{
upd(g[0][2][l],add(add(t[0][1][l],f[v[k]][0][0]),w[k]));
upd(g[1][2][l],add(add(t[1][1][l],f[v[k]][0][0]),w[k]));
upd(g[1][3][l],add(add(t[1][2][l],f[v[k]][0][0]),w[k]));
upd(g[1][5][l],add(add(t[1][4][l],f[v[k]][0][0]),w[k]));
}
// 2.根在v子树中
upd(g[1][0][1],add(add(t[1][0][0],f[v[k]][1][2]),w[k]));
upd(g[1][2][1],add(add(t[1][1][0],max(f[v[k]][1][0],f[v[k]][1][1])),w[k]));
upd(g[1][3][1],add(add(t[1][2][0],max(f[v[k]][1][0],f[v[k]][1][1])),w[k]));
upd(g[1][5][1],add(add(t[1][4][0],max(f[v[k]][1][0],f[v[k]][1][1])),w[k]));
}
f[u][0][0]=g[0][0][0],f[u][0][1]=g[0][2][0],f[u][1][0]=max(0,g[1][0][1]),
//f(u,1,0)实际意义是“根在u子树中除u以外的区域时的答案” （即u没儿子时为-inf）
//但当它转移到u的父亲时，它的意义是 “根在u子树中时的答案”（即u没儿子时为0）
//而且最终答案最小值为0，所以在这里将它改为0不影响统计答案
f[u][1][1]=g[1][3][1],f[u][1][2]=g[1][5][1];
}
int main()
{
memset(fir,-1,sizeof(fir));
n=read();
rep(i,1,n-1){int x=read(),y=read(),z=read();ade(x,y,z),ade(y,x,z);}
getans(1,0);
write(max(f[1][0][0],max(f[1][1][0],f[1][1][1])));
return 0;
}
/*
5
1 2 10
1 3 40
1 4 15
1 5 20
*/一些感想
sb猎人公会nmsl