题目链接
思路
- 这个题注意,图的结点应该是后面的序列中出现的所有的字符!
- 还有一点在于,对于某个序列[1,2,3],我们只需要创建1指向2的弧,2指向3的弧即可,并不需要创建1指向3的弧,尽管1的优先级比3高,但是只要建立了1->2的弧,那么就可以保证1不出去,3也绝对出不去!
- 至于如何判断是否只有一种拓扑排序的结果,只要每次都判断一下,此时队列里是不是只有一个元素!
- 最后注意一下判断数组相等的API
Arrays.equals(orgs,build.stream().mapToInt(e->e).toArray());
class Solution {
public boolean sequenceReconstruction(int[] org, List<List<Integer>> seqs) {
Map<Integer,Set<Integer>>graph = new HashMap<>();
Map<Integer,Integer> inDegrees = new HashMap<>();
for(List<Integer> seq : seqs){
for(int num :seq){
graph.putIfAbsent(num,new HashSet<>());
inDegrees.putIfAbsent(num,0);
}
}
for(List<Integer> seq : seqs){
for(int i = 1 ; i < seq.size() ; i++){
if(graph.get(seq.get(i)).contains(seq.get(i-1)))
return false;
if(!graph.get(seq.get(i-1)).contains(seq.get(i))){
graph.get(seq.get(i-1)).add(seq.get(i));
inDegrees.put(seq.get(i),inDegrees.get(seq.get(i))+1);
}
}
}
List<Integer> build = new ArrayList<>();
Queue<Integer> queue = new LinkedList<>();
for(int num : inDegrees.keySet()){
if(0 == inDegrees.get(num))
queue.offer(num);
}
while(1 == queue.size()){
int index = queue.poll();
build.add(index);
for(int next :graph.get(index)){
inDegrees.put(next,inDegrees.get(next)-1);
if(0 == inDegrees.get(next))
queue.offer(next);
}
}
int[] result = build.stream().mapToInt(e->e).toArray();
return Arrays.equals(result,org);
}
}