題意:給出一個只包含 ‘(‘, ‘)’, ‘{‘, ‘}’, ‘[’ 和 ‘]’ 的字符串,判斷它的輸入是否是有效的。
例子:括號必須是以正確的順序關閉的, “()” 和 “()[]{}” 都是有效的,但是 “(]” 和 “([)]” 是無效的。
分析:判斷括號匹配的合法性。需要使用一個棧來解決問題。遇到左括號入棧,遇到右括號,檢查棧頂的左括號是否匹配,如果匹配,彈棧,如果不匹配,返回錯誤。
需要注意的是如果棧爲空,而遇到右括號,同樣返回錯誤。遍歷完後,棧如果不空,同樣返回錯誤。因此在遇到右括號時,需要判斷棧是否爲空。
下圖是手擼的代碼,提交過一次沒有判斷棧是否爲空,因此stack.peek()方法錯誤,因此又加上遇到右括號時,需要判斷棧是否爲空,才得以通過。
import java.util.Stack;
class Solution {
public static boolean isValid(String s) {
int len = s.length();
Stack<Character> stack = new Stack<>();
for(int i = 0; i < len; i++){
if(s.charAt(i) == '(' || s.charAt(i) == '[' || s.charAt(i) == '{'){
stack.push(s.charAt(i));
}else{
if(stack.isEmpty()){
return false;
}
else if(s.charAt(i) == ')' && stack.peek() == '('){
stack.pop();
}else if(s.charAt(i) == ']' && stack.peek() == '['){
stack.pop();
}else if(s.charAt(i) == '}' && stack.peek() == '{'){
stack.pop();
}else{
return false;
}
}
}
return stack.isEmpty();
}
public static void main(String[] args) {
String s = "]";
System.out.println(isValid(s));
}
}
可以看出來,代碼很醜陋。因此去看了discuss的高票答案:當遇到左括號時,將對應的右括號入棧,這樣代碼就可以變得很簡潔:
public boolean isValid(String s) {
Stack<Character> stack = new Stack<Character>();
for (char c : s.toCharArray()) {
if (c == '(')
stack.push(')');
else if (c == '{')
stack.push('}');
else if (c == '[')
stack.push(']');
else if (stack.isEmpty() || stack.pop() != c)
return false;
}
return stack.isEmpty();
}