并不对劲的bzoj3677:p3647:[APIO2014]连珠线

题目大意

有一种生成$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

*/

一些感想

换根dp可以记一位状态表示“真的根在不在该子树里”。