码迷,mamicode.com
首页 > 其他好文 > 详细

设计一个集合栈

时间:2016-12-11 08:02:04      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:没有   desc   重要   组成   public   实现   return   code   个数   

题目描述

请实现一种数据结构SetOfStacks,由多个栈组成,其中每个栈的大小为size,当前一个栈填满时,新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。

给定一个操作序列int[][2] ope(C++为vector<vector<int>>),每个操作的第一个数代表操作类型,若为1,则为push操作,后一个数为应push的数字;若为2,则为pop操作,后一个数无意义。请返回一个int[][](C++为vector<vector<int>>),为完成所有操作后的SetOfStacks,顺序应为从下到上,默认初始的SetOfStacks为空。保证数据合法。

 

思路:用arraylist来实现栈,放进外层的arraylist里面。这个要用一个很重要的特点,就是在ans.add(temp),之后,temp.add(1),这样ans里面的temp还是会变化,因为ans里面传入的是引用不是值,所以在当前栈没有空或者没有满之前,上下文中的栈都可以使用这个引用,当发现这个stack空了,或者满了,这个时候就应该再new一个stack加到ans里面,或者去掉一个stack,还是用相同的名字来指向新的stack,进行下一补操作。

import java.util.*;

public class SetOfStacks {
    public ArrayList<ArrayList<Integer>> setOfStacks(int[][] ope, int size) {
        // write code here
        //要用arraylist来实现stack
        ArrayList<ArrayList<Integer>> ans=new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> temp=new ArrayList<Integer>();
        ans.add(temp);
        for(int i=0;i<ope.length;i++){
            if(ope[i][0]==1){//push
                if(temp.size()<size){
                    temp.add(ope[i][1]);
                }else{
                    temp=new ArrayList<Integer>();
                    temp.add(ope[i][1]);
                    ans.add(temp);
                }
            }else{
                if(temp.size()>0){
                    temp.remove(temp.size()-1);
                }else{
                    ans.remove(ans.size()-1);
                    temp=ans.get(ans.size()-1);
                    temp.remove(temp.size()-1);
                }
                
            }
        }
        return ans;
    }
}

 

设计一个集合栈

标签:没有   desc   重要   组成   public   实现   return   code   个数   

原文地址:http://www.cnblogs.com/tobemaster/p/6158866.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!