LC周赛289（卡常记录）

原创于 2022-04-17 23:42:32 发布 · 417 阅读

CC 4.0 BY-SA版权

文章标签：

这篇博客探讨了一种记录在线更新最大值和次大值的算法。通过递归的深度优先搜索（DFS），在给定的树结构中，动态维护每个节点的最大值和次大值。算法首先初始化最大值和次大值，然后遍历子节点，如果子节点的值大于当前最大值，更新最大值和次大值；如果子节点的值介于最大值和次大值之间，则更新次大值。最后，返回最长路径的长度。该算法注重效率，使用引用传递以提高速度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

记录在线更新最大值，次大值的算法

            if (tmp>mx1) {
                mx2=mx1;
                mx1=tmp;
            }
            else if (tmp>mx2) {
                mx2=tmp;
            }

注意：传递引用较快，传值较慢
也可将其放入成员变量中，较快

class Solution {
    int ans = 0;
    vector<int> cd[100000];
    string S;
public:
    int dfs(int root)
    {
        int max_1 = 0;
        int max_2 = 0;
        for(int i:cd[root])
        {
            int b = dfs(i);
            if(S[i] ==S[root]) continue;
            if(b > max_1)
            {
                max_2 = max_1;
                max_1 = b; 
            }
            else if(b > max_2)
            {
                max_2 = b;
            }
        }
        ans = max(ans,max_1 + max_2 + 1);
        return max(max_1 + 1,max_2 + 1);
    }
    int longestPath(vector<int>& parent, string s) {
        int root;
        for(int i = 0;i<parent.size();i++)
        {
            if(parent[i] == -1) root = i;
            else cd[parent[i]].push_back(i);
        }
        S = s;
        dfs(root);
        return ans;
    }
};