标签：dist 真题 int 18 queue 2022 graph new size
MT15 病毒传播描述
给出一个图 G(V,E) ，图上有 n 个点，m 条边，所有的边都是无向边。
最开始，也就是第 0 天的时候，这 n 个点中有一个点 v 感染了病毒，之后的每一天，凡是感染病毒的点都会向它的邻居点传播病毒。经过了 t 天之后，得到了感染病毒的点集 S 。要求找出第 0 天感染病毒的点 v 。如果 v 有很多不同的答案，把它们都找出来。 数据范围： 1 \le n,m \le 1000 \1≤n,m≤1000 ， 1 \le k \le n \1≤k≤n ，1 \le t \le 10^9 \1≤t≤109输入描述：
第一行两个数n，m，接下来有m行，每行两个数u，v，表示点u，v之间有一条无向边。接下来一行两个数k，t，其中k表示集合S的大小。最后一行k个数，集合S中的元素。输入的图可能有自环和重边，输入保证S中的数互不相同。
输出描述：
输出一行，如果不存在这样的v，输出-1。否则输出所有可能的v，按照从小到大的顺序输出，数字之间用空格隔开，不要在行末输出多余的空格。
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int m=sc.nextInt();
int[][] dist=new int[n][n];
for (int i=0;i<dist.length;i++){
Arrays.fill(dist[i],Integer.MAX_VALUE);
}
for (int i=0;i<m;i++){
int p=sc.nextInt();
int q=sc.nextInt();
// 自环
if (p!=q){
dist[p-1][q-1]=1;
dist[q-1][p-1]=1;
}
}
List<List<Integer>> graph=buildgraph(dist);
int k=sc.nextInt();
int t=sc.nextInt();
Set<Integer> set=new HashSet<>();
for (int i=0;i<k;i++){
int temp=sc.nextInt();
set.add(temp);
}
// bfs计算记录
List<Integer> ans=new ArrayList<>();
for (int i=0;i<n;i++){
//Set<Integer> temp=BFS(graph,i,t);
Set<Integer> temp=Dijkstra(dist,graph,i,t);
if (set.size()!=temp.size()) continue;
boolean flag=true;
for (int x:set){
if (!temp.contains(x)){
flag=false;
break;
}
}
if (flag) ans.add(i);
}
if (ans.size()==0){
System.out.println(-1);
}else{
for (int i=0;i<ans.size()-1;i++) System.out.print((ans.get(i)+1)+" ");
System.out.println(ans.get(ans.size()-1)+1);
}
}
public static List<List<Integer>> buildgraph(int[][] dist){
List<List<Integer>> graph=new ArrayList<>();
int n=dist.length;
for (int i=0;i<n;i++){
graph.add(new ArrayList<>());
}
for (int i=0;i<n;i++){
for (int j=i+1;j<n;j++){
if (dist[i][j]==1){
graph.get(i).add(j);
graph.get(j).add(i);
}
}
}
return graph;
}
public static Set<Integer> BFS(List<List<Integer>> graph,int start,int time){
Set<Integer> res=new HashSet<>();
Queue<Integer> queue=new LinkedList<>();
boolean[] visited=new boolean[graph.size()];
int[] dist=new int[graph.size()];
Arrays.fill(dist,Integer.MAX_VALUE);
dist[start]=0;
visited[start]=true;
queue.offer(start);
int dis=0;
while (!queue.isEmpty()){
int len=queue.size();
for (int i=0;i<len;i++){
int node=queue.poll();
dist[node]=dis;
for (int x:graph.get(node)){
if (!visited[x]){
queue.offer(x);
visited[x]=true;
}
}
}
dis++;
}
for (int i=0;i<dist.length;i++){
if (dist[i]<=time){
res.add(i+1);
}
}
return res;
}
public static Set<Integer> Dijkstra(int[][] dist,List<List<Integer>> graph,int start,int time){
// int[] node,dis
PriorityQueue<int[]> queue=new PriorityQueue<>((a,b)->(a[1]-b[1]));
Set<Integer> res=new HashSet<>();
queue.add(new int[]{start,0});
int[] dis=new int[graph.size()];
Arrays.fill(dis,Integer.MAX_VALUE);
dis[start]=0;
while (!queue.isEmpty()){
int[] temp=queue.poll();
int curNode=temp[0];
int curdis=temp[1];
if (dis[curNode]<curdis) continue;
for (int x:graph.get(curNode)){
if (curdis+dist[curNode][x]<dis[x]){
dis[x]=curdis+dist[curNode][x];
queue.offer(new int[]{x,dis[x]});
}
}
}
for (int i=0;i<dis.length;i++){
if (dis[i]<=time) res.add(i+1);
}
return res;
}
}
思路：两种方法求不同节点到其他节点的距离。因为权重都是1可以用BFS的方法遍历，也可以用Dijkstra方法计算。再比较两个集合。
标签：dist,真题,int,18,queue,2022,graph,new,size 来源： https://www.cnblogs.com/benbicao/p/16388170.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。