LintCode 605. 序列重构

2020-05-05  本文已影响0人  CW不要无聊的风格

题目描述

题目描述

判断是否序列 org 能唯一地由 seqs 重构得出,org是一个由从1到n的正整数排列而成的序列, 1 ≤ n ≤ 10^​4​​ 。

重构表示组合成 seqs 的一个最短的父序列 (意思是,一个最短的序列使得所有 seqs 里的序列都是它的子序列)。

判断是否有且仅有一个能从 seqs 重构出来的序列,并且这个序列是 org


测试样例

输入:org = [1,2,3], seqs = [[1,2],[1,3]]

输出: false

解释:

[1,2,3] 并不是唯一可以被重构出的序列,还可以重构出 [1,3,2]

输入: org = [1,2,3], seqs = [[1,2]]

输出: false

解释:

能重构出的序列只有 [1,2].

输入: org = [1,2,3], seqs = [[1,2],[1,3],[2,3]]

输出: true

解释:

序列 [1,2], [1,3], 和 [2,3] 可以唯一重构出 [1,2,3].

输入:org = [4,1,5,2,6,3], seqs = [[5,2,6,3],[4,1,5,2]]

输出:true


题意理解&思路

 > seqs 中的每个序列相当于图中的一条有向边;

> 由 seqs 中所有序列构成的树,每层节点数必须为1,否则这些序列就不止构成唯一序列(而是有多个);

> seqs 中的元素需和 org 完全一致

由于最终目标是 org,因此基于 org 中的元素来构建目标图节点,根据 seqs 中的序列,构建每个点的邻居,同时计算每个点的入度。

邻居构建和入度计算完毕,便可以使用 BFS,每次取入度为0的节点用于构建序列,同时更新它邻居的入度。若有某次入度为0的节点数不止1个,则说明 seqs 可构成多个序列,目标不成立。

最终判断构建出来的序列和 org 是否一致。


具体方法

I. 以 org 中的元素为 key 初始化2个 dict,分别用于存储邻居表和入度;

II. 初始化一个空集,存储 seqs 中不同的元素,用于在处理 seqs 中的每个序列时判断是否出现了 org 以外的元素。若是,则目标不成立;

III. 遍历 seqs 中每个序列的元素,在当前元素的邻居表中加入邻居,邻居为其后的一个元素,同时为这个邻居的入度加1;

IV. 待 seqs 中所有序列均处理完毕后,判断 org 中的元素与存储 seqs 中元素的集合是否完全一致,若不是,则目标不成立;

V. 每次取入度为0的元素,若这样的元素多余1个,则目标不成立;否则,将其加入列表构建序列;

VI. 更新取出元素的邻居的入度,将它们的入度减一,然后把这些邻居加入队列;

VII. 重复V、VI直至队列为空;

VIII. 判断构建出的序列与 org 是否完全一致


代码

from collections import deque

class Solution:

    """

    @param org: a permutation of the integers from 1 to n

    @param seqs: a list of sequences

    @return: true if it can be reconstructed only one or false

    """

    def sequenceReconstruction(self, org, seqs):

        # write your code here

        if not org:

            return True

        if not seqs or not any(seqs):

            return False

        # 构建图,入度和邻居

        indegree = {}.fromkeys(org, 0)

        neighbors = {n: [] for n in org}

        # 存放节点的集合,用于判断seqs中的节点和org是否完全一样

        nodes = set()

        for seq in seqs:

            nodes.update(seq)

            # 判断seqs中是否有org以外的节点

            if not nodes.issubset(org):

                return False

            for i in range(len(seq) - 1):

                indegree[seq[i+1]] += 1

                neighbors[seq[i]].append(seq[i+1])

        # 判断org中是否有seqs中没有的节点

        if set(org).difference(nodes):

            return False

        res = []

        queue = deque([start for start in indegree if not indegree[start]])

        while queue:

            # 要使得有且仅有一个能从 seqs重构出来的序列,则树的每层level节点数必须是1

            if len(queue) != 1:

                return False

            node = queue.popleft()

            res.append(node)

            # BFS

            for neighbor in neighbors[node]:

                indegree[neighbor] -= 1

                if not indegree[neighbor]:

                    queue.append(neighbor)

        return res == org

上一篇下一篇

猜你喜欢

热点阅读