括号匹配（栈）

Posted 2021-03-15 qdu-lkc

检查一段C语言代码的小括号( )、 中括号 [ ] 和大括号{ } 是否匹配。

输入格式:

在一行中输入一段C语言代码，长度不超过1000个字符（行末以换行符结束）。

输出格式:

第一行输出左括号的数量和右括号的数量，中间以一个空格间隔。
若括号是匹配的，在第二行打印YES，否则打印NO。

输入样例1:

for(int i=0; i<v; i++){ visited[i] = 0; for(int j=0; j<v; j++) scanf("%d",&(g->Adj[i][j])); }

 

输出样例1:

8 8
YES

 

输入样例2:

for(int i=0; i<v; i++) a(i]=0;

 

输出样例2:

2 2
NO

思路：
  注意两点：如果没有右括号与之匹配，不要break，要继续统计括号的数量；满足左括号都有右括号与之匹配，这时候可能有多余的左括号，因此不仅要判断flag=true，还要加上栈为空
代码：

import java.util.*;
public class Main{
      public static void main(String[] args) {
            Scanner scan=new Scanner(System.in);
            Stack<Character> sta=new Stack<Character>();
            Map<Character, Character> map=new HashMap<Character, Character>();
            map.put(‘(‘, ‘)‘);
            map.put(‘[‘, ‘]‘);
            map.put(‘{‘, ‘}‘);
            String s=scan.nextLine();
            char[] a=s.toCharArray();
            boolean flag=true;
            int cnt1=0,cnt2=0;
            for(int i=0;i<a.length;i++){
                if(a[i]==‘(‘ ||a[i]==‘[‘ ||a[i]==‘{‘)  {
                      sta.push(a[i]);
                      cnt1++;
                }
                else if(a[i]==‘)‘ ||a[i]==‘]‘ ||a[i]==‘}‘){
                      cnt2++;
                      if(!sta.isEmpty() && map.get(sta.peek())==a[i]){
                            sta.pop();
                      }
                     else{
                           flag=false;//不要break，因为需要继续统计数量
                     }
                }
            }
            System.out.println(cnt1+" "+cnt2);
            if(flag && sta.isEmpty()) System.out.println("YES");//判栈空
            else System.out.println("NO");
    }
}