【洛谷八连测R7】nzhtl1477-我回来了

【洛谷八连测R7】nzhtl1477-我回来了

题目描述

68号岛有n个商店，有的商店直接有小路连接，小路的长度都为1，格里克告诉了你哪些地方可能有做黄油蛋糕的原料。

但是那个人是个坑货，所以他会告诉你一些商店，然后告诉你距离这些商店距离 <= k的商店中都是可能有原料的。

然后你要把这些可能的商店每个都去一遍，你想知道你要去多少个商店。

由于你是勇者，所以有m次询问

简洁题意：

给你一个图，每次查询的时候给一堆特殊点以及一个数k，求图中有多少点距离至少一个特殊点距离不超过k，边是无向的。

输入输出格式

输入格式：

第一行三个数表示n，m，q

之后m行每行两个数x，y表示这两个点之间连有一条边~

之后q次询问，每个询问先给你一个数a和一个数k

之后一行a个数，表示a个特殊点

输出格式：

q行，每行一个数表示答案

说明

对于30%的数据，n，m，q <= 100，每次查询只给一个点

对于另外30%的数据，k=1

对于100%的数据，n，m，q <= 5000 , a的和 <= 500000

分析

相邻点之间的距离是1，直接用bfs就行了。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N=5010;
inline int read(){
    int x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}
int n,m,q,tot;
int head[N],dis[N];
bool vis[N];
queue<int>Q;
struct node{
    int next,to;
}e[N*2];
inline void ins(int from,int to){
    e[++tot].next=head[from];
    e[tot].to=to;
    head[from]=tot;
}
int main(){
    n=read();m=read();q=read();
    for(int i=1;i<=m;++i){
        int x=read(),y=read();
        ins(x,y); ins(y,x);
    }
    for(int i=1;i<=q;++i){
        int a=read(),k=read(),cnt=0;
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=a;++i){
            int x=read();
            if(!vis[x]){
                Q.push(x); ++cnt;
                vis[x]=true; dis[x]=0;
            }
        }
        do{
            int u=Q.front();Q.pop();
            for(int i=head[u];i;i=e[i].next)
            if(!vis[e[i].to]){
                dis[e[i].to]=dis[u]+1;
                if(dis[e[i].to]<=k) ++cnt;
                if(dis[e[i].to]<k) Q.push(e[i].to);
                vis[e[i].to]=true;
            }
        }while(!Q.empty());
        printf("%d\n",cnt);
    }
        return 0;
}