0
点赞
收藏
分享

微信扫一扫

【leetcode】1719. 重构一棵树的方案数

鱼板番茄 2022-02-16 阅读 46

题目:
1719. 重构一棵树的方案数
给你一个数组 pairs ,其中 pairs[i] = [xi, yi] ,并且满足:
pairs 中没有重复元素
xi < yi
令 ways 为满足下面条件的有根树的方案数:

树所包含的所有节点值都在 pairs 中。
一个数对 [xi, yi] 出现在 pairs 中 当且仅当 xi 是 yi 的祖先或者 yi 是 xi 的祖先。
注意:构造出来的树不一定是二叉树。
两棵树被视为不同的方案当存在至少一个节点在两棵树中有不同的父节点。
请你返回:
如果 ways == 0 ,返回 0 。
如果 ways == 1 ,返回 1 。
如果 ways > 1 ,返回 2 。
一棵 有根树 指的是只有一个根节点的树,所有边都是从根往外的方向。

我们称从根到一个节点路径上的任意一个节点(除去节点本身)都是该节点的 祖先 。根节点没有祖先。

示例 1:
在这里插入图片描述

输入:pairs = [[1,2],[2,3]]
输出:1
解释:如上图所示,有且只有一个符合规定的有根树。
示例 2:
在这里插入图片描述
输入:pairs = [[1,2],[2,3],[1,3]]
输出:2
解释:有多个符合规定的有根树,其中三个如上图所示。

示例 3:
输入:pairs = [[1,2],[2,3],[2,4],[1,5]]
输出:0
解释:没有符合规定的有根树。

提示:

1 <= pairs.length <= 105
1 <= xi < yi <= 500
pairs 中的元素互不相同。


```java
class Solution {
    public int checkWays(int[][] pairs) {
        Set<Integer>[] path=new Set[505];//记录每个点跟其他哪些点有祖孙关系
        for(int i=0;i<path.length;i++){path[i]=new HashSet<>();}
        int numOfNodes=0;
        int mostRelations=0;
        for(int i=0;i<pairs.length;i++){
            if(path[pairs[i][0]].size()==0){numOfNodes++;}
            if(path[pairs[i][1]].size()==0){numOfNodes++;}
            path[pairs[i][0]].add(pairs[i][1]);
            path[pairs[i][1]].add(pairs[i][0]);
            mostRelations=Math.max(mostRelations,Math.max(path[pairs[i][0]].size(),path[pairs[i][1]].size()));
        }
        if(numOfNodes!=mostRelations+1){return 0;}
        int rootVal=-1;//总的根节点
        int ans=1;
        for(int i=1;;i++){
            if(path[i].size()==mostRelations){
                rootVal=i;
                break;
            }
        }
        for(int p:path[rootVal]){
            //p是root的某一个后代,下面开始寻找p的父节点
            int dad=-1,partnerNum=505;
            for(int partner:path[p]){
                if(rootVal==p){continue;}
                if(partnerNum>path[partner].size()&&path[partner].size()>=path[p].size()){
                    dad=partner;
                    partnerNum=path[partner].size();
                }
            }
            if(partnerNum==path[p].size()){ans=2;}
            for(int a:path[p]){
                if(a==dad){continue;}
                if(!path[dad].contains(a)){return 0;}
            }
        }
        return ans;
    }
}
举报

相关推荐

0 条评论