算法大全(2)栈和队列

  声明,本文所有9道算法题目,覆盖了基本上所有常见的栈/队列问题,全都用C#实现,并测试通过,代码下载:StackAndQueue.zip

目录:

1.设计含min函数的栈,要求min、push和pop的时间复杂度都是o(1)。

2.设计含min函数的栈的另解

3.用两个栈实现队列

4.用两个队列实现栈 

5.栈的push、pop序列是否一致

6.递归反转一个栈,要求不得重新申请一个同样的栈,空间复杂度o(1)

7.给栈排个序

8..如何用一个数组实现两个栈

9..如何用一个数组实现三个栈

 

1.设计含min函数的栈,要求min、push和pop的时间复杂度都是o(1)。

  算法思想:需要设计一个辅助栈,用来存储当前栈中元素的最小值。网上有人说存储当前栈中元素的最小值的所在位置,虽然能节省空间,这其实是不对的,因为我在调用Min函数的时候,只能得到位置,还要对存储元素的栈不断的pop,才能得到最小值——时间复杂度o(1)。

  所以,还是在辅助栈中存储元素吧。

  此外,还要额外注意Push操作,第一个元素不用比较,自动成为最小值入栈。其它元素每次都要和栈顶元素比较,小的那个放到栈顶。 

public class NewStack
{
    private Stack dataStack;
    private Stack mindataStack;

    public NewStack()
    {
        dataStack = new Stack();
        mindataStack = new Stack();
    }

    public void Push(int element)
    {
        dataStack.Push(element);

        if (mindataStack.Count == 0)
            mindataStack.Push(element);
        else if (element <= (int)mindataStack.Peek())
            mindataStack.Push(element);
        else //(element > mindataStack.Peek)
            mindataStack.Push(mindataStack.Peek());
    }
    
    public int Pop()
    {
        if (dataStack.Count == 0)
            throw new Exception("The stack is empty");
        
        mindataStack.Pop();
        return (int)dataStack.Pop();
    }

    public int Min()
    {
        if (dataStack.Count == 0)
            throw new Exception("The stack is empty");
        
        return (int)mindataStack.Peek();
    }
}

  

2.设计含min函数的栈的另解

  话说,和青菜脸呆久了,就沾染了上海小市民意识,再加上原本我就很抠门儿,于是对于上一题目,我把一个栈当成两个用,就是说,每次push,先入站当前元素,然后入栈当前栈中最小元素;pop则每次弹出2个元素。

  算法代码如下所示(这里最小元素位于当前元素之上,为了下次比较方便):

public class NewStack
{
    private Stack stack;

    public NewStack()
    {
        stack = new Stack();
    }

    public void Push(int element)
    {
        if (stack.Count == 0)
        {
            stack.Push(element);
            stack.Push(element);
        }
        else if (element <= (int)stack.Peek())
        {
            stack.Push(element);
            stack.Push(element);
        }
        else //(element > stack.Peek)
        {
            object min = stack.Peek();
            stack.Push(element);
            stack.Push(min);            
        }
    }

    public int Pop()
    {
        if (stack.Count == 0)
            throw new Exception("The stack is empty");

        stack.Pop();
        return (int)stack.Pop();
    }

    public int Min()
    {
        if (stack.Count == 0)
            throw new Exception("The stack is empty");

        return (int)stack.Peek();
    }
}  

  之所以说我这个算法比较叩门,是因为我只使用了一个栈,空间复杂度o(N),节省了一半的空间(算法1的空间复杂度o(2N))。

 

 

3.用两个栈实现队列

  实现队列,就要实现它的3个方法:Enqueue(入队)、Dequeue(出队)和Peek(队头)。

1)stack1存的是每次进来的元素,所以Enqueue就是把进来的元素push到stack1中。

2)而对于Dequeue,一开始stack2是空的,所以我们把stack1中的元素全都pop到stack2中,这样stack2的栈顶就是队头。只要stack2不为空,那么每次出队,就相当于stack2的pop。

3)接下来,每入队一个元素,仍然push到stack1中。每出队一个元素,如果stack2不为空,就从stack2中pop一个元素;如果stack2为空,就重复上面的操作——把stack1中的元素全都pop到stack2中。

