#define OK 0
#define NULL 0

typedef int DataType;
typedef struct node{
    DataType data;
    struct node * next;
}Stack;

//创建栈，此时栈中没有任何元素
Stack* CreateStack()
{
    Stack *stack = (Stack*)malloc(sizeof(Stack));
    if(NULL != stack)
    {
       stack->next = NULL;
       return stack;
    }
    return NULL;
}

//撤销栈
void DestoryStack(Stack* stack)
{
    free(stack);
}
 
int IsEmpty(Stack* stack)
{
    return (stack->next == 0);
}
 
//入栈，成功返回1，失败返回0， 把元素 data 存入 栈 stack 中
int PushStack(Stack* stack, DataType data)
{
    Stack* newst = (Stack*)malloc(sizeof(Stack));
    if(NULL != newst)
    {
        newst->data = data;
        newst->next = stack->next;  //s->next = NULL;
        stack->next = newst;
        return 1;
    }
    return 0;
}
 
/*
    出栈，成功返回1，失败返回0，出栈不取出元素值，只是删除栈顶元素。
    如出栈要实现，取出元素值，并释放空间，可结合取栈顶元素函数做修改，这里不再给出。
 */
 
int PopStack(Stack* stack)
{
    Stack* tmpst;
    if(!IsEmpty(stack))
    {
        tmpst = stack->next;
        stack->next = tmpst->next;
        free(tmpst);
        return 1;
    }
    return 0;
}

int PopStackReturn(Stack* stack, DataType *data)
{
    Stack* tmpst;
    if(!IsEmpty(stack))
    {
        tmpst = stack->next;
        stack->next = tmpst->next;
        *data = tmpst->data;
        free(tmpst);
        return 1;
    }
    return 0;
}
 
//取栈顶元素，仅取出栈顶元素的值，取出之后，该元素，任然存在栈中。成功返回元素值，失败输出提示信息，并返回 -1
DataType GetTopElement(Stack* stack)
{
    if(!IsEmpty(stack))
    {
        return stack->next->data;
    }
    return -1;
}


int* dailyTemperatures(int* T, int TSize, int* returnSize) 
{
    int i;
    int temp;
    int topData;
    int ret;
    int *NewT = malloc(TSize * sizeof(int));
    if (NewT == NULL) 
    {
        printf("malloc fail");
        return NULL;
    }
    memset(NewT, 0, TSize * sizeof(int));

    Stack* stack = CreateStack();
    if (stack == NULL) 
    {
        printf("CreateStack fail");
        return NULL;
    }
    for (i = 0; i < TSize; i++) 
    {
        while (!IsEmpty(stack) && (T[i] > T[GetTopElement(stack)])) 
        {
            temp = GetTopElement(stack);
            NewT[temp] = i - temp;
            PopStack(stack);
        }
        PushStack(stack, i);
    }
    
    DestoryStack(stack);
    *returnSize = TSize;
    return NewT;
}