数据结构OJ:移除星号(栈)

OJ链接

简而言之:题给函数用指针接收字符串,你需要删除字符串中的星号和星号前的一个字符

这是典型可以用栈解决的问题

思路1:

我们将字符串中的字符一个一个push进栈,如果碰到星号就将前一个元素pop出栈,道理简单。值得注意的是,如果栈里已经没有元素了,遇到星号继续pop出栈就会出错了,所以在出栈时要对栈进行一个判空处理

typedef struct stack
{
    char *arr;
    int capacity;
    int top;
}stack;
void StackInit(stack*s)
{
    s->arr=NULL;
    s->capacity=0;
    s->top=0;
}
void StackPush(stack*s,char data)
{
    if(s->top==s->capacity)
    {
        int newcapacity=s->capacity==0?4:s->capacity*2;
        s->arr=realloc(s->arr,sizeof(int)*newcapacity);
        s->capacity=newcapacity;
    }
    s->arr[s->top]=data;
    s->top++;
}
void StackPop(stack*s)
{
    if(s->top==0)
    return;
    else
    s->top--;
}
char * removeStars(char * s){
    stack st;
    StackInit(&st);
    while(*s)
    {
        if(*s=='*')
        StackPop(&st);
        else
        StackPush(&st,*s);
        s++;
    }
    StackPush(&st,0);
    return st.arr;
}

博主在主函数内没有进行特殊判断是因为博主博主在实现出栈函数的时候进行了判空,所以在主函数内部就不需要再次处理。

事实上老实用栈处理会消耗掉空间和时间,效果不理想。一般在数据结构OJ中需要栈的地方应尽量考虑用数组来模拟栈

模拟栈思路:

考虑使用双指针(快慢指针),这种方法可以原地处理字符串,不需要额外开辟空间,所以空间复杂度是O(1),而思路1的空间复杂度是O(N).

快指针和慢指针的起始位置都为s[0],如果fast位置不为星号就将值赋给slow位置,fast和slow自增,

注意:如果fast位置是星号,则slow只需要回退即可,fast自增,同样的,如果slow退到s[0]也不能继续回退了。最后不要忘记末尾的'\0'需要自己加上去

ac代码:

char * removeStars(char * s){
    int slow=0;
    int fast=0;
    while(s[fast])
    {
        if(slow==0&&s[fast]=='*')
        {
            fast++;
            continue;
        }
        if(s[fast]=='*')
        slow--;
        else
        s[slow++]=s[fast];
        fast++;
    }
    s[slow]=0;
    return s;
}

你可能感兴趣的:(数据结构OJ,c语言,数据结构,算法)