问题描述:
思路概述:
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;
}