BZOJ3524 POI2014 Couriers 题解&代码

26 篇文章 0 订阅
5 篇文章 0 订阅

题意:给出一个长度为n的序列a满足1≤a[i]≤n。
又有m组询问,每次对于一个区间[l,r]问是否存在一个数在[l,r]中出现的次数大于(r-l+1)/2。如果存在,输出这个数,否则输出0。
思路:嘛,很纯粹的主席树…只不过空间限定太严格了点,竟然卡空间,我卡了好几次才AC…
不过我现在特别纠结为啥我BZOJ2223过不了

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 500005;
int n,m,l,r,x,cnt,rt[maxn],son[2][maxn*20],sum[maxn*20];
void Insert(int rl,int rn,int l,int r,int v)
{
    sum[rn]=sum[rl]+1;
    if(l==r)return;
    int mid = (l+r)/2;
    if(mid>=v) son[1][rn]=son[1][rl],son[0][rn]=++cnt,Insert(son[0][rl],son[0][rn],l,mid,v);
    else son[0][rn]=son[0][rl],son[1][rn]=++cnt,Insert(son[1][rl],son[1][rn],mid+1,r,v);

}
int query(int rl,int rn,int l,int r,int k)
{
    if(l==r)return l;
    int mid = (l+r)/2;
    if(sum[son[0][rn]]-sum[son[0][rl]]>k) return query(son[0][rl],son[0][rn],l,mid,k);
    if(sum[son[1][rn]]-sum[son[1][rl]]>k) return query(son[1][rl],son[1][rn],mid+1,r,k);
    return 0;
}
int main(void)
{
    scanf("%d%d",&n,&m);
    for(int i = 1; i <= n; i++)rt[i]=++cnt;
    for(int i = 1; i <= n; i++)
    {
        scanf("%d",&x);
        Insert(rt[i-1],rt[i],1,n,x);
    }
    for(int i = 1; i <= m; i++)
    {
        scanf("%d%d",&l,&r);
        printf("%d\n",query(rt[l-1],rt[r],1,n,(r-l+1)/2));
    }
    return 0;
}
以下是一份使用Treap实现的C++代码,仅供参考。 ```c++ #include <bits/stdc++.h> using namespace std; const int INF = 0x3f3f3f3f; const int MAXN = 1e5 + 5; int n, m; int root, cnt; struct Node { int val, key; int size; int ch[2]; } tr[MAXN]; inline int new_node(int val) { tr[++cnt].val = val; tr[cnt].key = rand(); tr[cnt].size = 1; return cnt; } inline void pushup(int p) { tr[p].size = 1 + tr[tr[p].ch[0]].size + tr[tr[p].ch[1]].size; } inline void split(int p, int k, int &x, int &y) { if (!p) { x = y = 0; return; } if (tr[p].val <= k) { x = p; split(tr[p].ch[1], k, tr[p].ch[1], y); } else { y = p; split(tr[p].ch[0], k, x, tr[p].ch[0]); } pushup(p); } inline int merge(int x, int y) { if (!x || !y) return x + y; if (tr[x].key < tr[y].key) { tr[x].ch[1] = merge(tr[x].ch[1], y); pushup(x); return x; } else { tr[y].ch[0] = merge(x, tr[y].ch[0]); pushup(y); return y; } } inline void insert(int val) { int x, y; split(root, val, x, y); root = merge(merge(x, new_node(val)), y); } inline void remove(int val) { int x, y, z; split(root, val, x, z); split(x, val - 1, x, y); y = merge(tr[y].ch[0], tr[y].ch[1]); root = merge(merge(x, y), z); } inline int kth(int k) { int p = root; while (p) { if (tr[tr[p].ch[0]].size >= k) { p = tr[p].ch[0]; } else if (tr[tr[p].ch[0]].size + 1 == k) { return tr[p].val; } else { k -= tr[tr[p].ch[0]].size + 1; p = tr[p].ch[1]; } } return -1; } inline int query_min() { int p = root; int res = INF; while (p) { res = min(res, tr[p].val); p = tr[p].ch[0]; } return res; } int main() { srand(time(NULL)); scanf("%d%d", &n, &m); for (int i = 1, opt, x; i <= n; ++i) { scanf("%d%d", &opt, &x); if (opt == 1) { insert(x); } else { remove(tr[cnt].val); } } printf("%d\n", query_min()); while (m--) { int k; scanf("%d", &k); printf("%d\n", kth(k)); } return 0; } ``` 代码实现中使用了一个结构体 `Node` 来表示平衡树的每一个节点。其中 `val` 表示节点存储的值,`key` 表示随机生成的优先级,`size` 表示以该节点为根的子树大小,`ch[0]` 和 `ch[1]` 分别表示左右儿子的下标。接下来就是一些经典的平衡树操作,例如 `split` 和 `merge` 用于分裂和合并平衡树,`insert` 和 `remove` 用于插入和删除节点,`kth` 和 `query_min` 用于查询平衡树中第 $k$ 小的数和最小值。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值