NOIP2018普及T4对称二叉树-基于回文子串判断的的更优解法

最新推荐文章于 2022-01-19 14:37:31 发布

寻夏先生

最新推荐文章于 2022-01-19 14:37:31 发布

阅读量319

点赞数

分类专栏：算法与OI

本文链接：https://blog.csdn.net/qq_42761323/article/details/88909659

版权

算法与OI 专栏收录该内容

3 篇文章 1 订阅

订阅专栏

NOIP2018普及T4对称二叉树-基于回文子串判断的的更优解法

题目：https://www.luogu.org/problemnew/show/P5018

NOIP2018普及组T4对称二叉树有着很像暴力的O(nlogn)正解。用树哈希可以优化到O(n),但是hash这类玄学算法毕竟让人不是那么舒服，因此天才的我想出了另一个**O(n)**算法，思路如下：

（1）dfs一遍，存下这棵树节点值的中序遍历和节点度的中序遍历。设为序列v[]和序列s[]。

（2）同时在这两个序列上跑Manancher算法，即向两边走的时候要求( ( v[i-p[i]] == v[i+p[i]] ) && ( s[i-p[i]] == s[i+p[i]] ) )。并且由于一棵子树只会是奇回文子串，因此不需要在字符间插入空字符。

（3）对以序列中的每一个位置为中心的回文串打个擂台，即ans=max(ans,p[i])。

这种思路好比把树压扁成一个序列，再用熟悉的Manancher解决，~~我真是个聪明的小机灵鬼~~。

#include<iostream>
#include<iomanip>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n=0,tot=0,vtmp[1000100],left_son[1000100],right_son[1000100];
int v[1000100],s[1000100],p[1000100];
void read()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&vtmp[i]);
    }
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&left_son[i],&right_son[i]);
    }
    return;
}
int dfs(int x)
{
    int cnt=0,tmp=0;
    if (left_son[x] != -1) cnt+=dfs(left_son[x]);
    tot++;tmp=tot;v[tot]=vtmp[x];
    if (right_son[x] != -1) cnt+=dfs(right_son[x]);
    s[tmp]=cnt;
    return cnt+1;
}
int manancher()
{
    int mx=0,id=0,ans=0;
    for (int i=1;i<=n;i++)
    {
        if (i<=mx)
        {
            p[i]=min(p[2*id-i],mx-i);
        }
        else
        {
            p[i]=1;
        }
        while ( ( v[i-p[i]] == v[i+p[i]] ) && ( s[i-p[i]] == s[i+p[i]] ) )
        {
            if ( ((i-p[i])<1) || ((i+p[i]>n)) ) break;
            p[i]++;
        }
        if (mx<p[i]+i)
        {
            id=i;
            mx=p[i]+i;
        }
        if (2*p[i]-1==s[i]+1) ans=max(ans,2*p[i]-1);
    }
    return ans;
}
int main()
{
    read();
    dfs(1);
    printf("%d\n",manancher());
    return 0;
}

Manancher算法讲解：https://segmentfault.com/a/1190000008484167