BZOJ4999: This Problem Is Too Simple！

开学了来刷刷水。

开个map每个值随便开个数据结构。卡了$O(n\log^2n)$的map版树状数组差评。

卡了半天才rk1，应该没几天就没了。已经没了。

/**************************************************************
    Problem: 4999
    User: T404
    Language: C++
    Result: Accepted
    Time:2900 ms
    Memory:164208 kb
****************************************************************/
 
#include<cstdio>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
#include<sys/mman.h>
struct ano{
    char b[1<<21],*s,*t;
    ano():t(b){
        s=(char*)mmap(0,1<<25,1,2,0,0);
    }
    ~ano(){fwrite(b,1,t-b,stdout);}
    char get(){
        while(*s<33)++s;
        return*s++;
    }
    void out(const char*v){
        while(*v)*t++=*v++;
        *t++=10;
    }
    operator int(){
        int x=0,y=0;
        while(*s<48)
            *s++==45?y=1:0;
        while(*s>32)
            x=x*10+*s++-48;
        return y?-x:x;
    }
    void out(int x){
        static char c[12];
        char*i=c;
        if(!x)*t++=48;
        else{
            if(x<0)*t++=45,x=-x;
            while(x){
                int y=x/10;
                *i++=x-y*10+48,x=y;
            }
            while(i!=c)*t++=*--i;
        }
        *t++=10;
    }
}buf;
using namespace std;
const int N=1e5+5;
struct edge{
    int v;
    edge*s;
}e[N*2];
edge*o=e,*h[N];
int dfn,f1[N],f2[N],siz[N],p[N],d[N],c1[N],c2[N],w[N];
void dfs1(int u){
    f1[u]=++dfn;
    siz[u]=1;
    for(edge*i=h[u];i;i=i->s)
        if(i->v!=p[u]){
            d[i->v]=d[p[i->v]=u]+1;
            dfs1(i->v);
            siz[u]+=siz[i->v];
            if(siz[c1[u]]<siz[i->v])
                c1[u]=i->v;
        }
    f2[u]=dfn;
}
int lca(int u,int v){
    while(c2[u]!=c2[v])
        d[c2[u]]>d[c2[v]]?u=p[c2[u]]:v=p[c2[v]];
    return d[u]<d[v]?u:v;
}
struct node{
    node*i,*j;
    int s;
}a[N*102];
node*l=a;
__gnu_pbds::cc_hash_table<int,node*>t;
void inc(int d,int u,node**o){
    for(int i=16;~i;--i){
        if(!*o)*o=l++;
        if(u>>i&1)o=&(*o)->j;
        else
            (*o)->s+=d,o=&(*o)->i;
    }
}
int ask(int u,node*o){
    int s=0;
    for(int i=16;~i;--i){
        if(!o)break;
        if(~u>>i&1)o=o->i;
        else
            s+=o->s,o=o->j;
    }
    return s;
}
void inc(int u,int d){
    node**o=&t[w[u]];
    inc(d,f1[u]-1,o);
    inc(-d,f2[u],o);
}
int ask(int u,int v,int k){
    node*o=t[k];
    int a=lca(u,v);
    return ask(f1[u],o)+ask(f1[v],o)-ask(f1[a],o)*2+(w[a]==k);
}
int main(){
    int n,m,u,v;
    n=buf,m=buf;
    for(int i=1;i<=n;++i)
        w[i]=buf;
    for(int i=2;i<=n;++i){
        u=buf,v=buf;
        *o={v,h[u]};
        h[u]=o++;
        *o={u,h[v]};
        h[v]=o++;
    }
    dfs1(1);
    for(int i=1;i<=n;++i)
        if(c1[p[i]]!=i)
            for(int j=i;j;j=c1[j])c2[j]=i;
    for(int i=1;i<=n;++i)
        inc(i,1);
    while(m--){
        char o=buf.get();
        u=buf,v=buf;
        if(o=='Q')
            buf.out(ask(u,v,buf));
        else{
            inc(u,-1);
            w[u]=v;
            inc(u,1);
        }
    }
}

要是人生能有追求就好了。

恋をしようか。