2020第十一届蓝桥杯省赛B组第一场 J. 网络分析

问题描述
时间限制: 1.0s 内存限制: 256.0MB 本题总分：25 分
【问题描述】
小明正在做一个网络实验。
他设置了 n 台电脑，称为节点，用于收发和存储数据。
初始时，所有节点都是独立的，不存在任何连接。
小明可以通过网线将两个节点连接起来，连接后两个节点就可以互相了。两个节点如果存在网线连接，称为相邻。
小明有时会测试当时的网络，他会在某个节点发送一条信息，信息会到每个相邻的节点，之后这些节点又会转发到自己相邻的节点，直到所有或间接相邻的节点都收到了信息。所有发送和接收的节点都会将信息存储一条信息只存储一次。
给出小明连接和测试的过程，请计算出每个节点存储信息的大小。
【输入格式】
输入的第一行包含两个整数 n, m，分别表示节点数量和操作数量。节1 至 n 编号。
接下来 m 行，每行三个整数，表示一个操作。
如果操作为 1 a b，表示将节点 a 和节点 b 通过网线连接起来。当 时，表示连接了一个自环，对网络没有实质影响。
如果操作为 2 p t，表示在节点 p 上发送一条大小为 t 的信息。
【输出格式】
输出一行，包含 n 个整数，相邻整数之间用一个空格分割，依次表示完上述操作后节点 1 至节点 n 上存储信息的大小。
【样例输入】

4 8
1 1 2
2 1 10
2 3 5
1 4 1
2 2 2
1 1 2
1 2 4
2 2 1
1
2
3
4
5
6
7
8
9
【样例输出】

13 13 5 3
1
【评测用例规模与约定】
对于 30% 的评测用例，1 ≤ n ≤ 20，1 ≤ m ≤ 100。
对于 50% 的评测用例，1 ≤ n ≤ 100，1 ≤ m ≤ 1000。
对于 70% 的评测用例，1 ≤ n ≤ 1000，1 ≤ m ≤ 10000。
对于所有评测用例，1 ≤ n ≤ 10000，1 ≤ m ≤ 100000，1 ≤ t ≤ 100。

解：通过了测试用例，不确定是否正确
1. 创建邻接表，List，记得判断是否存在再加入
2. 然后根据输入，先判断‘1’or‘2’
2.1 ‘1’添加链接
2.2 ‘2’ bfs存储信息，开辟一个数组存储每个节点的信息
一个vis，每次‘2’出现需要更新

import java.util.*;

public class Main {
    public static void main(String[] args) {
        /*
            1. 创建邻接表，List，记得判断是否存在再加入
            2. 然后根据输入，先判断‘1’or‘2’
            2.1 ‘1’添加链接
            2.2 '2' bfs存储信息，开辟一个数组存储每个节点的信息
                一个vis，每次‘2’出现需要更新
         */
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();

        int[] ans = new int[n+1];

        //1. 创建list，邻接表
        List<List<Integer>> list = new ArrayList<>();
        for(int i = 0; i <= n; i++) {
            list.add(new ArrayList<>());
        }

        //m次3个数据
        for(int i = 0; i < m; i++) {
            int a = sc.nextInt();
            int b = sc.nextInt();
            int c = sc.nextInt();
            if(a == 1) {// 1 则需要添加边
                list = add(b,c,list);
            } else {//等于 2 ，就要记录信息
                ans = info(b,c,list,ans);
            }
        }

        for(int i = 1; i <= n; i++) {
            System.out.print(ans[i] + " ");
        }
    }

    public static int[] info(int b, int c, List<List<Integer>> list, int[] ans) {
        int[] vis = new int[list.size()];
        Queue<Integer> queue = new LinkedList<>();
        queue.add(b);
        vis[b] = 1;
        while (!queue.isEmpty()) {
            int poll = queue.poll();
            ans[poll] += c;
            //poll的邻接点
            for(Integer num : list.get(poll)) {
                if(vis[num] == 0) {
                    queue.add(num);
                    vis[num] = 1;
                }
            }
        }
        return ans;
    }

    public static List<List<Integer>> add(int b, int c, List<List<Integer>> list) {
        if(!list.get(b).contains(c)) {
            list.get(b).add(c);
        }
        if(!list.get(c).contains(b)) {
            list.get(c).add(b);
        }
        return list;
    }
}