P1165 日志分析

日志分析

题目描述

\(M\) 海运公司最近要对旗下仓库的货物进出情况进行统计。目前他们所拥有的唯一记录就是一个记录集装箱进出情况的日志。该日志记录了两类操作：第一类操作为集装箱入库操作，以及该次入库的集装箱重量；第二类操作为集装箱的出库操作。这些记录都严格按时间顺序排列。集装箱入库和出库的规则为先进后出，即每次出库操作出库的集装箱为当前在仓库里所有集装箱中最晚入库的集装箱。

出于分析目的，分析人员在日志中随机插入了若干第三类操作――查询操作。分析日志时，每遇到一次查询操作，都要报告出当前仓库中最大集装箱的重量。

输入格式

包含\(N+1\) 行：

第一行为\(1\) 个正整数\(N\)，对应于日志内所含操作的总数。

接下来的$N $行，分别属于以下三种格式之一：

格式\(1\): \(0 X\) //一次集装箱入库操作，正整数\(X\)表示该次入库的集装箱的重量

格式\(2\): \(1\) //一次集装箱出库操作，（就当时而言）最后入库的集装箱出库

格式\(3\): \(2\) //一次查询操作，要求分析程序输出当前仓库内最大集装箱的重量

当仓库为空时你应该忽略出库操作，当仓库为空查询时你应该输出\(0\)。

输出格式

输出行数等于日志中查询操作的次数。每行为一个正整数，表示查询结果。

样例 #1

样例输入 #1

13
0 1
0 2
2
0 4
0 2
2
1
2
1
1
2
1
2

样例输出 #1

2
4
4
1
0

提示

对于\(20\%\)的数据，有\(N≤10\)；

对于\(40\%\)的数据，有\(N≤1000\)；

对于\(100\%\)的数据，有\(N≤200000,X≤10^8\)。

分析

• 如果直接循环查询最大值，\(O(n^2)\), TLE
• 考虑存储答案，查询的时候直接 \(O(1)\) 输出
• \(ans[head]\) 表示当前栈内元素的最大值，如果有元素入队，则更新一下
• 由于答案的更新总是比栈的操作慢一步，所以之前的最大值不会影响之后的答案。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=1e6+10, INF=0x3f3f3f3f;
int sta[N],ans[N],head=0,n,op,x;

int main(){
    scanf("%d",&n);
    while(n--){
        scanf("%d",&op);
        if(op==0){
            scanf("%d",&x);
            sta[++head] = x; // 入栈
            ans[head] = max(ans[head-1], x); // 更新答案
        }else if(op==1){
            if(head) head--;
        }else if(op==2){
            printf("%d\n",ans[head]);
        }
    }
    return 0;
}