1028. 语句匹配

Description

Pascal语言中，复合语句用begin...end表示，条件语句用if...then...else...表示，其中，else子句可以出现也可以不出现。现在提取出一些语句中的所有begin、end、if、then、else，编写一个程序检查它们能否匹配，以构成若干条合法的语句。

Input Format

输入包含两行。

第1行：一个整数N，表示接下来有N个字符串将读入。

第2行：包含N个字符串，每个字符串一定是{"begin","end","if","then","else"}之一（不包括引号）。这些字符串表示从一系列语句中依次提取出的所有元素。

Output Format

若输入可以构成若干条合法语句，输出一行"YES"，否则输出一行"NO"。（不包含引号）

Sample Input

9
if then begin if then begin end end else

Sample Output

YES

Sample Input

4
if begin end then

Sample Output

NO

说明

N≤100

注意begin...end不能作为判断条件。（见样例2）

注意then与else之后都允许出现不止一条if语句或复合语句。

#include<iostream>
#include<cstring>
using namespace std;

/*
1、输入完成后：num（前置）==num（后置）
2、输入过程中：num（前置）>=num (后置）
此条件同时包含了对else的判断
3、输入过程中：
if出现&&then未出现的情况下，不能出现匹配的begin和end
*/
int cmp(char[]);
int bucket[5];

int cmp(char arr[]){
    if(!strcmp(arr,"begin"))return 0;
    if(!strcmp(arr,"end"))return 1;
    if(!strcmp(arr,"if"))return 2;
    if(!strcmp(arr,"then"))return 3;
    if(!strcmp(arr,"else"))return 4;
}

int main(){
    int n;
    cin>>n;
    bool flag=false,isif=false,isbegin=false;
    char arr[101][6];
    for(int i=0;i<n;i++){
        cin>>arr[i];
        bucket[cmp(arr[i])]++;
        if(cmp(arr[i])==2) isif=true;
        if(cmp(arr[i])==3 && isif==true) isif=false;
        if(cmp(arr[i])==0 && isif==true) isbegin=true;
        if(cmp(arr[i])==1 && isif==true && isbegin==true){
            cout<<"NO";
            flag=true;
            break;
        }
        if(bucket[0]<bucket[1] || bucket[2]<bucket[3] || bucket[2]<bucket[4]){
            cout<<"NO";
            flag=true;
            break;
        }
    }
        if(flag==false){
            if(bucket[0]==bucket[1] && bucket[2]==bucket[3]){
                cout<<"YES";
            }
            else {
                cout<<"NO";
            }
        }
        return 0;
    }