题目链接
题意
一颗树,每个节点有个权值。
对于每个节点,求子树中存在多少节点大于他的权值
思路
先离散化,对每个节点建线段树,从叶子节点开始合并,求值。
代码
#include <bits/stdc++.h>
using namespace std;
const int N = 100005;
const int NN = 100005*30;
int a[N], ans[N], n, maxn, id, root[NN];
vector<int> lis, e[N];
struct Node
{
int l, r, sum;
}t[NN];
void ins(int &p, int l, int r, int pos)
{
p = ++id;
t[p].sum = 1;
if(l == r) return;
int mid = l+r >> 1;
if(pos <= mid) ins(t[p].l, l, mid, pos);
else ins(t[p].r, mid+1, r, pos);
}
int mer(int x, int y)
{
if(!x) return y;
if(!y) return x;
t[x].l = mer(t[x].l, t[y].l);
t[x].r = mer(t[x].r, t[y].r);
t[x].sum = t[t[x].l].sum + t[t[x].r].sum;
return x;
}
int query(int p, int l, int r, int ql, int qr)
{
if(ql <= l && r <= qr) return t[p].sum;
int mid = l+r >> 1;
if(qr <= mid) return query(t[p].l, l, mid, ql, qr);
if(ql > mid) return query(t[p].r, mid+1, r, ql, qr);
return query(t[p].l, l, mid, ql, qr) + query(t[p].r, mid+1, r, ql, qr);
}
void dfs(int u, int fa)
{
ins(root[u], 1, maxn, a[u]);
for(auto v : e[u])
{
if(v == fa) continue;
dfs(v, u);
root[u] = mer(root[v], root[u]);
}
ans[u] = query(root[u], 1, maxn, a[u]+1, maxn);
}
int main()
{
scanf("%d",&n);
for(int i = 1; i <= n; ++i) scanf("%d",&a[i]), lis.push_back(a[i]);
sort(lis.begin(),lis.end());
lis.insert(lis.begin(),lis[0]-1);
lis.erase(unique(lis.begin(),lis.end()) ,lis.end());
for(int i = 1; i <= n; ++i) a[i] = lower_bound(lis.begin(),lis.end(),a[i]) - lis.begin();
maxn = lis.size(); // 不减1,避免查询越界死循环
for(int i = 2; i <= n; ++i)
{
int tmp;
scanf("%d",&tmp);
e[tmp].push_back(i);
e[i].push_back(tmp);
}
id = 0;
dfs(1,0);
for(int i = 1; i <= n; ++i) printf("%d\n",ans[i]);
return 0;
}