牛客小白月赛8 F-数列操作 (有序表)

链接：https://www.nowcoder.com/acm/contest/214/F
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld

题目描述

clccle是个蒟蒻,她经常会在学校机房里刷题,也会被同校的dalao们虐,有一次,她想出了一个毒瘤数据结构,便兴冲冲的把题面打了出来,她觉得自己能5s内切掉就很棒了,结果evildoer过来一看,说:"这思博题不是1s就能切掉嘛",clccle觉得自己的信心得到了打击,你能帮她在1s中切掉这道水题嘛?

你需要写一个毒瘤(划掉)简单的数据结构,满足以下操作
1.插入一个数x(insert)
2.删除一个数x(delete)(如果有多个相同的数,则只删除一个)
3.查询一个数x的排名(若有多个相同的数，就输出最小的排名)
4.查询排名为x的数
5.求一个数x的前驱
6.求一个数x的后继

输入描述:

第一行,输入一个整数n,表示接下来需要输入n行接下来n行,输入 一个整数num和一个整数x

输出描述:

当num为3,4,5,6时,输出对应的答案

示例1

输入

复制

输出

复制

说明

大家自己手玩样例算了QWQ

备注:

对于全部数据n<=1e5,且3,4,5,6的操作数少于60000输入数据可能很多,推荐使用快读

好像是个平衡树的板子题

但是平衡树还没写过啊....

刚开始想用multiset实现

但是mutilset的树结构关系迭代器只重载了自加自减

无法随机查询

洗衣服的时候突然想到LIS的有序表插入

这道题用有序表加vector的各种操作应该可以实现

结果交了一发竟然过了

时间复杂度:

随机查询O(1)

按值查询O(log2n) //lower/upper_bound()

后插O(1) //arr.push_back()

中间插入平均O(n/2) //arr.insert(vector<int>::iterator it, val);

时间复杂度尚可接受

代码↓

//#pragma GCC optimize(2)
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <string>
#include <cstring>
#include <algorithm>
#include <stack>
#include <queue>
#include <set>
#include <sstream>
#include <map>
#include <ctime>
#include <vector>
#include <fstream>
#include <list>
#include <iomanip>
#include <numeric>
using namespace std;
typedef long long ll;const int MAXN = 1e6 + 10;std::vector <int> arr;int main()
{//ios::sync_with_stdio(false);//cin.tie(0);     cout.tie(0);//freopen("D://test.in", "r", stdin);//freopen("D://test.out", "w", stdout);int n, a, b;scanf("%d", &n);vector <int>::iterator it;while(n--){   scanf("%d%d", &a, &b);if(a == 1){it = lower_bound(arr.begin(), arr.end(), b);arr.insert(it, b);}else if(a == 2){it = lower_bound(arr.begin(), arr.end(), b);arr.erase(it);}else if(a == 3){  it = lower_bound(arr.begin(), arr.end(), b);printf("%d\n", it - arr.begin() + 1);  }else if(a == 4){printf("%d\n", arr[b - 1]);}else if(a == 5){it = lower_bound(arr.begin(), arr.end(), b);it--;printf("%d\n", *it);}else if(a == 6){it = upper_bound(arr.begin(), arr.end(), b);printf("%d\n", *it);}}return 0;
}