洛谷P3605 [USACO17JAN]Promotion Counting P

洛谷P3605 [USACO17JAN]Promotion Counting P

题目传送门
思路：
这道看起来很想线段树合并和树上启发式合并的题，而且也确实能这么做。
但是这两种做法实现起来却并不简单。
于是我们考虑用实现简单好理解的 $dfs+$ 树状数组实现。
树状数组肯定是用来求逆序对的，所以先要离散化。
设 $an{s}_i$ 表示 $i$ 节点的答案，那么 $an{s}_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;
}