bzoj4771 七彩树

Posted 2021-01-12 llppdd

七彩树

Time Limit: 5 Sec Memory Limit: 256 MB

Description

给定一棵n个点的有根树，编号依次为1到n，其中1号点是根节点。每个节点都被染上了某一种颜色，其中第i个节
点的颜色为c[i]。如果c[i]=c[j]，那么我们认为点i和点j拥有相同的颜色。定义depth[i]为i节点与根节点的距离
，为了方便起见，你可以认为树上相邻的两个点之间的距离为1。站在这棵色彩斑斓的树前面，你将面临m个问题。
每个问题包含两个整数x和d，表示询问x子树里且depth不超过depth[x]+d的所有点中出现了多少种本质不同的颜色
。请写一个程序，快速回答这些询问。

Input

第一行包含一个正整数T(1<=T<=500)，表示测试数据的组数。
每组数据中，第一行包含两个正整数n(1<=n<=100000)和m(1<=m<=100000)，表示节点数和询问数。
第二行包含n个正整数，其中第i个数为ci，分别表示每个节点的颜色。
第三行包含n-1个正整数，其中第i个数为fi+1，表示节点i+1的父亲节点的编号。
接下来m行，每行两个整数x(1<=x<=n)和d(0<=d<n)，依次表示每个询问。
输入数据经过了加密，对于每个询问，如果你读入了x和d，那么真实的x和d分别是x xor last和d xor last，
其中last表示这组数据中上一次询问的答案，如果这是当前数据的第一组询问，那么last=0。
输入数据保证n和m的总和不超过500000。

Output

对于每个询问输出一行一个整数，即答案。

Sample Input

1
5 8
1 3 3 2 2
1 1 3 3
1 0
0 0
3 0
1 3
2 1
2 0
6 2
4 1

Sample Output

1
2
3
1
1
2
1
1






本来是个离线题的，就是说你按深度插入每个点。（当然要先树剖一下啦！）
单独维护每一个颜色，两个相邻的点就在lca的地方减一就好了。
但是他是个在线就的很毒。。。
我们以毒攻毒，用可持久化线段树。。。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
struct lpl{
    int fa, dfn, top, son, size, deep;
}a[maxn];
struct ld{
    int id, dfn;
    inline bool operator < (const ld &A)const{return dfn < A.dfn;}
};
struct lpd{
    int sum, ls, rs;
}node[10000010];
int n, m, c[maxn];
int lastans, last, num, tot, opt, cnt, mxd, root = 1;
int d[maxn], rt[maxn * 5], ot[maxn], dp[maxn];
vector<int> point[maxn];
set<ld> s[maxn];
set<ld>::iterator iter, op, ed;
queue<int> q;

namespace HLD{
    void dfs1(int t, int fa){
        a[t].size = 1; int tmp = 0; mxd = max(mxd, a[t].deep);
        for(int i = point[t].size() - 1; i >= 0; --i){
            int now = point[t][i]; if(now == fa) continue;
            a[now].deep = a[t].deep + 1; dfs1(now, t);
            a[t].size += a[now].size; 
            if(tmp < a[now].size){tmp = a[now].size; a[t].son = now;}
        }
    }
    
    void dfs2(int t, int fa){
        a[t].dfn = ++num;
        if(a[t].son){a[a[t].son].top = a[t].top; dfs2(a[t].son, t);}
        for(int i = point[t].size() - 1; i >= 0; --i){
            int now = point[t][i]; if(now == fa || now == a[t].son) continue;
            a[now].top = now; dfs2(now, t);
        }
    }
    
    inline int LCA(int x1, int x2){
        if(a[a[x1].top].deep < a[a[x2].top].deep) swap(x1, x2);
        while(a[x1].top != a[x2].top){
            x1 = a[a[x1].top].fa; 
            if(a[a[x1].top].deep < a[a[x2].top].deep) swap(x1, x2);
        }
        return (a[x1].deep > a[x2].deep) ? x2 : x1;
    }
}