4)Peek操作,类似于Dequeue,只是不需要出队,所以我们调用stack2的Peek操作。当然,如果stack2为空,就把stack1中的元素全都pop到stack2中。

5)注意边界的处理,如果stack2和stack1都为空,才等于队列为空,此时不能进行Peek和Dequeue操作。

  按照上述分析,算法实现如下:

public class NewQueue
{
    private Stack stack1;
    private Stack stack2;

    public NewQueue()
    {
        stack1 = new Stack();
        stack2 = new Stack();
    }

    public void Enqueue(int element)
    {
        stack1.Push(element);
    }

    public int Dequeue()
    {
        if (stack2.Count == 0)
        {
            if (stack1.Count == 0)
                throw new Exception("The queue is empty");
            else
                while (stack1.Count > 0)
                    stack2.Push(stack1.Pop());
        }

        return (int)stack2.Pop();
    }

    public int Peek()
    {
        if (stack2.Count == 0)
        {
            if (stack1.Count == 0)
                throw new Exception("The queue is empty");
            else
                while (stack1.Count > 0)
                    stack2.Push(stack1.Pop());
        }

        return (int)stack2.Peek();
    }
} 

 

4.用两个队列实现栈

  这个嘛,就要queue1和queue2轮流存储数据了。这个“轮流”发生在Pop和Peek的时候,假设此时我们把所有数据存在queue1中(此时queue2为空),我们把queue1的n-1个元素放到queue2中,queue中最后一个元素就是我们想要pop的元素,此时queue2存有n-1个元素(queue1为空)。

  至于Peek,则是每次转移n个数据,再转移最后一个元素的时候,将其计下并返回。

  那么Push的操作,则需要判断当前queue1和queue2哪个为空,将新元素放到不为空的队列中。

public class NewStack
{
    private Queue queue1;
    private Queue queue2;

    public NewStack()
    {
        queue1 = new Queue();
        queue2 = new Queue();
    }

    public void Push(int element)
    {
        if (queue1.Count == 0)
            queue2.Enqueue(element);
        else
            queue1.Enqueue(element);
    }

    public int Pop()
    {
        if (queue1.Count == 0 && queue2.Count == 0)
            throw new Exception("The stack is empty");

        if (queue1.Count > 0)
        {
            while (queue1.Count > 1)
            {
                queue2.Enqueue(queue1.Dequeue());
            }

            //还剩一个
            return (int)queue1.Dequeue();
        }
        else  //queue2.Count > 0
        {
            while (queue2.Count > 1)
            {
                queue1.Enqueue(queue2.Dequeue());
            }

            //还剩一个
            return (int)queue2.Dequeue();
        }
    }

    public int Peek()
    {
        if (queue1.Count == 0 && queue2.Count == 0)
            throw new Exception("The stack is empty");

        int result = 0;

        if (queue1.Count > 0)
        {
            while (queue1.Count > 1)
            {
                queue2.Enqueue(queue1.Dequeue());
            }

            //还剩一个
            result = (int)queue1.Dequeue();
            queue2.Enqueue(result);

        }
        else  //queue2.Count > 0
        {
            while (queue2.Count > 1)
            {
                queue1.Enqueue(queue2.Dequeue());
            }

            //还剩一个
            result = (int)queue2.Dequeue();
            queue1.Enqueue(result);
        }

        return result;
    }
} 

 

5.栈的push、pop序列是否一致

  输入两个整数序列。其中一个序列表示栈的push顺序,判断另一个序列有没有可能是对应的pop顺序。为了简单起见,我们假设push序列的任意两个整数都是不相等的。

  比如输入的push序列是1、2、3、4、5,那么4、5、3、2、1就有可能是一个pop系列。因为可以有如下的push和pop序列:push 1,push 2,push 3,push 4,pop,push 5,pop,pop,pop,pop,这样得到的pop序列就是4、5、3、2、1。但序列4、3、5、1、2就不可能是push序列1、2、3、4、5的pop序列。

 

  网上的若干算法都太复杂了,现提出包氏算法如下:

  先for循环把arr1中的元素入栈,并在每次遍历时,检索arr2中可以pop的元素。如果循环结束,而stack中还有元素,就说明arr2序列不是pop序列。

