挑战程序设计竞赛（算法和数据结构）——12.5连通分量的JAVA实现

Posted 2021-12-31 小乖乖的臭坏坏

题目：

讲解：

注意：
定义了一个Vector类型的数组，不仅需要初始化Vector类型的数组，还需初始化每一个Vector类型的容器。
否则会出现以下异常：

Exception in thread “main” java.lang.NullPointerException: Cannot invoke “java.util.Vector.add(Object)” because “G[u]” is null
at ConnextedComponents.main(ConnextedComponents.java:21)

代码：

import com.sun.jdi.event.VMDeathEvent;

import java.io.BufferedInputStream;
import java.util.Scanner;
import java.util.Stack;
import java.util.Vector;

public class ConnextedComponents 
    public static void main(String[] args) 
        Vector<Integer> G[] = new Vector[1000];
//        for (int i=0;i<1000;i++)
//            G[i] = new Vector<Integer>();
//        

        Scanner cin = new Scanner(new BufferedInputStream(System.in));
        int n = cin.nextInt();
        int m = cin.nextInt();
        for (int i=0;i<m;i++)
            int u = cin.nextInt();
            int v = cin.nextInt();
            G[u].add(v);
            G[v].add(u);
        

        //给表示颜色的数组初始化
        int color[] = new int[1000];
        for (int i=0;i<1000;i++)
            color[i] = -1;
        

        //只要没有进行初始化，说明不属于其他连通块，那么就执行寻找连通块功能
        int id = 1;
        for (int i=0;i<n;i++)
            if(color[i]==-1)dfs(i, id++, color, G);
        

        int q = cin.nextInt();
        for (int i=0;i<q;i++)
            int s = cin.nextInt();
            int t = cin.nextInt();
            if(color[s]==color[t])
                System.out.println("yes");
            else
                System.out.println("no");
            
        
    

    //设r是当前id号，c为表示颜色的代号
    public static void dfs(int r, int c, int color[], Vector<Integer> G[])
        Stack<Integer> S = new Stack<>();//定义一个栈
        S.add(r);//栈将当前id号r值存入
        color[r] = c;//令r值设为当前颜色代号c
        while (!S.empty())//只要S不为空就执行以下操作
            int u = S.peek();S.pop();//取栈顶元素u，并弹栈
            for (int i=0;i<G[u].size();i++)//对于G[u] Vector尺寸中的每一个元素
                int v = G[u].get(i);//用临时变量v来保存每一个值。
                if(color[v]==-1)//只要v的color还是-1，那就赋值，并将v压入栈中。显然，栈内元素为同一个颜色的。
                    color[v] = c;
                    S.push(v);
                
            
        
    

输入：

10 9
0 1
0 2
3 4
5 7
5 6
6 7
6 8
7 8
8 9
3
0 1
5 9
1 3

输出：

yes
yes
no