蓝桥杯——仙境诅咒（dfs）

最新推荐文章于 2024-08-23 15:15:47 发布

本少侠今天躺平了

最新推荐文章于 2024-08-23 15:15:47 发布

阅读量348

点赞数 16

分类专栏：蓝桥杯省赛c++ 文章标签：蓝桥杯算法 c++ dfs

本文链接：https://blog.csdn.net/vs408/article/details/137107358

版权

蓝桥杯省赛c++ 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

文章讲述了如何使用C++编程语言解决一个关于病毒感染传播的问题，通过建立数组结构存储位置和距离信息，运用DFS算法模拟病毒扩散过程，以及注意事项如数组大小调整。

摘要由CSDN通过智能技术生成

问题描述：

思路概述：

1）准备工作：

（1）设立一个数组vector<pair<double, double>> p，存储每个人的位置；

（2）设立一个数组vector<vector<int>> a，记录每个人所对应的距离小于D的人都有谁；

（3）设立一个数组bool vis[12345] = { 0 }，记录每个人是否被污染；

2）搜索：

（1）第一个人已经被污染了，vis[0]=1；dfs（0）；

（2）遍历a[0]数组，都数组里的人的vis赋值为1，并赋值后直接开始dfs；

3）优化：

已经被污染的人不重新进行污染和遍历；

4）注意事项：

读入n之后，需要对vector<vector<int>> a重新定义数组大小，或者初始化时数组大小设为n；

满分代码：

#include<bits/stdc++.h>
using namespace std;
int n;
int D;
bool vis[12345] = { 0 };
pair<int, int> pq;
vector<pair<double, double>> p;
#define w1 first
#define w2 second
vector<vector<int>> a;

void dfs(int dep)
{
	for (int &i : a[dep])
	{
		if (vis[i] == 1)continue;
		vis[i] = 1;
		dfs(i);
	}
	return;
}

int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	cin >> n;
  a.resize(n);
	for (int i = 1; i <= n; i++)
	{
		cin >> pq.w1 >> pq.w2;
		p.push_back(pq);
	}
	cin >> D;
	for (int i = 0; i < n - 1; ++i)
	{
		for (int j = i + 1; j <n; ++j)
		{
			if ((int)(pow(p[i].w1 - p[j].w1, 2)) <= D*D - (int)(pow(p[i].w2 - p[j].w2, 2)))
			{
				a[i].push_back(j);
				a[j].push_back(i);
			}
		}
	}
	vis[0] = 1;
	dfs(0);
	for (int i = 0; i<n; i++)
	{
		cout << vis[i] << "\n";
	}
	return 0;
}