团建 | 第十五届蓝桥杯省赛C++A组

小蓝正在和朋友们团建，有一个游戏项目需要两人合作，两个人分别拿到一棵大小为 n 和 m 的树，树上的每个结点上有一个正整数权值。

两个人需要从各自树的根结点 1 出发走向某个叶结点，从根到这个叶结点的路径上经过的所有结点上的权值构成了一个正整数序列，两人的序列的最长公共前缀即为他们的得分。

给出两棵树，请计算两个人最多的得分是多少。

输入格式

输入的第一行包含两个正整数 n,m，用一个空格分隔。

第二行包含 nn 个正整数 c1,c2,…,cn，相邻整数之间使用一个空格分隔, 其中 ci 表示第一棵树结点 ii 上的权值。

第三行包含 mm 个正整数 d1,d2,…,dm，相邻整数之间使用一个空格分隔，其中 di 表示第二棵树结点 i 上的权值。

接下来 n−1 行，每行包含两个正整数 ui,vi 表示第一棵树中包含一条 ui 和 vi之间的边。

接下来 m−1 行，每行包含两个正整数 pi,qi 表示第二棵树中包含一条 pi 和 qi之间的边。

输出格式

输出一行包含一个整数表示答案。

数据范围

对于 20%20% 的评测用例，1≤n,m≤500；
对于所有评测用例，1≤n,m≤2×105，1≤ci,di≤108，1≤ui,vi≤n，1≤pi,qi≤m，对于任意结点，其儿子结点的权重互不相同。

输入样例1：

2 2
10 20
10 30
1 2
2 1

输出样例1：

1

输入样例2：

5 4
10 20 30 40 50
10 40 20 30
1 2
1 3
2 4
3 5
1 2
1 3
3 4

输出样例2：

2

样例解释

在第一个样例中，两个序列可以为 [10,20],[10,30]，最大前缀为 1;

在第二个样例中，两个序列可以为 [10,20,40],[10,20,30]，最大前缀为 2。

题解：

        采用邻接矩阵存储树，额外开两个数组记录路径。DFS遍历寻找判断即可。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<queue>
#include<stack>
#include<vector>
#include<unordered_set>
#include<unordered_map>
#include<map>
#include<set>
using namespace std;
typedef long long int ll;
const int INF=1e16;

int n,m;
int c[200005],d[200005];
vector<vector<int>> v1(200005);
vector<vector<int>> v2(200005);
vector<int> r1(200005,0);
vector<int> r2(200005,0);
int maxcnt=0;

void dfs(int t1,int t2,int cnt){
    if(cnt>maxcnt){
        maxcnt=cnt;
    }
    //cout << "1" << "\n";
    for(int i=0;i<v1[t1].size();i++){
        for(int j=0;j<v2[t2].size();j++){
            if(r1[v1[t1][i]]!=1 && r2[v2[t2][j]]!=1 && c[v1[t1][i]]==d[v2[t2][j]]){
                r1[v1[t1][i]]=1;r2[v2[t2][j]]=1;
                dfs(v1[t1][i],v2[t2][j],cnt+1);
                r1[v1[t1][i]]=0;r2[v2[t2][j]]=0;
            }
        }
    }
}

void solve(){
    cin >> n >> m;
    for(int i=1;i<=n;i++){
        cin >> c[i];
    }
    for(int i=1;i<=m;i++){
        cin >> d[i];
    }
    for(int i=1;i<n;i++){
        int x,y;
        cin >> x >> y;
        v1[x].push_back(y);
        v1[y].push_back(x);
    }
    for(int i=1;i<m;i++){
        int x,y;
        cin >> x >> y;
        v2[x].push_back(y);
        v2[y].push_back(x);
    }
    if(c[1]!=d[1]){
        cout << 0;
        return;
    }
    r1[1]=1;
    r2[1]=1;
    dfs(1,1,1);
    cout <<maxcnt;
}

int main(){
    solve();
}