P5018 [NOIP2018 普及组] 对称二叉树
这个题可以用树hash做
怎么搞呢,对于一棵树,我们要用hash记录他的形态
对于二叉树,我们要记录的就是左儿子的hash和自己和右儿子。
然后进行比较
我采用的是对于左右儿子和自己乘上不同质数的方式
还有自然溢出,以及两次hash
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
template<class T>inline void read(T &x)
{
x=0;register char c=getchar();register bool f=0;
while(!isdigit(c))f^=c=='-',c=getchar();
while(isdigit(c))x=(x<<3)+(x<<1)+(c^48),c=getchar();
if(f)x=-x;
}
template<class T>inline void print(T x)
{
if(x<0)putchar('-'),x=-x;
if(x>9)print(x/10);
putchar('0'+x%10);
}
unsigned long long prime[4];
int n;
int x,y;
int ans;
int sum[10000006];
int ls[1000006],rs[1000006];
unsigned long long has[1000006],val[10000006],Has[1000006],ha[10000006],Ha[10000006];
unsigned long long mod=89999794200117649;
unsigned long long modd=999999786000011449;
void dfs(int no){
if(ls[no]!=-1) dfs(ls[no]);
if(rs[no]!=-1) dfs(rs[no]);
sum[no]=sum[ls[no]]+sum[rs[no]]+1;
if(sum[ls[no]]==sum[rs[no]]&&has[ls[no]]==Has[rs[no]]&&ha[ls[no]]==Ha[rs[no]]){
ans=max(ans,sum[no]);
}
has[no]=has[ls[no]]*prime[0]+val[no]*prime[1]+has[rs[no]]*prime[2];
has[no]%=mod;
Has[no]=Has[rs[no]]*prime[0]+val[no]*prime[1]+Has[ls[no]]*prime[2];
Has[no]%=mod;
ha[no]=ha[ls[no]]*prime[2]+val[no]*prime[1]+ha[rs[no]]*prime[0];
ha[no]%=modd;
Ha[no]=Ha[rs[no]]*prime[2]+val[no]*prime[1]+Ha[ls[no]]*prime[0];
Ha[no]%=modd;
}
int main(){
prime[0]=20041204;
prime[1]=299999827;
prime[2]=999999751;
read(n);
for(int i=1;i<=n;++i){
read(val[i]);
}
for(int i=1;i<=n;++i){
read(x);read(y);
ls[i]=x;rs[i]=y;
}
dfs(1);
cout<<ans;
return 0;
}