refactor 582

fishercoder1534 · fishercoder1534 · commit afd6cb2df9f1 · 2019-07-15T08:47:46.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_582.java b/src/main/java/com/fishercoder/solutions/_582.java
@@ -1,6 +1,5 @@
 package com.fishercoder.solutions;
 
-
 import java.util.ArrayDeque;
 import java.util.Deque;
 import java.util.HashMap;
@@ -41,24 +40,26 @@
  */
 public class _582 {
 
-    public List<Integer> killProcess(List<Integer> pid, List<Integer> ppid, int kill) {
-        Map<Integer, List<Integer>> map = new HashMap<>();
-        for (int i = 0; i < pid.size(); i++) {
-            map.putIfAbsent(ppid.get(i), new LinkedList<>());
-            map.get(ppid.get(i)).add(pid.get(i));
-        }
-        List<Integer> result = new LinkedList<>();
-        Deque<Integer> stack = new ArrayDeque<>();
-        stack.offer(kill);
-        while (!stack.isEmpty()) {
-            int curr = stack.poll();
-            result.add(curr);
-            List<Integer> list = map.get(curr);
-            if (list != null) {
-                stack.addAll(list);
+    public static class Solution1 {
+        public List<Integer> killProcess(List<Integer> pid, List<Integer> ppid, int kill) {
+            Map<Integer, List<Integer>> map = new HashMap<>();
+            for (int i = 0; i < pid.size(); i++) {
+                map.putIfAbsent(ppid.get(i), new LinkedList<>());
+                map.get(ppid.get(i)).add(pid.get(i));
+            }
+            List<Integer> result = new LinkedList<>();
+            Deque<Integer> stack = new ArrayDeque<>();
+            stack.offer(kill);
+            while (!stack.isEmpty()) {
+                int curr = stack.poll();
+                result.add(curr);
+                List<Integer> list = map.get(curr);
+                if (list != null) {
+                    stack.addAll(list);
+                }
             }
+            return result;
         }
-        return result;
     }
 
 }
diff --git a/src/test/java/com/fishercoder/_582Test.java b/src/test/java/com/fishercoder/_582Test.java
@@ -13,15 +13,15 @@
  * Created by fishercoder on 5/18/17.
  */
 public class _582Test {
-    private static _582 test;
+    private static _582.Solution1 solution1;
     private static List<Integer> pid;
     private static List<Integer> ppid;
     private static List<Integer> expected;
     private static Integer kill;
 
     @BeforeClass
     public static void setup() {
-        test = new _582();
+        solution1 = new _582.Solution1();
     }
 
     @Test
@@ -30,7 +30,7 @@ public void test1() {
         ppid = Arrays.asList(3, 0, 5, 3);
         kill = 5;
         expected = Arrays.asList(5, 10);
-        assertEquals(expected, test.killProcess(pid, ppid, kill));
+        assertEquals(expected, solution1.killProcess(pid, ppid, kill));
     }
 
     @Test
@@ -39,6 +39,6 @@ public void test2() {
         ppid = Arrays.asList(3, 0, 5, 3);
         kill = 3;
         expected = Arrays.asList(3, 1, 5, 10);
-        assertEquals(expected, test.killProcess(pid, ppid, kill));
+        assertEquals(expected, solution1.killProcess(pid, ppid, kill));
     }
 }
