bzoj4756

http://www.lydsy.com/JudgeOnline/problem.php?id=4756

水题一枚。。。但是我写了一个小时。。。手贱打反查不出来。。。

就是每次线段树合并，先把自己的儿子都合并了， 最后和自己合并。。。

#include<bits/stdc++.h>
using namespace std;
const int N = ;
vector<int> G[N];
int n, sum, cnt;
int t[N], id[N], size[N * ], lc[N * ], rc[N * ], ans[N], root[N * ];
bool cp(int i, int j) { return t[i] < t[j]; }
void update(int l, int r, int &x, int pos)
{
    x = ++cnt; int mid = (l + r) >> ;
    ++size[x]; if(l == r) return;
    if(pos <= mid) update(l, mid, lc[x], pos);
    else update(mid + , r, rc[x], pos);
}
int merge(int x, int y)
{
    if(!x || !y) return x + y;
    sum += size[lc[x]] * size[rc[y]];
    lc[x] = merge(lc[x], lc[y]);
    rc[x] = merge(rc[x], rc[y]);
    size[x] += size[y];
    return x;
}
void dfs(int u)
{
    int last = ;
    for(int i = ; i < G[u].size(); ++i)
    {
        int v = G[u][i];
        dfs(v); root[v] = merge(root[v], root[last]);
        last = v;
    }
    sum = ; root[u] = merge(root[u], root[last]);
    ans[u] = sum;
}
int main()
{
    scanf("%d", &n);
    for(int i = ; i <= n; ++i)
    {
        scanf("%d", &t[i]); id[i] = i;
    }
    sort(id + , id + n + , cp);
    for(int i = ; i <= n; ++i) t[id[i]] = t[id[i - ]] + (t[id[i]] != t[id[i - ]]);
    for(int i = ; i <= n; ++i) update(, n, root[i], t[i]);
    for(int i = ; i <= n; ++i)
    {
        int x; scanf("%d", &x);
        G[x].push_back(i);
    }
    dfs();
    for(int i = ; i <= n; ++i) printf("%d\n", ans[i]);
    return ;
}