I - bLue的二叉树（DFS）

Description

Keke 是一个喜爱种树的人，他对各种树都有很深的研究。

MLE 听说 bLue 种了一些新品种的树，就想邀请 Keke 去围观一下。

PBH 在暗中把这一切尽收眼底，作为资深植树行家，他虽不屑，但也决定和他们一起去看一看。

于是，大家便一起到了 bLue 家去看树。

bLue 有两棵二叉树，分别有 n 和 m 个节点，编号分别为 1-n 和 1-m，每个节点都有一个权值，bLue 想知道第一棵树的所有子树中与第二棵树完全相同的个数(不考虑节点编号)。

Input

输入数据有多组（数据组数不超过 150），到 EOF 结束。
对于每组数据：
第一行有两个整数 n (0 < n <= 10^5) 和 m (0 < m <= 10^5)，表示第一棵树和第二棵树的节点个数
接下来 n 行，表示第一棵树：第 i (0 < i <= n) 行有 3 个整数，w[i] (0 < w[i] <= 10), lc[i], rc[i] (0 < lc[i], rc[i] <= n)，分别表示节点 i 的权值，该节点的左孩子编号和右孩子编号，若某个孩子不存在，则为 0 (数据保证每棵树都是合法的有根二叉树)
接下来 m 行，表示第二棵树：格式同第一棵树
保证：树的最大深度不会超过 10000。

Output

对于每组数据，输出一行一个整数 num，表示第一棵树的所有子树中与第二棵树完全相同的个数。

Sample

Input

7 4
1 6 3
2 0 4
1 7 0
3 0 0
1 2 1
2 0 0
2 0 0
2 0 0
1 4 0
1 1 2
2 0 0
3 3
1 0 0
2 1 3
3 0 0
1 0 3
2 1 0
3 0 0

Output

1
0

Hint

数据量比较大，推荐用 scanf 等读入。

答案：

#include <bits/stdc++.h>
#define ll long long
const int N = 1e5 + 10;
using namespace std;

struct node
{
    int k;
    int l;
    int r;
} a[N],b[N];

int avis[N];
int bvis[N];
int ka;
int kb;
int pa[3*N];
int pb[3*N];
int Next[3*N];

void ADFS(int x)
{
    pa[ka++]=a[x].k;
    if(!x)
        return ;
    ADFS(a[x].l);
    ADFS(a[x].r);
}

void BDFS(int x)
{
    pb[kb++]=b[x].k;
    if(!x)
        return ;
    BDFS(b[x].l);
    BDFS(b[x].r);
}

void get_Next(int n)
{
    int j=0;
    int i=1;
    Next[1]=0;
    while(i<=n)
    {
        if(!j||pb[i]==pb[j])
        {
            i++;
            j++;
            if(pb[i]==pb[j])
                Next[i]=Next[j];
            else
                Next[i]=j;
        }
        else
            j=Next[j];
    }
}

int Kmp(int n,int m)
{
    int i=1;
    int j=1;
    int cnt=0;
    while(i<=n)
    {
        if(!j||pa[i]==pb[j])
        {
            i++;
            j++;
        }
        else
            j=Next[j];
        if(j>=m+1)
        {
            cnt++;
            j=Next[j];
        }
    }
    return cnt;
}

int main()
{
    ios::sync_with_stdio(0);
    int n,m;
    while(cin>>n>>m)
    {
        memset(avis,0,sizeof(avis));
        memset(bvis,0,sizeof(bvis));
        int i;
        for(i=1; i<=n; i++)
        {
            cin>>a[i].k>>a[i].l>>a[i].r;
            avis[a[i].l]++;
            avis[a[i].r]++;
        }
        for(i=1; i<=m; i++)
        {
            cin>>b[i].k>>b[i].l>>b[i].r;
            bvis[b[i].l]++;
            bvis[b[i].r]++;
        }
        ka=kb=1;
        for(i=1; i<=n; i++)
        {
            if(!avis[i])
            {
                ADFS(i);
                break;
            }
        }
        for(i=1; i<=m; i++)
        {
            if(!bvis[i])
            {
                BDFS(i);
                break;
            }
        }
        get_Next(kb-1);
        int ans=Kmp(ka-1,kb-1);
        cout<<ans<<endl;
    }
    return 0;
}