洛谷 B3666 求数列所有后缀最大值的位置(Java)
传送门:B3666 求数列所有后缀最大值的位置
题目: 求数列所有后缀最大值的位置
题目描述
给定一个数列 a a a,初始为空。有 n n n 次操作,每次在 a a a 的末尾添加一个正整数 x x x。
每次操作结束后,请你找到当前 a a a 所有的后缀最大值的下标(下标从 1 开始)。一个下标 i i i 是当前 a a a 的后缀最大值下标当且仅当:对于所有的 i < j ≤ ∣ a ∣ i < j \leq |a| i<j≤∣a∣,都有 a i > a j a_i > a_j ai>aj,其中 ∣ a ∣ |a| ∣a∣ 表示当前 a a a 的元素个数。
为了避免输出过大,请你每次操作结束后都输出一个整数,表示当前数列所有后缀最大值的下标的按位异或和。
输入格式
第一行是一个整数,表示操作次数
n
n
n。
第二行有
n
n
n 个整数,依次表示
n
n
n 次操作所添加的整数
x
i
x_i
xi。
输出格式
每次操作后请输出一行一个整数,表示当前数列所有后缀最大值下标的按位异或和。
样例 #1
样例输入 #1
5
2 1 3 5 4
样例输出 #1
1
3
3
4
1
提示
数据规模与约定
对于全部的测试点,保证 1 ≤ n ≤ 1 0 6 1 \leq n \leq 10^6 1≤n≤106, 1 ≤ x i < 2 64 1 \leq x_i \lt 2^{64} 1≤xi<264。
分析:
求异或和
- 异或运算最重要的性质就是自反性。
(a⊕b)⊕b=a。
单调栈
首先,我们需要了解一下单调栈的原理:
每次给一个数,先判断这个数和队顶的大小,若栈非空且栈顶小于(这里可以随机应变)这个数,就弹出这个数,直到找到一个比这个数大的数或者栈空为止,然后把这个数入栈即可。
可以发现,这个单调栈就是专门针对“后缀最值”问题的数据结构。
最后就是由于 对于全部的测试点,保证
1
≤
n
≤
1
0
6
1 \leq n \leq 10^6
1≤n≤106,
1
≤
x
i
<
2
64
1 \leq x_i \lt 2^{64}
1≤xi<264。
,注意开 long。
代码:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main9 {
public static void main(String[] args) throws IOException {
BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
String [] s = br.readLine().split(" ");
br.close();
long [] a = new long [n+10];
int [] b = new int [n+10];
int tt = 0;long ans = 0;
for(int i = 1;i <= n;i++) {
long x = a[i]= Long.parseLong(s[i-1]);
while(tt>0&&x>=a[b[tt]]) {
ans ^= a[b[tt]];tt--;
}
tt++;b[tt] = i;
ans ^= i;
System.out.println(ans);
}
}
}