static bool JudgeSequenceIsPossible(int[] arr1, int[] arr2)
{
    Stack stack = new Stack();

    for (int i = 0, j = 0; i < arr1.Length; i++)
    {
        stack.Push(arr1[i]);

        while (stack.Count > 0 && (int)stack.Peek() == arr2[j])
        {
            stack.Pop();
            j++;
        }
    }

    return stack.Count == 0;
}

 

 

6.递归反转一个栈,要求不得重新申请一个同样的栈,空间复杂度o(1)

  算法思想:汉诺塔的思想,非常复杂,玩过九连环的人都想得通的

static void ReverseStack(ref Stack stack)
{
    if (stack.Count == 0)
        return;

    object top = stack.Pop();

    ReverseStack(ref stack);

    if (stack.Count == 0)
    {
        stack.Push(top);
        return;
    }

    object top2 = stack.Pop();
    ReverseStack(ref stack);

    stack.Push(top);
    ReverseStack(ref stack);
    stack.Push(top2);        
}
 

7.给栈排个序

  本题目是上一题目的延伸

static void Sort(ref Stack stack)
{
    if (stack.Count == 0)
        return;

    object top = stack.Pop();

    Sort(ref stack);

    if (stack.Count == 0)
    {
        stack.Push(top);
        return;
    }

    object top2 = stack.Pop();
    if ((int)top > (int)top2)
    {
        stack.Push(top);
        Sort(ref stack);
        stack.Push(top2);
    }
    else
    {
        stack.Push(top2);
        Sort(ref stack);
        stack.Push(top);
    }
}

 

 

8..如何用一个数组实现两个栈

  继续我所提倡的抠门儿思想,也不枉我和青菜脸相交一场。

  网上流传着两种方法:

  方法1  采用交叉索引的方法

一号栈所占数组索引为0, 2, 4, 6, 8......(K*2)  
二号栈所占数组索引为1,3,5,7,9 ......(K*2 + 1) 

  算法实现如下:

public class NewStack
{
    object[] arr;
    int top1;
    int top2;

    public NewStack(int capticy)
    {
        arr = new object[capticy];

        top1 = -1;
        top2 = -2;
    }

    public void Push(int type, object element)
    {
        if (type == 1)
        {
            if (top1 + 2 >= arr.Length)
                throw new Exception("The stack is full");
            else
            {
                top1 += 2;
                arr[top1] = element;
            }
        }

        else //type==2
        {
            if (top2 + 2 >= arr.Length)
                throw new Exception("The stack is full");
            else
            {
                top2 += 2;
                arr[top2] = element;
            }
        }
    }

    public object Pop(int type)
    {
        object obj = null;

        if (type == 1)
        {
            if (top1 == -1)
                throw new Exception("The stack is empty");
            else
            {
                obj = arr[top1];
                arr[top1] = null;
                top1 -= 2;
            }
        }

        else //type == 2
        {
            if (top2 == -2)
                throw new Exception("The stack is empty");
            else
            {
                obj = arr[top2];
                arr[top2] = null;
                top2 -= 2;
            }
        }

        return obj;
    }

    public object Peek(int type)
    {
        if (type == 1)
        {
            if (top1 == -1)
                throw new Exception("The stack is empty");

            return arr[top1];
        }

        else //type == 2
        {
            if (top2 == -2)
                throw new Exception("The stack is empty");

            return arr[top2];
        }
    }
}

  

  方法2:

第一个栈A:从最左向右增长
第二个栈B:从最右向左增长 

  代码实现如下:

public class NewStack
{
    object[] arr;
    int top1;
    int top2;

    public NewStack(int capticy)
    {
        arr = new object[capticy];

        top1 = 0;
        top2 = capticy;
    }

    public void Push(int type, object element)
    {
        if (top1 == top2)
            throw new Exception("The stack is full");

        if (type == 1)
        {
            arr[top1] = element;
            top1++;
        }
        else //type==2
        {
            top2--;
            arr[top2] = element;
        }            
    }

    public object Pop(int type)
    {
        object obj = null;

