倒计时47天!

深度优先搜索——DFS

温馨提示:学习dfs之前最好先了解一下递归的思想。

DFS基础——图的遍历

仙境诅咒

问题描述

在一片神秘的仙境中,有N位修仙者,他们各自在仙境中独立修炼,拥有自己独特的修炼之道和修炼之地,修仙者们彼此之间相互尊重、和谐相处。

然而,有一天,仙境的主宰者妮妮(第一位修仙者)受到了诅咒,该诅咒会向距离妮妮不超过D的范围内的修仙者传播。也就是说,如果一个修仙者被诅咒,那么在距离他不超过D的范围内的所有修仙者都会被诅咒。

现在,你需要预测哪些修仙者最终会被诅咒,以便及时采取措施,保护仙境的和平与安宁。

输入格式

第一行输入一个正整数

N

(

1

<

N

1

0

3

)

N(1

N(1

接下来N行,每行两个实数

X

i

X_i

Xi​和

Y

i

Y_i

Yi​$ (-103≤X_i,Y_i≤103)

,表示第

i

位修仙者的坐标

,表示第i位修仙者的坐标

,表示第i位修仙者的坐标(X_i,Y_i)$。第一位修仙者即仙境的主宰者妮妮。

最后一行输入一个正整数

D

(

1

<

=

D

<

=

1

0

3

)

D (1<=D<= 10^3)

D(1<=D<=103),表示诅咒传播的范围。

输出格式

输出N行,每行一个整数,第i行的整数为1表示第i位修仙者最终被诅咒,为0则表示第i位修仙者没有被诅咒。

样例输入

5

0 0

1 1

0 1

1 0

2 2

1

样例输出

1

1

1

1

0

题目分析

距离被诅咒者距离不超过D是其它修仙者都会被诅咒感染,也就是我可以从当前被诅咒者走到距离不超过D的其它修仙者。我们可以用数组v[i]=1表示修仙者i已经被诅咒。那么dfs过程代码如下,

private static void dfs(int u) {

v[u] = 1;

for(int i = 1;i <=n;i++)

if(v[i]==0&&dis(u,i)<=d)

dfs(i);

}

dfs(u)这里的u是已经被诅咒的修仙者,那么v[u]就要被标记为1,然后for循环遍历其它修仙者,如果其它修仙者没有被诅咒,并且与当前节点u的距离小于d,那么说明当前修仙者会被传染成为新的被诅咒者,这个时候就要进入dfs(i)去看i能传染给哪些人。

为什么要判断v[i]==0?防止重复遍历,比如我从节点2进入了节点3,即dfs(2)进入了dfs(3),在dfs(3)运行时,我判断了dis(2,3)<=d,如果我没有v[i]==0的约束,我会从dfs(3)进入dfs(2),再从dfs(2)进入dfs(3),最终产生了死循环。

dis函数就是已知两点坐标求两点距离的公式,很简单,但是注意,这里有开根号,那么会有小数,在定义变量的时候要注意变量的类型。

private static double dis(int u, int v) {

return Math.sqrt(Math.pow(x[u]-x[v], 2)+Math.pow(y[u]-y[v], 2));

}

最后通过数组v的值是否为1,可以判断当前点是否被传染。

for(int i = 1;i <=n;i++) System.out.println(v[i]==0?0:1);

题目代码

import java.util.Scanner;

public class Main{

static int n,d;

static double x[],y[];

static int v[];

public static void main(String[] args) {

Scanner scanner = new Scanner(System.in);

n = scanner.nextInt();

x = new double[n+1];

y = new double[n+1];

v = new int[n+1];

for(int i = 1;i <= n;i++) {

x[i] = scanner.nextInt();

y[i] = scanner.nextInt();

}

d = scanner.nextInt();

dfs(1);

for(int i = 1;i <=n;i++) System.out.println(v[i]==0?0:1);

}

private static void dfs(int u) {

// TODO Auto-generated method stub

v[u] = 1;

for(int i = 1;i <=n;i++)

if(v[i]==0&&dis(u,i)<=d)

dfs(i);

}

private static double dis(int u, int v) {

// TODO Auto-generated method stub

return Math.sqrt(Math.pow(x[u]-x[v], 2)+Math.pow(y[u]-y[v], 2));

}

}

相关链接

评论可见,请评论后查看内容,谢谢!!!评论后请刷新页面。