namespace SGT{
    inline void update(int t){node[t].sum = node[node[t].ls].sum + node[node[t].rs].sum;}
    
    void build(int &t, int l, int r){
        t = ++cnt; node[t].sum = 0; if(l == r) return;
        int mid = (l + r) >> 1;
        build(node[t].ls, l, mid); build(node[t].rs, mid + 1, r);
    }
    
    void Modify(int &t, int l, int r, int pos, int val){
        t = ++cnt; node[t] = node[last];
        if(l == r){node[t].sum += val; return;}
        int mid = (l + r) >> 1;
        if(pos <= mid){last = node[last].ls; Modify(node[t].ls, l, mid, pos, val);}
        else{last = node[last].rs; Modify(node[t].rs, mid + 1, r, pos, val);}
        update(t);
    }
    
    int Query(int t, int l, int r, int ql, int qr){
        if(l == ql && r == qr) return node[t].sum;
        int mid = (l + r) >> 1;
        if(qr <= mid) return Query(node[t].ls, l, mid, ql, qr);
        if(mid < ql) return Query(node[t].rs, mid + 1, r, ql, qr);
        return Query(node[t].ls, l, mid, ql, mid) + Query(node[t].rs, mid + 1, r, mid + 1, qr);
    }
}

template<typename T>
inline void read(T &x){
    x = 0; static char c = getchar();
    while(!isdigit(c)) c = getchar();
    while(isdigit(c)){x = x * 10 + c - '0'; c = getchar();}
}

template<typename T>
void write(T x){
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}

inline void putit(){
    read(n); read(m); lastans = 0; mxd = 0;
    for(int i = 1; i <= n; ++i) {
        point[i].clear(); s[i].clear(); 
        a[i].son = 0; read(c[i]);
    }
    for(int f, i = 1; i < n; ++i){
        read(f); a[i + 1].fa = f; 
        point[f].push_back(i + 1);
    }
}

inline void prepare(){
    a[root].deep = 1; num = 0; memset(rt, 0, sizeof(rt));
    HLD::dfs1(root, 0); a[1].top = 1; HLD::dfs2(root, 0);
    q.push(root); int now; tot = 0;
    while(!q.empty()){
        now = q.front(); q.pop(); d[++tot] = now; 
        for(int i = point[now].size() - 1; i >= 0; --i) q.push(point[now][i]);
    }
    cnt = 0; SGT::build(rt[0], 1, n); opt = 0;
    bool qian, hou;
    for(int color, i = 1; i <= tot; ++i){
        color = c[d[i]];
        iter = s[color].insert((ld){d[i], a[d[i]].dfn}).first;
        qian = (iter != s[color].begin()); iter++; hou = (iter != s[color].end()); iter--;
        if(qian){iter--; op = iter; iter++;}
        if(hou){iter++; ed = iter; iter--;} 
        if(qian && hou){
            last = rt[opt]; 
            SGT::Modify(rt[++opt], 1, n, a[HLD::LCA(op -> id, ed -> id)].dfn, 1);
        }
        if(qian){
            last = rt[opt];
            SGT::Modify(rt[++opt], 1, n, a[HLD::LCA(op -> id, iter -> id)].dfn, -1);    
        }
        if(hou){
            last = rt[opt];
            SGT::Modify(rt[++opt], 1, n, a[HLD::LCA(iter -> id, ed -> id)].dfn, -1);    
        }
        last = rt[opt]; SGT::Modify(rt[++opt], 1, n, iter -> dfn, 1);
        ot[d[i]] = opt; dp[a[d[i]].deep] = d[i];
    }
}

inline void workk(){
    int x, D;
    while(m--){
        read(x); read(D); 
        x ^= lastans; D ^= lastans;
        D += a[x].deep; D = min(D, mxd);
        D = ot[dp[D]];
        lastans = SGT::Query(rt[D], 1, n, a[x].dfn, a[x].dfn + a[x].size - 1);
        write(lastans); putchar(10);
    }
}

int main()
{
    int T; read(T); 
    while(T--){
        putit();
        prepare();
        workk();        
    } 
    return 0;
}