Handle incorrect inputs in StackPostfixNotation (#4261)

This commit is contained in:
Piotr Idzik
2023-07-26 08:20:56 +02:00
committed by GitHub
parent dec3b98e4b
commit 44dcebb699
2 changed files with 43 additions and 10 deletions

View File

@ -1,18 +1,14 @@
package com.thealgorithms.others; package com.thealgorithms.others;
import java.util.*; import java.util.Scanner;
import java.util.Stack;
public class StackPostfixNotation { public final class StackPostfixNotation {
private StackPostfixNotation() {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
String post = scanner.nextLine(); // Takes input with spaces in between eg. "1 21 +"
System.out.println(postfixEvaluate(post));
scanner.close();
} }
// Evaluates the given postfix expression string and returns the result. // Evaluates the given postfix expression string and returns the result.
public static int postfixEvaluate(String exp) { public static int postfixEvaluate(final String exp) {
Stack<Integer> s = new Stack<Integer>(); Stack<Integer> s = new Stack<Integer>();
Scanner tokens = new Scanner(exp); Scanner tokens = new Scanner(exp);
@ -28,12 +24,16 @@ public class StackPostfixNotation {
case "+" -> s.push(num1 + num2); case "+" -> s.push(num1 + num2);
case "-" -> s.push(num1 - num2); case "-" -> s.push(num1 - num2);
case "*" -> s.push(num1 * num2); case "*" -> s.push(num1 * num2);
default -> s.push(num1 / num2); case "/" -> s.push(num1 / num2);
default -> throw new IllegalArgumentException("exp contains an unknown operation.");
} }
// "+", "-", "*", "/" // "+", "-", "*", "/"
} }
} }
tokens.close(); tokens.close();
if (s.size() != 1) {
throw new IllegalArgumentException("exp is not a proper postfix expression.");
}
return s.pop(); return s.pop();
} }
} }

View File

@ -0,0 +1,33 @@
package com.thealgorithms.others;
import static java.util.Map.entry;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import java.util.Map;
import org.junit.jupiter.api.Test;
public class StackPostfixNotationTest {
@Test
public void testEvaluate() {
final Map<String, Integer> testCases = Map.ofEntries(entry("1 1 +", 2), entry("2 3 *", 6), entry("6 2 /", 3), entry("-5 -2 -", -3), entry("5 2 + 3 *", 21), entry("-5", -5));
for (final var tc : testCases.entrySet()) {
assertEquals(tc.getValue(), StackPostfixNotation.postfixEvaluate(tc.getKey()));
}
}
@Test
public void testIfEvaluateThrowsExceptionForEmptyInput() {
assertThrows(IllegalArgumentException.class, () -> StackPostfixNotation.postfixEvaluate(""));
}
@Test
public void testIfEvaluateThrowsExceptionForInproperInput() {
assertThrows(IllegalArgumentException.class, () -> StackPostfixNotation.postfixEvaluate("3 3 3"));
}
@Test
public void testIfEvaluateThrowsExceptionForInputWithUnknownOperation() {
assertThrows(IllegalArgumentException.class, () -> StackPostfixNotation.postfixEvaluate("3 3 !"));
}
}