        if (type == 1)
        {
            if (top1 == 0)
                throw new Exception("The stack is empty");
            else
            { 
                top1--;
                obj = arr[top1];
                arr[top1] = null;
            }
        }

        else //type == 2
        {
            if (top2 == arr.Length)
                throw new Exception("The stack is empty");
            else
            {
                obj = arr[top2];
                arr[top2] = null;
                top2++;
            }
        }

        return obj;
    }

    public object Peek(int type)
    {
        if (type == 1)
        {
            if (top1 == 0)
                throw new Exception("The stack is empty");

            return arr[top1 - 1];
        }

        else //type == 2
        {
            if (top2 == arr.Length)
                throw new Exception("The stack is empty");

            return arr[top2];
        }
    }
}

  综合比较上述两种算法,我们发现,算法1实现的两个栈,每个都只有n/2个空间大小;而算法2实现的两个栈,如果其中一个很小,另一个则可以很大,它们的和为常数n。

 

9..如何用一个数组实现三个栈

  最后,让我们把抠门儿进行到底,相信看完本文,你已经从物质和精神上都升级为一个抠门儿主义者。

  如果还使用交叉索引的办法,每个栈都只有N/3个空间。

  让我们只好使用上个题目的第2个方法,不过这只能容纳2个栈,我们还需要一个位置存放第3个栈,不如考虑数组中间的位置——第3个栈的增长规律可以如下:

第1个入栈C的元素进mid处
第2个入栈C的元素进mid+1处
第3个入栈C的元素进mid-1处
第4个入栈C的元素进mid+2处

  这个方法的好处是, 每个栈都有接近N/3个空间。

public class NewStack
{
    object[] arr;
    int top1;
    int top2;

    int top3_left;
    int top3_right;
    bool isLeft;

    public NewStack(int capticy)
    {
        arr = new object[capticy];

        top1 = 0;
        top2 = capticy;

        isLeft = true;
        top3_left = capticy / 2;
        top3_right = top3_left + 1;
    }

    public void Push(int type, object element)
    {
        if (type == 1)
        {
            if (top1 == top3_left + 1)
                throw new Exception("The stack is full");

            arr[top1] = element;
            top1++;
        }
        else if (type == 2)
        {
            if (top2 == top3_right)
                throw new Exception("The stack is full");

            top2--;
            arr[top2] = element;
        }
        else //type==3
        {
            if (isLeft)
            {
                if (top1 == top3_left + 1)
                    throw new Exception("The stack is full");

                arr[top3_left] = element;
                top3_left--;
            }
            else
            {
                if (top2 == top3_right)
                    throw new Exception("The stack is full");

                arr[top3_right] = element;
                top3_right++;
            }

            isLeft = !isLeft;
        }
    }

    public object Pop(int type)
    {
        object obj = null;

        if (type == 1)
        {
            if (top1 == 0)
                throw new Exception("The stack is empty");
            else
            {
                top1--;
                obj = arr[top1];
                arr[top1] = null;
            }
        }
        else if (type == 2)
        {
            if (top2 == arr.Length)
                throw new Exception("The stack is empty");
            else
            {
                obj = arr[top2];
                arr[top2] = null;
                top2++;
            }
        }
        else //type==3
        {
            if (top3_right == top3_left + 1)
                throw new Exception("The stack is empty");

            if (isLeft)
            {
                top3_left++;
                obj = arr[top3_left];
                arr[top3_left] = null;
            }
            else
            {
                top3_right--;
                obj = arr[top3_right];
                arr[top3_right] = null;
            }

            isLeft = !isLeft;
        }

        return obj;
    }

    public object Peek(int type)
    {
        if (type == 1)
        {
            if (top1 == 0)
                throw new Exception("The stack is empty");

            return arr[top1 - 1];
        }

        else if (type == 2)
        {
            if (top2 == arr.Length)
                throw new Exception("The stack is empty");

            return arr[top2];
        }
        else //type==3
        {
            if (top3_right == top3_left + 1)
                throw new Exception("The stack is empty");

            if (isLeft)
                return arr[top3_left + 1];
            else
                return arr[top3_right - 1];
        }
    }
}

 

posted @ 2009-12-21 00:35  包建强  Views(11953)  Comments(34Edit  收藏  举报