Codeforces Gym 100342F Move to Front 平衡树

题目大意：

有一个链表其中的元素为1->2->3->4->…….->INF，每次询问一个元素在链表中的位置，然后把该元素放到链表的首位置。

做法：

beta version

• 如果输入数字不大，那么我们可以直接先把1到最大数字打上时间戳加到平衡树（按时间戳从新到老排序）里面去；
• 每次询问，就询问时间戳在平衡树中的rank即可；
• 在平衡树中删除该时间戳，将该数重新打上一个新的时间戳加入到平衡树中即可（相当于移到链表首）。

AC version

• 但是输入到了$10^9$级别，我们没有办法将所有数都加到平衡树中
• 所以我们先将出现的所有数字离散化；
• 询问时先检查当前数(记为Ai)是否被打上过时间戳；
• 如果被打上了时间戳，那么做法和上面的beta version一样；
• 如果没有被打上时间戳，那么它的rank = Ai + 已经打了时间戳的数字个数 - 打了时间戳的数字里比Ai小的数的个数；
• 然后按照与beta version一样的做法给该数打上新的时间戳。
• 注：不一定使用平衡树，使用能够实现查询rank的数据结构即可。

代码：

4545769	WHU_FFT	F	Accepted	8824 KB	310 ms	GNU G++ 4.9.2

#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;
const int maxn=200000;
int num[maxn],cop[maxn],t[maxn];
tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update>bst,bb;
//exSTL平衡树,bst为时间戳平衡树,bb普通平衡树 
int main()
{
    freopen("mtf.in","r",stdin);
#ifdef ONLINE_JUDGE
    freopen("mtf.out","w",stdout);
#endif
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&num[i]);
    memcpy(cop,num,sizeof num);
    sort(cop+1,cop+n+1);
    int len=unique(cop+1,cop+n+1)-&cop[1];
    //离散化 
    for(int i=1;i<=n;i++)
    {
        int s=lower_bound(cop+1,cop+1+len,num[i])-&cop[0];
        if(bst.find(t[ s ]) != bst.end())
        {
            //已经打了时间戳的情况 
            printf("%d",(int)bst.order_of_key(t[ s ])+1);
            bst.erase(t[ s ]);
        }
        else printf("%d",num[i] + (int)bst.size() - (int)bb.order_of_key(num[i]) );
        //没有打时间戳的情况 
        t[ s ]=n-i+1;
        bst.insert( t[ s ]);
        if(i!=n)putchar(' ');
        else putchar('\n');
        bb.insert( num[i] );
    }
    return 0;
}