• 主席树 入门


    思想类似前缀和,访问某状态的线段树可通过末减初状态进行求解。

    hdu4417 第二道模板题

    有很多细节需要注意。

    1.题目给定ai的高度可能为0,但通过离散化事实上不影响结论。

    2.给定的访问区间[x,y]以及高度h也可能为零,因而x,y需对应++。查询依旧是root[y] - root[x-1]。

    3.对于高度h,为了获得其所对应的离散化后的高度,应该在b数组(离散化前,排序去重后的数组)中跑一遍upper_bound()。

    4.upper_bound(b + 1, b + sz + 1, h)返回第一个严格大于h的下标,此处返回的下标应该减一,从而符合题目“小于等于h的均能被hit”的规律。

    5.此处比较巧妙的点在于,当返回的是原数据最小,也即返回下标1,则减一为0,因而可通过!k进行判断是否有解。无解直接输出0。

     1 #include<iostream>
     2 #include<algorithm>
     3 #include<cstdio>
     4 #include<cstring>
     5 #include<iomanip>
     6 using namespace std;
     7 
     8 const int maxn = 1e5 + 10;
     9 int n, q;
    10 int a[maxn], b[maxn];//b用于记录去重后离散化前的a
    11 int root[maxn];//存储每棵树的根节点对应编号(cnt)
    12  
    13 int cnt = 0;//标记可以使用的新节点 
    14 
    15 struct node
    16 {
    17     int l, r, val;    
    18 }tree[maxn << 5];
    19 
    20 void init()
    21 {
    22     cnt = 0;
    23     for(int i = 1 ; i <= n ; i++){
    24         a[i] = b[i] = root[i] = 0;
    25         tree[i].l = tree[i].r = tree[i].val = 0;
    26     }
    27 }
    28 
    29 int upd(int pre, int pl, int pr, int x)
    30 {
    31     int now = ++cnt;
    32     tree[now].l = tree[pre].l;
    33     tree[now].r = tree[pre].r;
    34     tree[now].val = tree[pre].val + 1;
    35     int mid = (pl + pr) >> 1;
    36     if(pl < pr){
    37         if(x <= mid){//修改左子树 
    38             tree[now].l = upd(tree[pre].l, pl, mid, x);
    39         }else{//修改右子树
    40             tree[now].r = upd(tree[pre].r, mid + 1, pr, x);
    41         }
    42     }
    43     return now;//返回节点编号 
    44 }
    45 
    46 int query(int u, int v, int pl, int pr, int h)
    47 {
    48     if(pl == pr)    return tree[v].val - tree[u].val;
    49     int mid = (pl + pr) >> 1;
    50     if(h <= mid){//线段树mid位于左端点 
    51         return query(tree[u].l, tree[v].l, pl, mid, h);
    52     }else{
    53         return tree[tree[v].l].val - tree[tree[u].l].val + query(tree[u].r, tree[v].r, mid + 1, pr, h);    
    54     }
    55 }
    56 
    57 int main(){
    58     int T;scanf("%d",&T);
    59     for(int cas = 1 ; cas <= T ; cas++){
    60         printf("Case %d:
    ",cas);
    61         scanf("%d%d",&n,&q);
    62         for(int i = 1 ; i <= n ; i++){
    63             scanf("%d",&a[i]);
    64             b[i] = a[i];
    65         }
    66         sort(b + 1, b + n + 1);
    67         int sz = unique(b + 1, b + n + 1) - b - 1;//不重复的元素个数
    68     
    69         for(int i = 1 ; i <= n ; i++){
    70             a[i] = lower_bound(b + 1, b + sz + 1, a[i]) - b;//范围记得注意 
    71             
    72             root[i] = upd(root[i - 1], 1, sz, a[i]);
    73         //建第i棵线段树,root[i]是第i棵线段树的根结点
    74         }
    75         while(q--){
    76             int x, y, h;
    77             scanf("%d%d%d",&x,&y,&h);
    78             x++;y++;
    79             int k = upper_bound(b + 1, b + sz + 1, h) - b - 1;//upper_bound necessity
    80             if(!k)    printf("0
    ");
    81             else    printf("%d
    ",query(root[x - 1], root[y], 1, sz, k));
    82             //第y棵线段树减第x-1棵线段树,就是区间[x,y]的线段树
    83         }
    84         init();
    85     }
    86     
    87     return 0;
    88 }
  • 相关阅读:
    阿里云提示Discuz memcache+ssrf GETSHELL漏洞如何解决
    mysql总是无故退出, InnoDB: mmap(68681728 bytes) failed; errno 12
    ElasticSearch6 报错FORBIDDEN/12/index read-only / allow delete (api)
    Yum安装时提示多库版本保护 Multilib version problems found
    redis:CLUSTER cluster is down 解决方法
    虚拟主机如何绑定网站根目录到子目录中
    iscroll遇到的两个坑
    前端之Sass/Scss实战笔记
    JS与jquery书写插件规范
    css命名推荐
  • 原文地址:https://www.cnblogs.com/ecustlegendn324/p/13803814.html
Copyright © 2020-2023  润新知