mirror of
https://github.com/TheAlgorithms/Java.git
synced 2025-07-09 03:59:38 +08:00
Handle incorrect inputs in StackPostfixNotation
(#4261)
This commit is contained in:
@ -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();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -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 !"));
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user