mirror of
https://github.com/TheAlgorithms/Java.git
synced 2025-07-27 06:23:08 +08:00
Add QuadTree
data structure (#5681)
This commit is contained in:
@ -0,0 +1,58 @@
|
||||
package com.thealgorithms.datastructures.trees;
|
||||
|
||||
import java.util.List;
|
||||
import org.junit.jupiter.api.Assertions;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
public class QuadTreeTest {
|
||||
int quadTreeCapacity = 4;
|
||||
BoundingBox boundingBox = new BoundingBox(new Point(0, 0), 500);
|
||||
QuadTree quadTree = new QuadTree(boundingBox, quadTreeCapacity);
|
||||
|
||||
@Test
|
||||
public void testNullPointInsertIntoQuadTree() {
|
||||
Assertions.assertFalse(quadTree.insert(null));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInsertIntoQuadTree() {
|
||||
Assertions.assertTrue(quadTree.insert(new Point(10, -10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(-10, 10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(-10, -10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(10, 10)));
|
||||
Assertions.assertFalse(quadTree.insert(new Point(1050, 1050)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInsertIntoQuadTreeAndSubDivide() {
|
||||
Assertions.assertTrue(quadTree.insert(new Point(10, -10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(-10, 10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(-10, -10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(10, 10)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(-100, 100)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(100, -101)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(-100, -100)));
|
||||
Assertions.assertTrue(quadTree.insert(new Point(100, 100)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testQueryInQuadTree() {
|
||||
quadTree.insert(new Point(10, -10));
|
||||
quadTree.insert(new Point(-10, 10));
|
||||
quadTree.insert(new Point(-10, -10));
|
||||
quadTree.insert(new Point(10, 10));
|
||||
quadTree.insert(new Point(-100, 100));
|
||||
quadTree.insert(new Point(100, -100));
|
||||
quadTree.insert(new Point(-100, -100));
|
||||
quadTree.insert(new Point(100, 100));
|
||||
|
||||
List<Point> points = quadTree.query(new BoundingBox(new Point(0, 0), 100));
|
||||
Assertions.assertEquals(8, points.size());
|
||||
|
||||
points = quadTree.query(new BoundingBox(new Point(5, 5), 5));
|
||||
Assertions.assertEquals(1, points.size());
|
||||
|
||||
points = quadTree.query(new BoundingBox(new Point(-200, -200), 5));
|
||||
Assertions.assertEquals(0, points.size());
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user