平衡树做法,因为最近正好自己看了点平衡树相关的知识,便觉得这个题就是专门为了平衡树准备的。(完全没想到set,其实也是不会用)
因为查排名,如果我们有一棵二叉搜索树那就会很好办,每个点都有相应的判断位置的值,从根节点入手,如果当前节点比过题数较多就往左儿子查,过题少就往右儿子查,相等就是判断罚分然后向左或者向右儿子搜索。一旦发现与当前节点过题数和罚分都相等的节点就返回左方所有节点囊括的点数,在那基础上加一就是需要的解。
为了维持二叉搜索树查询的效率,我们需要平衡树,我暂时只会treap(就是堆+树),通过堆的性质维护使BST不至于失衡。代码里应该都是treap基本的操作:插入,删除,树枝的旋转(这个学了好久)。然后自写个递归查询即可。
至于treap的维护,每一次读入数据后删去该点本身在书中的数据,更新后再插入即可。查询打印
具体见AC代码注释:(因为是初学平衡树,码的太丑别嫌弃www)
#include<bits/stdc++.h>
#define maxn 10000005
#define maxm 30000005
#define FOR(a, b, c) for(int a=b; a<=c; a++)
#define qhqh 1000000007
#define inf 2147483647
#define llinf 9223372036854775807
#define ll long long
#define pi acos(-1.0)
#define ls ch[p][0]
#define rs ch[p][1]
using namespace std;
//快读
inline int read()
{
char c=getchar();long long x=0,f=1;
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
return x*f;
}
int n, m, x, penalty;
int ch[maxn][2]; //左右儿子
int dat[maxn];
int siz[maxn], cnt[maxn]; //siz[]表示以当前节点为根的子树大小,cnt[ ]表示当前节点包含的副本数
int root, tot;
struct node
{
int x, p;
}val[maxn], rua[maxn]; //val[ ]