#include "stack_.h"
 
// 栈表操作实现文件
//
 
 
Stack* Stack_Create(int maxlen)
// 创建栈
{
    Stack* stk = (Stack*)malloc(sizeof(Stack));
    stk->data = (T*)malloc(sizeof(T)*maxlen);
    stk->max = maxlen;
    stk->top = -1;
    return stk;
}
 
void Stack_Free(Stack* stk)
// 释放栈
{
    free(stk->data);
    free(stk);
}
 
void Stack_MakeEmpty(Stack* stk)
// 置为空栈
{
    stk->top = -1;
}
 
bool Stack_IsEmpty(Stack* stk)
// 判断栈是否空
{
    return -1 == stk->top;
}
 
bool Stack_IsFull(Stack* stk)
// 判断栈是否满
{
    return stk->top == stk->max-1;
}
 
T Stack_Top(Stack* stk)
// 获取当前栈顶元素
{
    return stk->data[stk->top];
}
 
T Stack_Push(Stack* stk, T e)
// 将元素e压入栈顶
// 返回栈顶点元素
{
    if(Stack_IsFull(stk)) {
        printf("Stack_IsFull(): stack full error when push element to the stack!\n");
        Stack_Free(stk);
        exit(0);
    }
    else{
        stk->top += 1;
        stk->data[stk->top] = e;
        return Stack_Top(stk);
    }
}
 
T Stack_Pop(Stack* stk)
// 将栈顶元素出栈
// 返回栈顶元素
{
    if(Stack_IsEmpty(stk)) {
        printf("Stack_IsEmpty(): stack empty error when pop element of the stack top!\n");
        Stack_Free(stk);
        exit(0);
    }
    else{
        T topE = Stack_Top(stk);
        stk->top -= 1;
        return topE;
    }
}
 
void Stack_Print(Stack* stk)
// 打印栈顶到栈低的元素
{
    if (Stack_IsEmpty(stk)) {
        printf("The stack is empty.\n");
        return;
    }
    
    //printf("The stack contains: ");
    for (int i=stk->top; i>=0; i--) {
        printf("%d", stk->data[i]);
    }
    printf("\n");
    
}
 

void Bracket_Match(T* str, int len)
// 利用stack栈判断括号是否匹配
// 输入参数：字符串序列，字符串长度
// 若匹配输出YES，否则输出NO，末尾换行
{
    // 创建一个栈实例
    Stack *stk = Stack_Create(32); // 假设32足够存储所有括号

    for (int j = 0; j < len; j++)
    {
        if (str[j] == '{' || str[j] == '(' || str[j] == '[')
        {
            // 如果是左括号，压入栈中
            Stack_Push(stk, str[j]);
        }
        else if (str[j] == '}' || str[j] == ')' || str[j] == ']')
        {
            // 如果是右括号，检查栈顶元素是否匹配
            if (Stack_IsEmpty(stk))
            {
                // 如果栈为空，说明没有相应的左括号与之匹配
                printf("NO\n");
                Stack_Free(stk);
                return;
            }
            else
            {
                // 弹出栈顶元素
                T topElement = Stack_Pop(stk);
                // 检查是否匹配
                if ((topElement == '{' && str[j] != '}') ||
                    (topElement == '(' && str[j] != ')') ||
                    (topElement == '[' && str[j] != ']'))
                {
                    // 如果不匹配，输出NO
                    printf("NO\n");
                    Stack_Free(stk);
                    return;
                }
            }
        }
        else
        {
            // 如果字符不是括号，则忽略
        }
    }

    // 检查栈是否为空，如果为空说明所有括号都匹配
    if (Stack_IsEmpty(stk))
    {
        printf("YES\n");
    }
    else
    {
        // 如果栈不为空，说明仍有未匹配的左括号
        printf("NO\n");
    }

    // 释放栈占用的内存
    Stack_Free(stk);
}