코테 떨어질만한 코드인지... 평가 부탁드려요

Question

첫번째 문제는 N개의 자연수의 위치를 swap하여 인접한 원소의 차가 k 이하가 되게 하는 문제입니다. N의 개수가 8이라서 부르프 포스로 모든 순열을 생성후 최소의 swap을 찾았습니다.import java.util.*; public class Solution { public static void main(String[] args) { int[] number1 = {10, 40, 30, 20}; int[] number2 = {3, 7, 2, 8, 6, 4, 5, 1}; int solution1 = solution(number1, 20); System.out.println("solution1 = " + solution1); int solution2 = solution(number2, 3); System.out.println("solution2 = " + solution2); } public static int solution(int[] numbers, int k) { List> allPerms = generatePermutations(numbers); int minSwaps = Integer.MAX_VALUE; for (List perm : allPerms) { if (isConditionSatisfied(perm, k)) { int swaps = countMinSwaps(numbers, perm); minSwaps = Math.min(minSwaps, swaps); } } return minSwaps == Integer.MAX_VALUE ? -1 : minSwaps; } private static List> generatePermutations(int[] arr) { List> result = new ArrayList<>(); permute(result, new ArrayList<>(), arr, new boolean[arr.length]); return result; } private static void permute(List> result, List temp, int[] arr, boolean[] used) { if (temp.size() == arr.length) { result.add(new ArrayList<>(temp)); } else { for (int i = 0; i < arr.length; i++) { if (used[i]) continue; used[i] = true; temp.add(arr[i]); permute(result, temp, arr, used); used[i] = false; temp.remove(temp.size() - 1); } } } private static boolean isConditionSatisfied(List arr, int k) { for (int i = 0; i < arr.size() - 1; i++) { if (Math.abs(arr.get(i) - arr.get(i + 1)) > k) return false; } return true; } private static int countMinSwaps(int[] original, List perm) { int[] arr = Arrays.copyOf(original, original.length); Map valueToIndex = new HashMap<>(); for (int i = 0; i < arr.length; i++) { valueToIndex.put(arr[i], i); } int swaps = 0; for (int i = 0; i < arr.length; i++) { if (arr[i] != perm.get(i)) { swaps++; int idx = valueToIndex.get(perm.get(i)); valueToIndex.put(arr[i], idx); int temp = arr[i]; arr[i] = arr[idx]; arr[idx] = temp; } } return swaps; } }두번째 문제는 time < k일때 해당 노드를 추가하고 dfs로 해당 위치까지 갈 수 있는지 판별하는 문제입니다.import java.util.*; public class Solution { public static void main(String[] args) { int n1 = 6, k1 = 17; int[][] roads1 = {{5, 4, 6}, {5, 2, 5}, {0, 4, 2}, {2, 3, 3}, {1, 2, 7}, {0, 1, 3}}; System.out.println(solution(n1, k1, roads1)); int n2 = 4, k2 = 10; int[][] roads2 = {{0, 1, 2}, {0, 2, 3}}; System.out.println(solution(n2, k2, roads2)); int n3 = 4, k3 = 11; int[][] roads3 = {{0, 1, 2}, {1, 2, 7}, {2, 3, 10}, {3, 0, 9}}; System.out.println(solution(n3, k3, roads3)); } public static int[] solution(int n, int k, int[][] roads) { // 그래프 구성 List> graph = new ArrayList<>(); for (int i = 0; i < n; i++) { graph.add(new ArrayList<>()); } for (int[] road : roads) { int a = road[0], b = road[1], t = road[2]; graph.get(a).add(new Pair(b, t)); graph.get(b).add(new Pair(a, t)); } Set reachableStations = new HashSet<>(); dfs(0, k, graph, 0, reachableStations); // 결과 반환 if (reachableStations.isEmpty()) { return new int[]{-1}; } List resultList = new ArrayList<>(reachableStations); Collections.sort(resultList); int[] answer = new int[resultList.size()]; for (int i = 0; i < resultList.size(); i++) { answer[i] = resultList.get(i); } return answer; } private static void dfs(int node, int k, List> graph, int time, Set reachableStations) { if (time == k) { reachableStations.add(node); return; } else if (time > k) { return; } for (Pair next : graph.get(node)) { dfs(next.node, k, graph, time + next.time, reachableStations); } } private static class Pair { int node, time; Pair(int node, int time) { this.node = node; this.time = time; } } }문제를 그대로 올리면 안된다 그래서 제가 쓴 코드만 작성합니다...코테 이후로 왜 떨어졌는지에 대한 피드백을 못받아서요..어느 부분을 개선해야 할 지 고견 부탁드립니다.

Hide_D · Answer

동작보다도 외부에서 키보드로 받든(stdin) 파일로 받든 데이터를 입력하는 부분이 없는데, 그 부분이 크게 작용한게 아닐까요?

코테 떨어질만한 코드인지... 평가 부탁드려요

1개의 답변

Q&A 목록