蓝桥杯——发现环

小明的实验室有N台电脑,编号1~N。原本这N台电脑之间有N-1条数据链接相连,恰好构成一个树形网络。在树形网络上,任意两台电脑之间有唯一的路径相连。
不过在最近一次维护网络时,管理员误操作使得某两台电脑之间增加了一条数据链接,于是网络中出现了环路。环路上的电脑由于两两之间不再是只有一条路径,使得这些电脑上的数据传输出现了BUG。
为了恢复正常传输。小明需要找到所有在环路上的电脑,你能帮助他吗?
样例输入
5
1 2
3 1
2 4
2 5
5 3
样例输出
1 2 3 5

思路:这题就是用拓扑排序,因为这是一个有向图,拓扑排序只能用在有向图,通过拓扑排序去判断环,注意,有的节点虽然在环上,但是它不参与闭合,那么就是它有入度,但是出度为0,那么只需要再记录一下每个结点的出度,然后在最后不把没有出度的节点加入环汇总就能拿到正确答案了。
代码如下:注解详细

static int MAX = Integer.MAX_VALUE;
public static void main(String[] args)
{
    Scanner in = new Scanner(System.in);
    int n = in.nextInt();
    int[][] e = new int[n][n];
    for (int i = 0; i < n; i++)
    {
        int a = in.nextInt();
        int b = in.nextInt();
        e[a-1][b-1] = 1;
    }
    for (int i = 0; i < e.length; i++)
    {
        for (int j = 0; j < e[i].length; j++)
        {
            if(i!=j && e[i][j]!=1)
                e[i][j]=MAX;
        }
    }
    System.out.println(topSort(n,e));
}

public static List<Integer> topSort(int n, int[][] e) {
    LinkedList<Integer> path = new LinkedList<>();
    LinkedList<Integer> ring = new LinkedList<>();
    int[] inDegree = new int[n];
    int[] outDegree = new int[n];
    //获取出度
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(e[i][j]==1)
                outDegree[i] += e[i][j];
        }
    }
    //获取入度
    for(int j=0;j<n;j++)
    {
        for(int i=0;i<n;i++)
        {
            if(e[i][j]==1)
                inDegree[j] += e[i][j];
        }
    }
    Queue<Integer> q = new LinkedList<>();
    //将所有入度为0的节点加入队列
    for (int i = 0; i < n; i++) {
        if (inDegree[i] == 0)
            q.offer(i);
    }
    while (!q.isEmpty()) {
        int cur = q.poll();
        path.add(cur);
        for(int j=0;j<n;j++)
        {
            if(e[cur][j]!=MAX)
                e[cur][j] = 0;
            if(e[cur][j]!=MAX && inDegree[j]!=0)
            {
                inDegree[j]--;
                if(inDegree[j]==0)
                    q.offer(j);
            }
        }
    }
    //这里去掉出度为0的值
    if(path.size()!=n)
    {
        for (int i = 0; i < n; i++)
        {
            if(!path.contains(i) && outDegree[i]!=0)
                ring.add(i+1);//输出的值加1
        }
    }
    return path.size() == n ? path : ring;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值