ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

BZOJ4756: [Usaco2017 Jan]Promotion Counting(线段树合并)

2019-02-18 08:49:39  阅读:301  来源: 互联网

标签:Usaco2017 int BZOJ4756 mid Jan SS MAXN ql getchar


题意

题目链接

Sol

线段树合并板子题

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 400000, SS = MAXN * 21;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int N, p[MAXN], fa[MAXN], root[MAXN], tot, ans[MAXN];
vector<int> v[MAXN];
int ls[SS], rs[SS], sum[SS];
void insert(int &k, int l, int r, int p, int v) {
    if(!k) k = ++tot;
    sum[k]++;
    if(l == r) return ;
    int mid = l +  r >> 1;
    if(p <= mid) insert(ls[k], l, mid, p, v);
    else insert(rs[k], mid + 1, r, p, v);
}
int Query(int k, int l, int r, int ql, int qr) {
    if(ql <= l && r <= qr) return sum[k];
    int mid = l + r >> 1;
    if(ql > mid) return Query(rs[k], mid + 1, r, ql, qr);
    else if(qr <= mid) return Query(ls[k], l, mid, ql, qr);
    else return Query(ls[k], l, mid, ql, qr) + Query(rs[k], mid + 1, r, ql, qr);
}
int Merge(int x, int y) {
    if(!x || !y) return x ^ y;
    ls[x] = Merge(ls[x], ls[y]);
    rs[x] = Merge(rs[x], rs[y]);
    sum[x] += sum[y];
    return x;
}
void dfs(int x) {
    for(int i = 0; i < v[x].size(); i++) {
        int to = v[x][i]; dfs(to);
        root[x] = Merge(root[x], root[to]);
    }
    ans[x] = Query(root[x], 1, N, p[x] + 1, N);
    insert(root[x], 1, N, p[x], 1);
}
void Des() {
    static int date[MAXN], num = 0;
    for(int i = 1; i <= N; i++) date[++num] = p[i];
    sort(date + 1, date + num + 1);
    num = unique(date + 1, date + num + 1) - date - 1;
    for(int i = 1; i <= N; i++) p[i] = lower_bound(date + 1, date + N + 1, p[i]) - date;
}
int main() {
    N = read();
    for(int i = 1; i <= N; i++) p[i] = read();
    Des();
    for(int i = 2; i <= N; i++) fa[i] = read(), v[fa[i]].push_back(i);
    dfs(1);
    for(int i = 1; i <= N; i++) printf("%d\n", ans[i]);
    return 0;
}
/*
*/

标签:Usaco2017,int,BZOJ4756,mid,Jan,SS,MAXN,ql,getchar
来源: https://www.cnblogs.com/zwfymqz/p/10393690.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有