题目链接
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;
}