Uso de Stack + Indegree para realizar la clasificación topológica

Uso de Stack + Indegree para realizar la clasificación topológica

class Solution {
      Stack<Integer> stack = new Stack<Integer>();
    boolean flag[] ;
    int []result;
    Map<Integer,List<Integer>> map =new HashMap<Integer,List<Integer>>();
    void init(int numCourses, int[][] prerequisites)
    {
        flag =  new boolean[numCourses];
        result = new int[numCourses];
        for(int i=0;i<numCourses;i++)
        {
            map.put(i,new ArrayList<Integer>());
        }
        for(int i=0;i<prerequisites.length;i++)
        {
            map.get(prerequisites[i][1]).add(prerequisites[i][0]);
        }
    }
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        if(numCourses==0)
            return new int[0];
        init(numCourses, prerequisites);
        for(int i=0;i<numCourses;i++)
        {
          BFS_Stack(i);
        }
        int track=0;
        while(!stack.isEmpty())
        {
            result[track++]=stack.pop();
        }
        return result;
    }

    void  BFS_Stack(Integer start)
    {
        if(flag[start])
            return;
        else if(map.get(start).size()==0)
        {
            stack.add(start);
            flag[start]=true;
            return;
        }
        else
        {
            flag[start]=true;
            for(int i=0;i<map.get(start).size();i++)
            {
                BFS_Stack(map.get(start).get(i));
            }
            stack.add(start);

        }
        return ;
    }
}
Publicado 28 artículos originales · elogiado 0 · visitas 619

Supongo que te gusta

Origin blog.csdn.net/qq_44932835/article/details/105423192
Recomendado
Clasificación