SPOJ 10628. SPOJ COT Count on a tree 可持久化线段树

　　这题是裸的主席树，每个节点建一棵主席树，再加个lca就可以了。

　　历尽艰辛，终于A掉了这一题，这般艰辛也显示出了打代码的不熟练。

　　错误：1、lca倍增的时候i和j写反了，RE了5次，实在要吸取教训

　　　　　2、主席树插入操作的时候，如果插入到的那个点（叶节点）原来有值，而没有加上，导致了WA

　　以下是历尽艰辛的代码，还很长。

　

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <map>

using namespace std;

const int maxn = ;
int n, m, w[maxn];
int head[maxn], label;
struct Edge
{
    int v, next;
    Edge (int v = , int next = ):
        v(v), next(next) {}
}e[maxn*];
int depth[maxn], f[maxn][], root[maxn], temp[maxn], s_num;
queue <int> q;
map <int, int> id;
struct Tree
{
    int sum[maxn*], ls[maxn*], rs[maxn*], cnt;
    Tree ()
    {
        sum[] = , cnt = ;
    }
    void PushUp(int rt)
    {
        sum[rt] = sum[ls[rt]]+sum[rs[rt]];
    }
    void insert(int las_rt, int rt, int l, int r, int p, int d)
    {
        if (l == r)
        {
            sum[rt] = sum[las_rt]+d;
            return ;
        }
        int mid = (l+r)>>;
        if (p <= mid)
        {
            ls[rt] = ++cnt, rs[rt] = rs[las_rt];
            insert(ls[las_rt], ls[rt], l, mid, p, d);
        }
        else
        {
            ls[rt] = ls[las_rt], rs[rt] = ++cnt;
            insert(rs[las_rt], rs[rt], mid+, r, p, d);
        }
        PushUp(rt);
    }
    int query(int u_rt, int v_rt, int lca_rt, int lca_pos, int l, int r, int k)
    {
        if (l == r)
            return l;
        int mid = (l+r)>>;
        int s_l = sum[ls[u_rt]]+sum[ls[v_rt]]-*sum[ls[lca_rt]]+(lca_pos >= l && lca_pos <= mid);
        if (k <= s_l)
            return query(ls[u_rt], ls[v_rt], ls[lca_rt], lca_pos, l, mid, k);
        else
            return query(rs[u_rt], rs[v_rt], rs[lca_rt], lca_pos, mid+, r, k-s_l);
    }
}T;

void ins(int u, int v)
{
    e[++label] = Edge(v, head[u]);
    head[u] = label;
}

void in()
{
    scanf("%d %d", &n, &m);
    for (int i = ; i <= n; ++i)
        scanf("%d", &w[i]);
    for (int i = ; i <= n; ++i)
        head[i] = -;
    label = -;
    for (int i = ; i < n; ++i)
    {
        int u, v;
        scanf("%d %d", &u, &v);
        ins(u, v), ins(v, u);
    }
}

void Build_lca()
{
    for (int i = ; i <= ; ++i)
        for (int j = ; j <= n; ++j)
            f[j][i] = -;
    q.push();
    depth[] = ;
    f[][] = -;
    while (!q.empty())
    {
        int u = q.front();
        for (int i = head[u]; i != -; i = e[i].next)
        {
            int v = e[i].v;
            if (v == f[u][])
                continue ;
            f[v][] = u;
            depth[v] = depth[u]+;
            q.push(v);
        }
        q.pop();
    }
    for (int i = ; i <= ; ++i)
        for (int j = ; j <= n; ++j)
        {
            if (f[j][i-] == -)
                continue ;
            f[j][i] = f[f[j][i-]][i-];
        }
}

void Hash_a()
{
    for (int i = ; i <= n; ++i)
        temp[i] = w[i];
    sort(temp+, temp+n+);
    s_num = ;
    for (int i = ; i <= n; ++i)
        if (temp[i] != temp[i-] || i == )
        {
            temp[++s_num] = temp[i];
            id[temp[i]] = s_num;
        }
}

void dfs(int u)
{
    root[u] = ++T.cnt;
    T.insert(u ==  ?  : root[f[u][]], root[u], , s_num, id[w[u]], );
    for (int i = head[u]; i != -; i = e[i].next)
    {
        int v = e[i].v;
        if (v == f[u][])
            continue ;
        dfs(v);
    }
}

void Build_tree()
{
    Hash_a();
    dfs();/*
    for (int i = 1; i <= n; ++i)
        printf("%d %d\n", i, T.sum[root[i]]);*/
}

void prepare()
{
    Build_lca();
    Build_tree();
}

int lca(int u, int v)
{
    if (depth[u] < depth[v])
        swap(u, v);
    for (int i = ; i >= ; --i)
    {
        if (f[u][i] == -)
            continue ;
        if (depth[f[u][i]] >= depth[v])
        {
            u = f[u][i];
            if (depth[u] == depth[v])
                break ;
        }
    }
    if (u == v)
        return u;
    for (int i = ; i >= ; --i)
    {
        if (f[u][i] == -)
            continue ;
        if (f[u][i] != f[v][i])
        {
            u = f[u][i];
            v = f[v][i];
        }
    }
    return f[u][];
}

void work()
{
    prepare();
    while (m --)
    {
        int u, v, k;
        scanf("%d %d %d", &u, &v, &k);
        int t = lca(u, v);
        int pos = T.query(root[u], root[v], root[t], id[w[t]], , s_num, k);
        printf("%d\n", temp[pos]);
    }
}

int main()
{
    in();
    work();
    return ;
}