diff --git a/src/comp1110/ass2/State.java b/src/comp1110/ass2/State.java index b1d364f..bc7eeed 100644 --- a/src/comp1110/ass2/State.java +++ b/src/comp1110/ass2/State.java @@ -561,56 +561,48 @@ public class State { */ public int scoreLinks(int playerID) { - int numOfIslands = 0; - Coord[] playerCoords = players[playerID].getPieces(); // playerCoords + Coord[] playerCoords = players[playerID].getPieces(); Set playerCoordsSet = new HashSet<>(Arrays.asList(playerCoords)); - Set playerLongestLink = findLongestLink(playerCoordsSet, islands); + int maxScore = findLongestLinkScore(playerCoordsSet, islands); - outerLoop: - for(Island island : islands) { - for ( Coord playerCoord : playerLongestLink ) { - if (island.containsCoord(playerCoord)) { - numOfIslands++; - continue outerLoop; - } - } - } - - return numOfIslands * 5; + return maxScore; } - public static Set findLongestLink ( Set allCoords, Island[] islands) { + public static int findLongestLinkScore ( Set allCoords, Island[] islands) { Set longest = new HashSet<>(); Set current = new HashSet<>(); Coord now; + int max = scoreForLink(longest,islands); + for(Island islandIter : islands) { for ( Coord coords : allCoords) { if(islandIter.containsCoord(coords)) { now = coords; current.add(coords); - recurssionLink(allCoords, current, longest, now); + recursionLink(allCoords, current, longest, now); // if the score is bigger, update the Set - if(scoreForLink(current, islands) > scoreForLink(longest,islands)) { + if(scoreForLink(current, islands) > max) { longest.clear(); longest.addAll(current); + max = scoreForLink(longest, islands); } current = new HashSet<>(); } } } - return longest; + return max; } - public static void recurssionLink (Set allCoords, Set current, Set longest, Coord now) { + public static void recursionLink(Set allCoords, Set current, Set longest, Coord now) { current.add(now); for(Coord c : allCoords) { - if( (now.isAdjacent(c) || now.isAdjacentDiagonal(c)) && !current.contains(c) ) { - recurssionLink(allCoords, current, longest, c); + if( now.isAdjacentDiagonal(c) && !current.contains(c) ) { + recursionLink(allCoords, current, longest, c); } } }