牛客网 2018校招真题 美团点评 病毒传播

Description

牛客网 2018校招真题 病毒传播

Solving Ideas

以每个被感染的点为起点，使用bfs进行模拟，得到t时间的传染结果，然后将其与真实结果对比

Solution

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;

/**
 * @author wylu
 */
public class Main {
    //感染病毒的点为true, 未感染的为false
    static boolean[] infected;
    static ArrayList<Integer>[] graph;
    static int n, m, k, t;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] strs = br.readLine().split(" ");
        n = Integer.parseInt(strs[0]);
        m = Integer.parseInt(strs[1]);
        infected = new boolean[n + 1];
        graph = new ArrayList[n + 1];
        for (int i = 1; i <= n; i++) graph[i] = new ArrayList<>();

        //建图
        for (int i = 0; i < m; i++) {
            strs = br.readLine().split(" ");
            int u = Integer.parseInt(strs[0]), v = Integer.parseInt(strs[1]);
            graph[u].add(v);
            graph[v].add(u);
        }

        strs = br.readLine().split(" ");
        k = Integer.parseInt(strs[0]);
        t = Integer.parseInt(strs[1]);
        for (String s : br.readLine().split(" ")) {
            infected[Integer.parseInt(s)] = true;
        }

        int res = 0;
        for (int i = 1; i <= n; i++) {
            //只有感染的点才需要检查
            if (infected[i] && bfs(i)) {
                System.out.print((res++ > 0 ? " " : "") + i);
            }
        }
        System.out.println((res == 0) ? "-1" : "");
    }

    //以x为起点传播t天的结果和实际结果比较是否相同
    private static boolean bfs(int x) {
        //每个点被传染需要的时间, 为0表明没有被传染
        int[] cost = new int[n + 1];
        LinkedList<Integer> queue = new LinkedList<>();
        //按照题意起点被传染需要的时间应该为0,
        //此处置1, 则整体被传染时间都会偏移1, 这样做是为了方便用0来表示未感染
        //(当然也可以将cost[]初始为-1, 用-1表示未感染, 此时起点可以置0)
        cost[x] = 1;
        queue.offer(x);
        while (!queue.isEmpty()) {
            int cur = queue.poll();
            //因为整体偏移了1, 所以这里不是 cost[cur] >= t
            if (cost[cur] > t) break;
            for (Integer e : graph[cur]) {
                if (cost[e] == 0) {
                    cost[e] = cost[cur] + 1;
                    queue.offer(e);
                }
            }
        }

        for (int i = 1; i <= n; i++) {
            if (!infected[i] && cost[i] != 0) return false;
            if (infected[i] && cost[i] == 0) return false;
        }
        return true;
    }
}