P3605 Promotion Counting P(树状数组+dfs

#include 
//using int_max = 0x3f3f3f3f;
#define long_max 9223372036854775807;
using namespace std;
typedef long long ll;
typedef pair PII;
typedef pair PDD;
using VI = vector;
typedef unsigned long long ull;
const int mod = 1e8 - 3;
int n;
VI g[500050];
PII a[500050];
int q[500050];
int res[500050];
int tree[500050];

int lowbit(int x){
    return x & -x;
}
int query(int pos){
    int s = 0;
    for(int i = pos;i;i -= lowbit(i)){
        s += tree[i];
    }
    return s;
}
void change(int pos,int x){
    for(int i=pos;i<=n;i += lowbit(i)){
        tree[i] += x;
    }
}
void dfs(int now){

    res[now] = -(query(n) - query(q[now]));
    for(int i=0;i>n;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        a[i] = {x,i};
    }
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++){
        q[a[i].second] = i;
    }

    for(int i=2;i<=n;i++){
        int x;
        cin>>x;
        g[x].push_back(i);
    }
    dfs(1);
    for(int i=1;i<=n;i++){
        cout<

考虑在一条dfs 路径上,只关心子节点是否比当前节点的值大

res[x] = -已经大于该节点的数  + 总共比该节点大的数

所以先将前序比该节点的大的数都去掉,然后将后序中的数字插入,

你可能感兴趣的:(数据结构题,算法,深度优先,数据结构)