题目链接
 143. 最长同值路径(卡码网周赛第二十五期(23年B站笔试真题))
题目描述
给定一个二叉树的 root ,返回最长的路径的长度,这个路径中的每节点具有相同值。这条路径可以经过也可以不经过根节点。两个节点之间的路径长度 由它们之间的边数表示。
树的节点数的范围是 [0,10^4] -1000 <= Node.val <= 1000
树的深度将不超过 18 层
输入
输入共两行,第一行是一个整数 n,表示第二行的字符串数。
第二行包含 n 个字符串,空格隔开,数字的字符串代表该节点存在,并且值为数字,null 代表是一个空结点。
输出
输出一个正整数,代表最长路径长度。
样例1输入
7
5 4 5 1 1 null 5
样例1输出
2
样例1提示
通过层序遍历构建二叉树如下:
样例2输入
9
7 2 8 null 4 null 6 4 5
样例2输出
1
样例2提示
通过层序遍历构建二叉树如下:
题解1(C++版本)
#include<bits/stdc++.h>
using namespace std;
 
const int N = 1e4 + 10;
 
int n, val[N]; // val[i]表示编号为i的节点的值
char s[10];
vector<int> edge[N];
int dp[N][2]; // dp[i][0/1]表示根节点编号为i,沿左/右子树这边走的最长路径长度
 
int ans = 0;
queue<int> q; 
void dfs(int idx){
    int sz = edge[idx].size();
    if(sz >= 1){ // 存在左孩子
        int leftIdx = edge[idx][0];
        dfs(leftIdx);
        if(val[idx] == val[leftIdx]) // 左孩子的值等于根节点的值
        dp[idx][0] = max(dp[leftIdx][0], dp[leftIdx][1]) + 1;
        
    }
    if(sz >= 2){ // 存在右孩子
        int rightIdx = edge[idx][1];
        dfs(rightIdx); 
        if(val[idx] == val[rightIdx]) // 右孩子的值等于根节点的值
        dp[idx][1] = max(dp[rightIdx][1], dp[rightIdx][0]) + 1;
    }
    
    ans = max(ans, dp[idx][0] + dp[idx][1]);
}
 
int main(){
    scanf("%d", &n);
    memset(dp, 0, sizeof dp);
    q.push(0);
    for(int i = 1; i <= n; i++){
        int now = q.front(); q.pop(); // 注意是按照层序遍历的方式建树的
        scanf("%s", s);
        if(strcmp(s, "null") != 0) {
            int n = strlen(s);
            int num = 0;
            for(int j = 0; j < n; j++){
                num = num * 10 + s[j] - '0';
            }
            val[i] = num;
            
            edge[now].push_back(i);
            q.push(i);
        }
        if(i > 1&& (i + 1 <= n)){ // 根节点只有一个
            i++;
            scanf("%s", s);
            if(strcmp(s, "null") != 0) {
                int n = strlen(s);
                int num = 0;
                for(int j = 0; j < n; j++){
                    num = num * 10 + s[j] - '0';
                }
                val[i] = num;
                
                edge[now].push_back(i);
                q.push(i);
            }
        }
       
    }
    dfs(1);
    printf("%d\n", ans); // ans表示的最长路径上的节点数,边数=节点数-1
    return 0;
}





















