algorithm/Java
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Java.git synced 2025-07-27 06:23:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add Gale-Shapley Algorithm and Tests (#5494)

Browse Source
This commit is contained in:
B Karthik
2024-10-02 19:20:30 +05:30
committed by GitHub
parent dab8ff3d1d
commit e6f597acaf
3 changed files with 139 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
src/test/java/com/thealgorithms/greedyalgorithms/GaleShapleyTest.java Normal file
View File

@ -0,0 +1,72 @@
package com.thealgorithms.greedyalgorithms;
import static org.junit.jupiter.api.Assertions.assertEquals;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.Test;
public class GaleShapleyTest {
@Test
public void testStableMatch() {
Map<String, LinkedList<String>> womenPrefs = new HashMap<>();
womenPrefs.put("A", new LinkedList<>(List.of("X", "Y", "Z")));
womenPrefs.put("B", new LinkedList<>(List.of("Y", "X", "Z")));
womenPrefs.put("C", new LinkedList<>(List.of("X", "Y", "Z")));
Map<String, LinkedList<String>> menPrefs = new HashMap<>();
menPrefs.put("X", new LinkedList<>(List.of("A", "B", "C")));
menPrefs.put("Y", new LinkedList<>(List.of("B", "A", "C")));
menPrefs.put("Z", new LinkedList<>(List.of("A", "B", "C")));
Map<String, String> result = GaleShapley.stableMatch(womenPrefs, menPrefs);
Map<String, String> expected = new HashMap<>();
expected.put("A", "X");
expected.put("B", "Y");
expected.put("C", "Z");
assertEquals(expected, result);
}
@Test
public void testSinglePair() {
Map<String, LinkedList<String>> womenPrefs = new HashMap<>();
womenPrefs.put("A", new LinkedList<>(List.of("X")));
Map<String, LinkedList<String>> menPrefs = new HashMap<>();
menPrefs.put("X", new LinkedList<>(List.of("A")));
Map<String, String> result = GaleShapley.stableMatch(womenPrefs, menPrefs);
Map<String, String> expected = new HashMap<>();
expected.put("A", "X");
assertEquals(expected, result);
}
@Test
public void testEqualPreferences() {
Map<String, LinkedList<String>> womenPrefs = new HashMap<>();
womenPrefs.put("A", new LinkedList<>(List.of("X", "Y", "Z")));
womenPrefs.put("B", new LinkedList<>(List.of("X", "Y", "Z")));
womenPrefs.put("C", new LinkedList<>(List.of("X", "Y", "Z")));
Map<String, LinkedList<String>> menPrefs = new HashMap<>();
menPrefs.put("X", new LinkedList<>(List.of("A", "B", "C")));
menPrefs.put("Y", new LinkedList<>(List.of("A", "B", "C")));
menPrefs.put("Z", new LinkedList<>(List.of("A", "B", "C")));
Map<String, String> result = GaleShapley.stableMatch(womenPrefs, menPrefs);
Map<String, String> expected = new HashMap<>();
expected.put("A", "X");
expected.put("B", "Y");
expected.put("C", "Z");
assertEquals(expected, result);
}
}