传送门:牛客
题目描述
奶牛们又一次试图创建一家创业公司,还是没有从过去的经验中吸取教训–牛是可怕的管理者!
为了方便,把奶牛从
1
∼
n
1\sim n
1∼n 编号,把公司组织成一棵树,1 号奶牛作为总裁(这棵树的根节点)。除了总裁以外的每头奶牛都有一个单独的上司(它在树上的 “双亲结点”)。
所有的第
i
i
i 头牛都有一个不同的能力指数
p
i
p_i
pi,描述了她对其工作的擅长程度。如果奶牛
i
i
i 是奶牛
j
j
j 的祖先节点,那么我们我们把奶牛
j
j
j 叫做
i
i
i 的下属。
不幸地是,奶牛们发现经常发生一个上司比她的一些下属能力低的情况,在这种情况下,上司应当考虑晋升她的一些下属。你的任务是帮助奶牛弄清楚这是什么时候发生的。简而言之,对于公司的中的每一头奶牛
i
i
i,请计算其下属
j
j
j 的数量满足
p
j
>
p
i
p_j > p_i
pj>pi。
输入:
5
804289384
846930887
681692778
714636916
957747794
1
1
2
3
输出:
2
0
1
0
0
刚开始我以为是一道树链剖分的题目.然后发现这道题本质上应该是一道树上 d f s dfs dfs的题目
当然树链剖分暴力解决这道题也是可以做的,对于树剖,复杂度时log级别的.可以考虑使用树剖将树形结构转化为线性结构,然后考虑维护区间逆序对个数.此时我们无法使用线段树进行维护.对于维护区间逆序对个数,我们考虑使用分块进行维护,朴素莫队可以在 N N l o g N N\sqrt{N}logN NNlogN解决.复杂度还是满足本题的.但是使用上述方法解决本题是在是杀鸡用牛刀,本题还是用不到那么复杂的维护方法的
对于本题来说,我们只需要计算一个节点的所有儿子的贡献即.因为父亲对儿子是没有贡献的,所以我们考虑使用 d f s dfs dfs来解决这道题.使用权值树状数组(当然权值线段树也是可以的)来维护每一个权值.那么对于一个节点 u u u来说,我们只需要遍历他的所有儿子节点,并且将所有权值都存入权值树状数组中,然后对于 u u u节点的答案来说,就是当前比他大的节点的个数.但是这么做的话存在一个问题,因为我们的这棵BIT此时存的不只是当前结点的儿子的权值,之前遍历其他节点的时候也存了,所以此时会导致一些错误.解决此问题的方案是,在遍历该节点的儿子节点之前先减去之前所有已在节点的贡献即可.这样的话就相当于减去了其他不满足子树的贡献
本题需要进行离散化操作
下面是具体的代码部分:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
ll x=0,w=1;char ch=getchar();
for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int tree[maxn];int n;
int lowbit(int x) {
return x&(~x+1);
}
void Add(int pos,int val) {
while(pos<=n) {
tree[pos]+=val;
pos+=lowbit(pos);
}
}
int query(int pos) {
int ans=0;
while(pos) {
ans+=tree[pos];
pos-=lowbit(pos);
}
return ans;
}
int w[maxn];vector<int>v;int Size;
vector<int>edge[maxn];int ans[maxn];
int get_id(int x) {
return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
void dfs1(int u,int per_u) {
int x=get_id(w[u]);
ans[u]-=query(Size)-query(x);
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==per_u) continue;
dfs1(v,u);
}
ans[u]+=query(Size)-query(x);
Add(x,1);
}
int main() {
n=read();
for(int i=1;i<=n;i++) {
w[i]=read();
v.push_back(w[i]);
}
sort(v.begin(),v.end());
Size=v.size();
for(int i=2;i<=n;i++) {
int u=read();
edge[u].push_back(i);
}
dfs1(1,0);
for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
return 0;
}