leetcode 2246. Longest Path With Different Adjacent Characters(不同相邻字母的最长路径)

leetcode 2246. Longest Path With Different Adjacent Characters(不同相邻字母的最长路径)_第1张图片
给一棵以节点0为根的树(不一定是二叉树),共有n个节点,0~n-1,
同样的,有长度为n的数组parent, parent[i]表示第 i 个节点的parent, 0的parent是-1.
长度为n的字符串,s(i)表示第i个节点的字母。

返回最长的路径,路径中相邻的节点字母不能相同。

思路:

首先探讨一下什么叫路径。
树的路径可以是(1) 从根节点到叶子节点的路径,(2)任一节点到其他任一节点的路径。
这里是定义(2).

再说下树的直径,定义为树上最远的两个节点之间的距离。

所以本题其实求的是树的直径,长度为路径上的节点数。

case 1:
先讨论一个简单的情形,Example1, 以根节点0为例,
节点0连接的child(即节点2,1)为出发点,能得到最长路径mx1=2(节点2->5),和第二长的路径mx2=2(节点1->3)。

然后从节点0来看,节点2这条路也应该舍弃,因为节点2和节点0的字母相同。所以mx1=2只剩节点1->3这条,所以mx1=2, mx2=0(顺延)。

而节点0将这两个最长路径连接了起来,加上节点0本身,最大路径长度为1+mx1+mx2.

而每个child会返回由它出发的最长路径,所以dfs(node)会返回遍历node所有child所得到的mx1和它自己的和,即1+mx1.

case 2:
节点0-1-2(只有一个分叉的树),s = “aab”
根节点0和1字母都是’a’,不满足条件,是不是要把节点1为根的这个分支砍了,只留节点0,然后最大路径是1?
前面说过了,最大路径是以任一节点出发的,如果从2出发,那最大路径就是2.
所以,以节点1为出发点的路径长度mx1计算出来后,要用它去更新最大路径长度,因为这一段本身也是路径。

另外,本题明确规定了节点的parent关系,可以按有向图处理。

class Solution {
    List<Integer>[] graph;
    int res = 0;

    public int longestPath(int[] parent, String s) {
        int n = parent.length;
        res = 1;
        graph = new ArrayList[n];

        for(int i = 0; i < n; i++) graph[i] = new ArrayList<>();
        for(int i = 1; i < n; i++) graph[parent[i]].add(i); //有向图
        dfs(0, s);
        return res;
    }

    int dfs(int cur, String s) {
        if(graph[cur].size() == 0) return 1;  //叶子节点

        int mx1 = 0;
        int mx2 = 0;

        for(int child : graph[cur]) {
            int len = dfs(child, s);
            res = Math.max(res, len); //case 2

            if(s.charAt(child) == s.charAt(cur)) continue;
            if(len > mx1) {  //case 1
                mx2 = mx1;
                mx1 = len;
            } else {
                mx2 = Math.max(mx2, len);
            }
        }

        res = Math.max(res, 1+mx1+mx2);
        return 1+mx1;
    }
}

你可能感兴趣的:(leetcode,leetcode,算法,深度优先)