[ABC254E] Small d and k 题解

题目传送门

ABC 的 E 题这么简单？

题意

给定 $n$ 个点 $m$ 条边的简单无向图，特别地，保证每个点的度数不超过 $3$。$q$ 次询问，给定 $x,k$，求所有距离 $x$ 不超过 $k$ 的点（包括 $x$）的编号和。

Solution

看到题面中“求所有距离 $x$ 不超过 $k$ 的点（包括 $x$）的编号和”这句话，考虑广搜，因为广搜是按照距离搜的，这样更好的计算距离。

每次询问时，BFS 一次，初始时定义 $d_x=0$，即 $x$ 与 $x$ 的距离为 $0$，然后将 $x$ 进队。每次取出队头 $y$，扫描与 $y$ 相连的所有边，这些边的另一端点 $i$ 与 $y$ 的距离 $d_j=d_y+1$。这样扫下去就行了，记得用 $vis$ 数组标记该店是否访问过。

由于广搜是按照距离搜的，所以访问的点与点 $x$ 的距离是 $1,2,3$ 这样递增的，所以扫到 $d_y>k$ 时，就跳出循环，因为剩下的距离都是大于 $k$ 的。

Code

本人较懒，用 vector 存图，码量少多了。vector 真的太方便了！

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=150010;
int n,m,Q,x,k,d[N];
vector<int>G[N];
ll ans;
bool vis[N];
void bfs()
{
	ans=0;//答案每次清零
	memset(vis,0,sizeof(vis));//标记数组清零
	queue<int>q;
	q.push(x);
	d[x]=0;
	vis[x]=1;
	while(q.size())
	{
		int y=q.front();q.pop();
		if(d[y]>k) break;//距离大于k
		ans+=y;
		for(auto i:G[y])//扫描与y相连的边，此处"i"即为该边另一端点
		{
			if(!vis[i]) vis[i]=1,q.push(i),d[i]=d[y]+1;
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int u,v;
		cin>>u>>v;
		G[u].push_back(v);
		G[v].push_back(u);//无向图存边存两次
	}
	cin>>Q;
	while(Q--)
	{
		cin>>x>>k;
		bfs();
		cout<<ans<<'\n';
	}
	return 0;
}