P1801

黑匣子

题目描述

Black Box 是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量 \(i\)。最开始的时候 Black Box 是空的.而 \(i=0\)。这个 Black Box 要处理一串命令。

命令只有两种:

  • ADD(x):把 \(x\) 元素放进 Black Box;

  • GET\(i\)\(1\),然后输出 Black Box 中第 \(i\) 小的数。

记住:第 \(i\) 小的数,就是 Black Box 里的数的按从小到大的顺序排序后的第 \(i\) 个元素。

我们来演示一下一个有11个命令的命令串。(如下表所示)

序号 操作 \(i\) 数据库 输出
1 ADD(3) \(0\) \(3\) /
2 GET \(1\) \(3\) \(3\)
3 ADD(1) \(1\) \(1,3\) /
4 GET \(2\) \(1,3\) \(3\)
5 ADD(-4) \(2\) \(-4,1,3\) /
6 ADD(2) \(2\) \(-4,1,2,3\) /
7 ADD(8) \(2\) \(-4,1,2,3,8\) /
8 ADD(-1000) \(2\) \(-1000,-4,1,2,3,8\) /
9 GET \(3\) \(-1000,-4,1,2,3,8\) \(1\)
10 GET \(4\) \(-1000,-4,1,2,3,8\) \(2\)
11 ADD(2) \(4\) \(-1000,-4,1,2,2,3,8\) /

现在要求找出对于给定的命令串的最好的处理方法。ADD 命令共有 \(m\) 个,GET 命令共有 \(n\) 个。现在用两个整数数组来表示命令串:

  1. \(a_1,a_2,\cdots,a_m\):一串将要被放进 Black Box 的元素。例如上面的例子中 \(a=[3,1,-4,2,8,-1000,2]\)

  2. \(u_1,u_2,\cdots,u_n\):表示第 \(u_i\) 个元素被放进了 Black Box 里后就出现一个 GET 命令。例如上面的例子中 \(u=[1,2,6,6]\) 。输入数据不用判错。

输入格式

第一行两个整数 \(m\)\(n\),表示元素的个数和 GET 命令的个数。

第二行共 \(m\) 个整数,从左至右第 \(i\) 个整数为 \(a_i\),用空格隔开。

第三行共 \(n\) 个整数,从左至右第 \(i\) 个整数为 \(u_i\),用空格隔开。

输出格式

输出 Black Box 根据命令串所得出的输出串,一个数字一行。

样例 #1

样例输入 #1

7 4
3 1 -4 2 8 -1000 2
1 2 6 6

样例输出 #1

3
3
1
2

提示

数据规模与约定

  • 对于 \(30\%\) 的数据,\(1 \leq n,m \leq 10^{4}\)
  • 对于 \(50\%\) 的数据,\(1 \leq n,m \leq 10^{5}\)
  • 对于 \(100\%\) 的数据,\(1 \leq n,m \leq 2 \times 10^{5},|a_i| \leq 2 \times 10^{9}\),保证 \(u\) 序列单调不降。

可能有重复元素 所以用pair<int,int> make_pair()

#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_cxx;
using namespace __gnu_pbds;
#define int long long
const int N=2e5+5;
#define KOITO_YUU 小糸侑
typedef pair<int,int>pii;
#define mp make_pair
typedef tree<pii,null_type,less<pii>,rb_tree_tag,tree_order_statistics_node_update>Tree;
Tree tr;
int m,n,a[N],b[N],times=0;
#ifdef KOITO_YUU
signed main() {
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1; i<=n; i++)cin>>a[i];
	for(int i=1; i<=m; i++)cin>>b[i];
	int pt=1,rk=0;
	for(int i=1; i<=n; i++) {
		times++;
		tr.insert(mp(a[i],times));
		while(i==b[pt]) {
			pt++;
			rk++;
			auto it=tr.find_by_order(rk-1);
			cout<<(*it).first<<"\n";
		}
	}
	return 1+1==3;
}
#endif
posted @ 2023-05-06 09:41  N0zoM1z0  阅读(4)  评论(0编辑  收藏  举报