博客园 首页 私信博主 显示目录 隐藏目录 管理 动画

Codeforces.741D.Arpa’s letter-marked tree and Mehrdad’s Dokhtar-kosh paths(dsu on tree 思路)

题目链接

\(Description\)

给定一棵树,每条边上有一个字符(a~v)。对每个节点,求它的子树中一条最长的路径,满足 路径上所有边上的字符可以重新排列成一个回文串。输出其最长长度。
\(n\leq 5\times10^5\)

\(Solution\)

可以构成回文串,即要么所有字符都出现了偶数次,要么有一个出现了奇数次、其余都出现了偶数次。
转化为异或!把每个字符c(0~21)映射到1<<c上去。
\(s[x]\)表示根节点到\(x\)路径上边权的异或和。那么路径\((u,v)\)满足条件当且仅当\(s[u]\ xor\ s[v]\)等于\(0\)或是某个二次幂。
而路径\((u,v)\)的答案是\(dep[u]+dep[v]-dep[LCA]*2\)。在LCA处计算,这样只需要对每个状态求它最大的\(dep\)
而且更新时只有23种方式(对于\(s[v]\),可以从\(\max\{dep[s[v]]\}\)\(\max\{dep[s[v]\ xor\ 2^i]\}\)更新)。
dsu on tree求每个子树的\(\max\{dep[s]\}\)就好了。
复杂度\(O(23n\log n)\)

//608ms	79100KB
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define MAXIN 300000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=5e5+5,INF=0x3f3f3f3f;

int Enum,H[N],nxt[N],to[N],ch[N],s[N],f[(1<<22)+2],L[N],R[N],A[N],dep[N],sz[N],son[N],Ans[N];
char IN[MAXIN],*SS=IN,*TT=IN;

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
inline void AE(int u,int v,int c)
{
	to[++Enum]=v, nxt[Enum]=H[u], ch[Enum]=c, H[u]=Enum;
}
void DFS1(int x)
{
	static int Index=0;
	A[L[x]=++Index]=x;
	int mx=0; sz[x]=1;
	for(int i=H[x],v; i; i=nxt[i])
		dep[v=to[i]]=dep[x]+1, s[v]=s[x]^ch[i], DFS1(v), sz[x]+=sz[v], sz[v]>mx&&(mx=sz[v],son[x]=v);
	R[x]=Index;
}
inline int Add(int s,int d,int delta)
{
	int ans=f[s]+d-delta;
	for(int i=0; i<22; ++i) ans=std::max(ans,f[s^(1<<i)]+d-delta);//d[u]+d[v]-d[LCA]*2
	return ans;
}
void DFS2(int x,int keep)
{
	int ans=0;
	for(int i=H[x]; i; i=nxt[i]) if(to[i]!=son[x]) DFS2(to[i],0),ans=std::max(ans,Ans[to[i]]);
	if(son[x]) DFS2(son[x],1),ans=std::max(ans,Ans[son[x]]);

	ans=std::max(ans,Add(s[x],0,dep[x])), f[s[x]]=std::max(f[s[x]],dep[x]);
	for(int i=H[x],v,delta=dep[x]<<1; i; i=nxt[i])
		if((v=to[i])!=son[x])
		{
			for(int j=L[v]; j<=R[v]; ++j) ans=std::max(ans,Add(s[A[j]],dep[A[j]],delta));
			for(int j=L[v]; j<=R[v]; ++j) f[s[A[j]]]=std::max(f[s[A[j]]],dep[A[j]]);
		}
	Ans[x]=ans;
	if(!keep) for(int i=L[x]; i<=R[x]; ++i) f[s[A[i]]]=-INF;
}

int main()
{
	int n=read();
	for(int i=2,x,c; i<=n; ++i)
	{
		x=read(),c=gc(); while(!isalpha(c)) c=gc();
		AE(x,i,1<<c-'a');
	}
	memset(f,-0x3f,sizeof f);//没有的值不能用0更新 
	DFS1(1), DFS2(1,1);
	for(int i=1; i<=n; ++i) printf("%d ",Ans[i]);

	return 0;
}
posted @ 2018-11-25 20:46  SovietPower  阅读(150)  评论(0编辑  收藏  举报