洛谷 3605 [USACO17JAN]Promotion Counting晋升者计数(线段树合并)

版权声明:没人会转的( ̄▽ ̄) https://blog.csdn.net/j2_o2/article/details/89329577
题目链接
题意

一颗树,每个节点有个权值。
对于每个节点,求子树中存在多少节点大于他的权值

思路

先离散化,对每个节点建线段树,从叶子节点开始合并,求值。

代码
#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;
}
展开阅读全文

没有更多推荐了,返回首页