BZOJ 3685 普通van Emde Boas树 权值线段树（zkw）

    第一眼看到这题，没错就拿他来做treap的练手了，然而我错了，卡treap，我哭了，写了两三次treap（），这两天几乎都在写数据结构了。后来我又可耻地看了题解，原来这道题已经给了数列中数的范围，可以写成权值线段树，当然线段树要写成zkw才更快。 最重要的是，在我看过的一个人的题解中， 他是用一个bool 数组来判断一个数是否出现过，比我们这些第一反应写treap，还要写个find，去查找这个数是否存在的高级多了，节省了不少时间，关键是还学到了，输出优化。加油吧！ 对于权值线段树的理解和应用也要加强。

这份代码3000多毫秒。

#include<cstdio>
#include<iostream>
#define rep(i,j,k) for(int i = j; i <= k; i++)
using namespace std;
 
int tree[1<<21|1];
int po = 1, cnt = 0;
bool in[1<<20];
 
inline int read()
{
    int s =0, t =1; char c = getchar();
    while( !isdigit(c) ){
        if( c == '-' ) t = -1; c = getchar();
    }
    while( isdigit(c) ){
        s = s * 10 + c - '0'; c = getchar();
    }
    return s * t;
}
 
void update(int t,int d)
{
    for(tree[t+=po]=d, t>>=1; t; t>>=1 ){
        tree[t] = tree[t<<1] | tree[t<<1|1];
    }
}
 
int get_min(int t)
{
    for(;t <= po; )
    {
        if( tree[t<<1] ) t = (t<<1);
        else t = (t<<1)+1;
    }
    return t-po;
}
 
int get_max(int t)
{
    for(;t <= po; ){
        if( tree[t<<1|1] ) t = (t<<1)+1;
        else t = t << 1;
    }
    return t-po;
}
 
void put(int x)
{
    if( x / 10 ) put(x/10);
    putchar(x%10+'0');
}
 
int main()
{
    int n = read(), m = read(); while( 1<<po < n ) po++; po = (1<<po) -1;
    while( m-- ){
        int x = read();
        if( x != 3 && x != 4 ){
            int y = read()+1;
            if( x == 1 ){
                if( !in[y] ){
                    update(y,in[y]=1); cnt++;
                }
            }
            if(x == 2 ){
                if( in[y] ){
                    update(y,in[y]=0); cnt--;
                }
            }
            if( x == 5 ){
                int t;
                for(t = y+po; t != 1; t>>=1){
                    if( (t & 1) && tree[t^1] ) {
                        t = t ^ 1; break;
                    }
                }
                if( t == 1 ) puts("-1");
                else {
                    put(get_max(t)-1); puts("");
                }
            }
            if( x == 6 ){
                int t;
                for(t = y+po; t != 1; t>>=1){
                    if( !(t & 1) && tree[t^1] ) {
                        t = t ^ 1; break;
                    }
                }
                if( t == 1 ) puts("-1");
                else {
                    put(get_min(t)-1); puts("");
                }
            }
            if( x == 7 ){
                if( in[y] ) {
                    put(1); puts("");
                }
                else puts("-1"); 
            }
        }
        else {
            if( !cnt ) {
                puts("-1"); continue;
            }
            if( x == 3 ){
                put(get_min(1)-1); puts("");
            }
            else if( x == 4 ) {
                put(get_max(1)-1); puts("");
            }
        }
    }
    return 0;
}

另附这道题，我刚才突发奇想用vector写了分块，一开始由于鲁棒性不强RE了，后来改了后AC了，用时6800多毫秒，还是很快的。（其实昨天就写了vector，但由于没分块，TLE了）

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#define mod 1005
#define rep(i,j,k) for(int i = j; i <= k; i++)
#define down(i,j,k) for(int i = j; i >= k; i--)
using namespace std;

int read()
{
    int s = 0, t = 1; char c = getchar();
    while( !isdigit(c) ){
        if( c == '-' )t = -1; c = getchar();
    }
    while( isdigit(c) ){
        s = s * 10 + c - '0'; c = getchar();
    }
    return s * t;
} 

bool in[1000006] = {0}; int tot = 0;

struct node{
vector<int> a;
int cnt;
node(){
  a.insert(a.begin(),0x7fffffff); cnt = 0;
}
} nod[1005];
vector<int> s; int cnt = 0;

void insert(int y)
{
    tot++; 
    int key = y / mod;
    if( !nod[key].cnt ) {
        s.insert(lower_bound(s.begin(),s.end(),key),key);
        cnt++;
    }
    nod[key].cnt++;
    nod[key].a.insert(lower_bound(nod[key].a.begin(),nod[key].a.end(),y),y);
}

void remove(int y)
{
    tot--; 
    int key = y / mod;
    nod[key].cnt--;
    if( !nod[key].cnt ) {
        s.erase(lower_bound(s.begin(),s.end(),key));
        cnt--;
    }
    nod[key].a.erase(lower_bound(nod[key].a.begin(),nod[key].a.end(),y));
}

void put(int x)
{
    if(x / 10 ) put(x/10);
    putchar(x%10+'0');
}

int minl()
{
    if( !tot ) return -1;
    else return nod[s[0]].a[0];
}

int maxl()
{
    if( !tot ) return -1;
    else return nod[s[cnt-1]].a[nod[s[cnt-1]].cnt-1];
}

int pre(int y)
{
    if( !tot || minl() >= y ) return -1;
    int key = y / mod;
    int k = lower_bound(s.begin(),s.end(),key) - s.begin();
    down(i,k,0){
        int x = s[i];
        int zhi = lower_bound(nod[x].a.begin(),nod[x].a.end(),y) - nod[x].a.begin();
        if( zhi == 0 ) continue;  return nod[x].a[zhi-1];
    }
}

int suc(int y)
{
    if( !tot || maxl() <= y ) return -1;
    int key = y / mod;
    int k = lower_bound(s.begin(),s.end(),key) -s.begin();
    rep(i,k,cnt-1){
        int x = s[i];
        int zhi = upper_bound(nod[x].a.begin(),nod[x].a.end(),y) - nod[x].a.begin();
        if( nod[x].a[zhi] != 0x7fffffff ) return nod[x].a[zhi];
    }
}

int main()
{
    int n = read(), m = read(); s.insert(s.begin(),0x7fffffff); 
    while( m-- ){
        int x = read(), y, key;
        switch( x ){
            case 1: y = read()+1; if( !in[y] ) insert(y), in[y] = 1; break;
            case 2: y = read()+1; if( in[y] ) remove(y), in[y] = 0;  break;
            case 3: key = minl(); if( key == -1 ) puts("-1"); else put(key-1), puts(""); break;
            case 4: key = maxl(); if( key == -1 ) puts("-1"); else put(key-1), puts(""); break;
            case 5: y = read()+1;  key = pre(y); if( key == -1 ) puts("-1"); else put(key-1), puts(""); break;
            case 6: y = read()+1;  key = suc(y); if( key == -1 ) puts("-1"); else put(key-1), puts(""); break;
            case 7: y = read()+1; if( in[y] ) put(1), puts(""); else puts("-1"); break;
        }
    }
    return 0;
}

加油，相信自己，能行的。