洛谷P3605 [USACO17JAN]Promotion Counting P
题目传送门
思路:
这道看起来很想线段树合并和树上启发式合并的题,而且也确实能这么做。
但是这两种做法实现起来却并不简单。
于是我们考虑用实现简单好理解的
d
f
s
+
dfs+
dfs+ 树状数组实现。
树状数组肯定是用来求逆序对的,所以先要离散化。
设
a
n
s
i
ans_i
ansi 表示
i
i
i 节点的答案,那么
a
n
s
i
ans_i
ansi
=
=
= 加上
i
i
i 的子树后比
i
i
i 强的
−
-
− 原先就比
i
i
i 强的。
这一步我们可以在回溯过程中实现。
代码:
#include<bits/stdc++.h>
#define freo(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout)
#define ll long long
#define lowbit(x) (x&(-x))
using namespace std;
const int maxn=1e5+5;
int a[maxn],b[maxn],head[maxn],ecnt,n,tree[maxn],ans[maxn];
struct edge
{
int nxt,to;
}e[maxn];
inline ll read()
{
ll ret=0;char ch=' ',c=getchar();
while(!(c<='9'&&c>='0')) ch=c,c=getchar();
while(c<='9'&&c>='0') ret=(ret<<1)+(ret<<3)+c-'0',c=getchar();
return ch=='-'?-ret:ret;
}
void add(int x,int y)
{
e[++ecnt]=(edge){head[x],y};
head[x]=ecnt;
}
void modify(int x,int v) {for(int i=x;i<=n;i+=lowbit(i)) tree[i]+=v;}
int query(int x)
{
int ret=0;
for(int i=x;i;i-=lowbit(i)) ret+=tree[i];
return ret;
}
void dfs(int u)
{
ans[u]-=query(n)-query(a[u]);
for(int i=head[u];i;i=e[i].nxt) dfs(e[i].to);
ans[u]+=query(n)-query(a[u]);
modify(a[u],1);
}
int main()
{
//freo("1");
n=read();
for(int i=1;i<=n;i++) b[i]=read(),a[i]=b[i];
sort(b+1,b+n+1);
for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+n+1,a[i])-b;
for(int i=2;i<=n;i++)
{
int x=read();
add(x,i);
}
dfs(1);
for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
return 0;
}