爱学习 昂塞 挂

1.     static class Message{
   
2.         int id;
   
3.         String type;
   
4.         public Message(int id, String type){
   
5.             this.id = id;
   
6.             this.type = type;
   
7.         }
   
8.         public Message(String type){
   
9.             this.type = type;
   
10.         }
    
11.     }
    
12. 
    
13.     public static List<Message> process(List<Message> messageList){
    
14.         Stack<Message> stackA = new Stack<>();
    
15.         Stack<Message> stackB = new Stack<>();
    
16.         Stack<Message> stackC = new Stack<>();
    
17.         for(Message message : messageList){
    
18.             if(message.type.equals("A")){
    
19.                 stackA.push(message);
    
20.             }
    
21.             else if(message.type.equals("B")){
    
22.                 stackB.push(message);
    
23.             }
    
24.             else if(message.type.equals("C")){
    
25.                 stackC.push(message);
    
26.                 while(!stackA.isEmpty() && !stackC.isEmpty() && stackA.peek().id == stackC.peek().id){
    
27.                     stackB.pop().id = stackA.peek().id;
    
28.                     stackA.pop();
    
29.                     stackC.pop();
    
30.                 }
    
31.             }
    
32.         }
    
33.         return messageList;
    
34.     }

复制代码

是这样吗？测了几个例子没啥问题。。不过好像有点难证明。。